def schema_database(db_factory): dbname = "procrastinate_schema" db_factory(dbname=dbname) # apply the current procrastinate schema to the "procrastinate_schema" database connector = aiopg_connector.AiopgConnector(dbname=dbname) connector.open() schema_manager = schema.SchemaManager(connector=connector) schema_manager.apply_schema() connector.close() return dbname
def setup_db(): dbname = "procrastinate_test_template" db_create(dbname=dbname) connector = aiopg_connector.AiopgConnector(dbname=dbname) schema_manager = schema.SchemaManager(connector=connector) schema_manager.apply_schema() # We need to close the psycopg2 underlying connection synchronously connector.close() yield dbname db_drop(dbname=dbname)
def schema_database(db_factory, pum): dbname = "procrastinate_schema" db_factory(dbname=dbname) # apply the current procrastinate schema to procrastinate_schema connector = aiopg_connector.AiopgConnector(dbname=dbname) schema_manager = schema.SchemaManager(connector=connector) schema_manager.apply_schema() connector.close() # set the baseline version in procrastinate_schema # This db is as far as can be. pum("baseline", f"--pg_service {dbname} --baseline 999.999.999") return dbname
def schema_manager(self) -> schema.SchemaManager: return schema.SchemaManager(connector=self.connector)