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)