def db(local_config): db = PostgresDb.from_config(local_config, application_name='test-run', validate_db=False) # Drop and recreate tables so our tests have a clean db. drop_db(db._connection) remove_dynamic_indexes() ensure_db(db._engine) return db
def db(local_config, request): timezone = request.param db = PostgresDb.from_config(local_config, application_name='test-run', validate_connection=False) # Drop and recreate tables so our tests have a clean db. with db.connect() as connection: _core.drop_db(connection._connection) remove_dynamic_indexes() # Disable informational messages since we're doing this on every test run. with _increase_logging(_core._LOG) as _: _core.ensure_db(db._engine) c = db._engine.connect() c.execute('alter database %s set timezone = %r' % (local_config.db_database, str(timezone))) c.close() # We don't need informational create/drop messages for every config change. _dynamic._LOG.setLevel(logging.WARN) yield db db.close()
def db(local_config): db = PostgresDb.from_config(local_config, application_name='test-run', validate_db=False) # Drop and recreate tables so our tests have a clean db. drop_db(db._connection) remove_dynamic_indexes() ensure_db(db._engine) return db
def test_db_init(global_integration_cli_args, db, local_config): drop_db(db._connection) assert not has_schema(db._engine, db._connection) # Run on an empty database. opts = list(global_integration_cli_args) opts.extend(['-v', 'system', 'init']) cli_method = datacube.scripts.cli_app.cli result = _run_cli(cli_method, opts) assert result.exit_code == 0 assert 'Created.' in result.output assert has_schema(db._engine, db._connection)
def test_db_init(global_integration_cli_args, db, local_config): with db.connect() as connection: drop_db(connection._connection) assert not has_schema(db._engine, connection._connection) # Run on an empty database. cli_method = datacube.scripts.cli_app.cli result = _run_cli(global_integration_cli_args, cli_method, ['-v', 'system', 'init']) assert result.exit_code == 0 assert 'Created.' in result.output with db.connect() as connection: assert has_schema(db._engine, connection._connection)
def db(local_config: LocalConfig): db = PostgresDb.from_config(local_config, application_name='dea-test-run', validate_connection=False) # Drop and recreate tables so our tests have a clean db. with db.connect() as connection: _core.drop_db(connection._connection) remove_dynamic_indexes() # Disable informational messages since we're doing this on every test run. with _increase_logging(_core._LOG) as _: _core.ensure_db(db._engine) # We don't need informational create/drop messages for every config change. _dynamic._LOG.setLevel(logging.WARN) yield db db.close()
def test_db_init(global_integration_cli_args, db, local_config): drop_db(db._connection) assert not has_schema(db._engine, db._connection) # Run on an empty database. opts = list(global_integration_cli_args) opts.extend( [ '-v', 'system', 'init' ] ) cli_method = datacube.scripts.cli_app.cli result = _run_cli(cli_method, opts) assert result.exit_code == 0 assert 'Done.' in result.output assert has_schema(db._engine, db._connection)