def init_db(ec): print("initializing db") if "sqlite://" in ec.db_uri: directory = os.path.dirname(ec.db_path) if not os.path.exists(directory): os.makedirs(directory) smart = smartdb.SmartDatabaseClient(ec.db_uri) tries = 60 for i in range(tries): try: smart.create_base() @smartdb.cockroach_transaction def op(caller=init_db.__name__): with smart.new_session() as session: ops.health_check_purge(session) op(caller=init_db.__name__) return except ConnectionError as e: print("%d/%d %s" % (i + 1, tries, e)) time.sleep(1) raise ConnectionError(ec.db_uri)
def test_03(self): ss = smartdb.SmartDatabaseClient("sqlite:///%s/dbs/smart.sqlite" % (self.unit_path)) self.assertEqual(smartdb._SingleEndpoint, type(ss)) self.assertEqual(1, len(ss.engines)) ss.create_base() ss.connected_cockroach_session()
def test_05(self): ss = smartdb.SmartDatabaseClient( "cockroachdb://[email protected]:16257,cockroachdb://[email protected]:16257" ) self.assertEqual(smartdb._MultipleEndpoints, type(ss)) self.assertEqual(2, len(ss.engines)) with self.assertRaises(ConnectionError): with ss.new_session() as session: pass
def test_04(self): ss = smartdb.SmartDatabaseClient("cockroachdb://[email protected]:16257") self.assertEqual(smartdb._SingleEndpoint, type(ss)) self.assertEqual(1, len(ss.engines)) with self.assertRaises(ConnectionError): with ss.connected_cockroach_session() as session: pass with ss.new_session() as session: pass
def setUpClass(cls): uris = [ "cockroachdb://root@localhost:26257", "cockroachdb://root@localhost:26258", "cockroachdb://root@localhost:26259" ] db_uri = ",".join(uris) cls.smart = smartdb.SmartDatabaseClient(db_uri) cls.repositories = RepositoriesRegister(cls.smart) cls.set_up_class_checks(cls.smart)
def setUpClass(cls): db = "%s/%s.sqlite" % (cls.dbs_path, TestModelSQLiteMemory.__name__.lower()) try: os.remove(db) except OSError: pass assert os.path.isfile(db) is False db_uri = 'sqlite:///%s' % db cls.smart = smartdb.SmartDatabaseClient(db_uri) cls.repositories = RepositoriesRegister(cls.smart) cls.set_up_class_checks(cls.smart)
def setUpClass(cls): cls.smart = smartdb.SmartDatabaseClient(EC.db_uri) cls.p_matchbox = Process( target=TestEnjoliverCockroach.process_target_matchbox) cls.p_api = Process(target=TestEnjoliverCockroach.process_target_api) print("PPID -> %s\n" % os.getpid()) cls.p_matchbox.start() assert cls.p_matchbox.is_alive() is True cls.p_api.start() assert cls.p_api.is_alive() is True cls.api_running(EC.api_uri, cls.p_api) cls.smart.create_base()
def test_02(self): ss = smartdb.SmartDatabaseClient( "cockroachdb://[email protected]:26257,cockroachdb://[email protected]:26257,cockroachdb://[email protected]:26257" ) self.assertEqual(smartdb._MultipleEndpoints, type(ss)) self.assertEqual(3, len(ss.engines))
def test_01(self): ss = smartdb.SmartDatabaseClient("cockroachdb://root@localhost:26257") self.assertEqual(smartdb._SingleEndpoint, type(ss)) self.assertEqual(1, len(ss.engines))
def test_00(self): ss = smartdb.SmartDatabaseClient( "sqlite:////var/lib/enjoliver/enjoliver.sqlite") self.assertEqual(smartdb._SingleEndpoint, type(ss)) self.assertEqual(1, len(ss.engines))
def setUpClass(cls): db_uri = 'sqlite:///:memory:' cls.smart = smartdb.SmartDatabaseClient(db_uri)
def setUpClass(cls): db_uri = "postgresql://postgres@localhost:5432" cls.smart = smartdb.SmartDatabaseClient(db_uri) cls.repositories = RepositoriesRegister(cls.smart) cls.set_up_class_checks(cls.smart)
def setUpClass(cls): db_uri = 'sqlite:///:memory:' cls.smart = smartdb.SmartDatabaseClient(db_uri) cls.repositories = RepositoriesRegister(cls.smart) cls.set_up_class_checks(cls.smart)