43 lines
2.0 KiB
Python
43 lines
2.0 KiB
Python
import glob
|
|
|
|
def fix_file(path):
|
|
with open(path, 'r') as f:
|
|
content = f.read()
|
|
|
|
# 1. security_test.rs run_migrations
|
|
content = content.replace('db::run_migrations(&provider)', 'nx9_auth::db::run_migrations(&pool)')
|
|
content = content.replace('nx9_auth::db::run_sqlite_migrations(&pool)', 'nx9_auth::db::run_migrations(&pool)')
|
|
|
|
# Fix setup_test_db returning provider instead of pool? Wait! `setup_test_db` in `tests/security_test.rs` currently returns `(provider, db_path)`.
|
|
# Let me make it return `(provider, pool, db_path)` like integration_test.rs did.
|
|
content = content.replace(
|
|
'let provider = std::sync::Arc::new(nx9_auth::db::provider::SqliteProvider::new(pool));\n (provider, db_path)',
|
|
'let provider = std::sync::Arc::new(nx9_auth::db::provider::SqliteProvider::new(pool.clone()));\n (provider, pool, db_path)'
|
|
)
|
|
content = content.replace(
|
|
'let (provider, db_path) = setup_test_db().await;',
|
|
'let (provider, pool, db_path) = setup_test_db().await;'
|
|
)
|
|
|
|
# 2. security_test.rs user_repo
|
|
content = content.replace('user_repo::find_by_username(&provider,', 'provider.users().find_by_username(')
|
|
content = content.replace('user_repo::find_by_id(&provider,', 'provider.users().find_by_id(')
|
|
|
|
# role_repo
|
|
content = content.replace('role_repo::assign_role(&provider,', 'nx9_auth::identity::roles::assign_role(&provider,')
|
|
|
|
# 3. security_test.rs ServerConfig
|
|
content = content.replace(
|
|
'server: nx9_auth::config::ServerConfig {\n host: "127.0.0.1".into(),\n port: 8080,\n }',
|
|
'server: nx9_auth::config::ServerConfig {\n host: "127.0.0.1".into(),\n port: 8080,\n cookie_secure: false,\n production: false,\n }'
|
|
)
|
|
|
|
# 4. cli_test.rs create_sqlite_pool
|
|
content = content.replace('nx9_auth::db::create_sqlite_pool', 'nx9_auth::db::create_pool')
|
|
|
|
with open(path, 'w') as f:
|
|
f.write(content)
|
|
|
|
for f in glob.glob('tests/*.rs'):
|
|
fix_file(f)
|