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)
