def edb(pytestconfig: Config) -> ElectrolyteDB: """ Create and populate an EDB instance """ mock_allowed = not pytestconfig.option.edb_no_mock if ElectrolyteDB.can_connect(): _edb = ElectrolyteDB() else: if mock_allowed: _edb = MockDB() else: pytest.fail("EDB could not connect to a database instance, but mocking is not allowed") _load_bootstrap(_edb) yield _edb _reset(_edb)
def connect_to_edb(test_invalid_host=False): print("connecting to " + str(ElectrolyteDB.DEFAULT_URL)) db = ElectrolyteDB() connected = db.can_connect() return (db, connected)