def tearDown(self) -> None:
local_postgres_helpers.teardown_on_disk_postgresql_database(
OperationsBase)
local_postgres_helpers.teardown_on_disk_postgresql_database(
self.schema_base())
self.metadata_patcher.stop()
self.entity_matching_error_threshold_patcher.stop()
def test_enums_match_schema(self) -> None:
with runner(self.default_config(), self.engine) as r:
r.migrate_up_to("head")
# Fetch enum values
migration_enums = self.fetch_all_enums()
# Doing teardown/setup to generate a new postgres instance
local_postgres_helpers.restore_local_env_vars(self.overridden_env_vars)
local_postgres_helpers.stop_and_clear_on_disk_postgresql_database(
self.db_dir)
self.db_dir = local_postgres_helpers.start_on_disk_postgresql_database(
)
self.overridden_env_vars = (
local_postgres_helpers.update_local_sqlalchemy_postgres_env_vars())
local_postgres_helpers.use_on_disk_postgresql_database(
self.database_key)
# Check enum values
schema_enums = self.fetch_all_enums()
# Assert that they all match
self.assertEqual(len(migration_enums), len(schema_enums))
for enum_name, migration_values in migration_enums.items():
schema_values = schema_enums[enum_name]
self.assertCountEqual(migration_values, schema_values)
# Cleanup needed for this method.
local_postgres_helpers.teardown_on_disk_postgresql_database(
self.database_key)
def tearDown(self) -> None:
local_postgres_helpers.teardown_on_disk_postgresql_database(
self.operations_database_key)
local_postgres_helpers.teardown_on_disk_postgresql_database(
self.main_database_key)
self.metadata_patcher.stop()
self.entity_matching_error_threshold_patcher.stop()
def tearDown(self) -> None:
self.email_generation_patcher.stop()
self.project_id_patcher.stop()
self.gcs_file_system_patcher.stop()
self.get_secret_patcher.stop()
self.top_opps_email_recipient_patcher.stop()
local_postgres_helpers.teardown_on_disk_postgresql_database(self.database_key)
def test_enums_match_schema(self):
with runner(self.default_config(), self.engine) as r:
r.migrate_up_to("head")
# Fetch enum values
migration_enums = self.fetch_all_enums()
# Doing teardown/setup to generate a new postgres instance
local_postgres_helpers.restore_local_env_vars(self.overridden_env_vars)
local_postgres_helpers.stop_and_clear_on_disk_postgresql_database(
self.db_dir)
self.db_dir = local_postgres_helpers.start_on_disk_postgresql_database(
)
self.overridden_env_vars = (
local_postgres_helpers.update_local_sqlalchemy_postgres_env_vars())
declarative_base = SQLAlchemyEngineManager.declarative_method_for_schema(
self.schema_type)
local_postgres_helpers.use_on_disk_postgresql_database(
declarative_base)
# Check enum values
schema_enums = self.fetch_all_enums()
# Assert that they all match
self.assertEqual(len(migration_enums), len(schema_enums))
for enum_name, migration_values in migration_enums.items():
schema_values = schema_enums[enum_name]
self.assertEqual(
len(migration_values),
len(schema_values),
msg=f"{enum_name} lengths differ",
)
self.assertEqual(
len(migration_values),
len(migration_values.intersection(schema_values)),
msg=f"{enum_name} values differ",
)
# Cleanup needed for this method.
local_postgres_helpers.teardown_on_disk_postgresql_database(
declarative_base)
def tearDown(self) -> None:
local_postgres_helpers.teardown_on_disk_postgresql_database(
self.db_key)
local_postgres_helpers.restore_local_env_vars(self.env_vars)
def tearDown(self) -> None:
local_postgres_helpers.teardown_on_disk_postgresql_database(StateBase)
def tearDown(self) -> None:
self.auth0_client_patcher.stop()
local_postgres_helpers.teardown_on_disk_postgresql_database(
self.database_key)
def tearDown(self) -> None:
local_postgres_helpers.restore_local_env_vars(self.overridden_env_vars)
local_postgres_helpers.teardown_on_disk_postgresql_database(
SQLAlchemyDatabaseKey.for_schema(SchemaType.CASE_TRIAGE))
def tearDown(self) -> None:
local_postgres_helpers.teardown_on_disk_postgresql_database(StateBase)
self.bq_client_patcher.stop()
self.storage_client_patcher.stop()
self.task_client_patcher.stop()
def tearDown(self) -> None:
local_postgres_helpers.teardown_on_disk_postgresql_database(
JusticeCountsBase)
def tearDown(self) -> None:
local_postgres_helpers.teardown_on_disk_postgresql_database(
SQLAlchemyDatabaseKey.canonical_for_schema(SchemaType.STATE))
def tearDown(self) -> None:
local_postgres_helpers.teardown_on_disk_postgresql_database(
OperationsBase)
local_postgres_helpers.teardown_on_disk_postgresql_database(
self.schema_base())
def tearDown(self) -> None:
super().tearDown()
local_postgres_helpers.teardown_on_disk_postgresql_database(
self.database_key)
def tearDown(self) -> None:
local_postgres_helpers.restore_local_env_vars(self.overridden_env_vars)
local_postgres_helpers.teardown_on_disk_postgresql_database(
CaseTriageBase)
def tearDown(self) -> None:
local_postgres_helpers.teardown_on_disk_postgresql_database(
self.operations_key)
