class TestDatabase_:
def setUp(self):
self.database = DatabaseType()
def test_alias(self):
assert database is Database
def __shouldnt_import(self, name):
try:
__import__(name)
raise Exception
except ImportError:
pass
def test_init(self):
assert self.database.conn is None
assert self.database.configs == {}
assert self.database.dbapi.module.__name__ in DBAPI_MAPPINGS
# test load orders
name = self.database.dbapi.module.__name__
if name == 'MySQLdb':
assert __import__('MySQLdb')
elif name == 'pymysql':
assert __import__('pymysql')
self.__shouldnt_import('MySQLdb')
elif name == 'sqlite3':
assert __import__('sqlite3')
self.__shouldnt_import('MySQLdb')
self.__shouldnt_import('pymysql')
def test_set_dbapi(self):
self.database.set_dbapi(dbapi)
assert self.database.dbapi.module.__name__ == dbapi_name
class TestDatabase_:
def setUp(self):
self.database = DatabaseType()
def test_alias(self):
assert database is Database
def __shouldnt_import(self, name):
try:
__import__(name)
raise Exception
except ImportError:
pass
def test_init(self):
assert self.database.conn is None
assert self.database.configs == {}
assert self.database.dbapi.module.__name__ in DBAPI_MAPPINGS
# test load orders
name = self.database.dbapi.module.__name__
if name == 'MySQLdb':
assert __import__('MySQLdb')
elif name == 'pymysql':
assert __import__('pymysql')
self.__shouldnt_import('MySQLdb')
elif name == 'sqlite3':
assert __import__('sqlite3')
self.__shouldnt_import('MySQLdb')
self.__shouldnt_import('pymysql')
def test_set_dbapi(self):
self.database.set_dbapi(dbapi)
assert self.database.dbapi.module.__name__ == dbapi_name
def setUp(self):
self.database = DatabaseType()
def setUp(self):
self.database = DatabaseType()
self.database.set_dbapi(dbapi)
super(TestDatabase, self).setUp()
class TestDatabase(Test):
def setUp(self):
self.database = DatabaseType()
self.database.set_dbapi(dbapi)
super(TestDatabase, self).setUp()
def test_config(self):
assert self.database.configs == {}
self.database.config(**configs)
assert self.database.configs
assert not self.database.conn
self.database.connect()
assert self.database.conn
self.database.config(**configs)
assert not self.database.dbapi.conn_is_open(self.database.conn)
def test_connect(self):
assert self.database.conn is None
self.database.config(**configs)
conn1 = self.database.connect()
conn2 = self.database.connect()
assert conn1 is not conn2
assert self.database.dbapi.conn_is_open(conn1)
assert self.database.dbapi.conn_is_open(conn2)
def test_get_conn(self):
assert self.database.conn is None
self.database.config(**configs)
conn = self.database.get_conn()
assert conn and self.database.dbapi.conn_is_open(conn)
assert conn and self.database.dbapi.conn_is_alive(conn)
conn1 = self.database.get_conn()
assert conn1 is conn
def test___del__(self):
self.database.config(**configs)
self.database.connect()
conn = self.database.conn
dbapi = self.database.dbapi
del self.database
conn = None
assert not dbapi.conn_is_open(conn)
def test_execute(self):
self.database.config(**configs)
cursor = self.database.execute(
"insert into t_user (name, email) values ('jack', '*****@*****.**')")
assert cursor.lastrowid == 1
def test_execute_sql(self):
pass
def change(self):
self.database.configs(**configs)
self.database.connect()
old_conn = self.database.conn
self.database.execute('create database skylarktests2')
self.database.change('skylarktests2')
assert self.database.configs != configs
if db_type == 'mysql':
self.database.conn is old_conn
else:
self.database.conn is not old_conn
assert not self.database.dbapi.conn_is_open(old_conn)
def test_transaction(self):
db = self.database
db.config(**configs)
db.execute("insert into t_user (name, email) values ('j', '*****@*****.**')")
db.set_autocommit(False)
t = db.transaction()
try:
db.execute("insert into t_user set x;") # syntax error
except Exception:
t.rollback()
else:
raise Exception
cursor = db.execute('select count(*) from t_user;')
db.conn.commit() # !important
assert cursor.fetchone()[0] == 1
with db.transaction() as t:
db.execute(
"insert into t_user (name, email) values ('a', '*****@*****.**')")
db.execute(
"insert into t_user (name, email) values ('a', '*****@*****.**')")
cursor = db.execute('select count(*) from t_user')
db.commit()
assert cursor.fetchone()[0] == 3
def setUp(self):
self.database = DatabaseType()
def setUp(self):
self.database = DatabaseType()
self.database.set_dbapi(dbapi)
super(TestDatabase, self).setUp()
class TestDatabase(Test):
def setUp(self):
self.database = DatabaseType()
self.database.set_dbapi(dbapi)
super(TestDatabase, self).setUp()
def test_config(self):
assert self.database.configs == {}
self.database.config(**configs)
assert self.database.configs
assert not self.database.conn
self.database.connect()
assert self.database.conn
self.database.config(**configs)
assert not self.database.dbapi.conn_is_open(self.database.conn)
def test_connect(self):
assert self.database.conn is None
self.database.config(**configs)
conn1 = self.database.connect()
conn2 = self.database.connect()
assert conn1 is not conn2
assert self.database.dbapi.conn_is_open(conn1)
assert self.database.dbapi.conn_is_open(conn2)
def test_get_conn(self):
assert self.database.conn is None
self.database.config(**configs)
conn = self.database.get_conn()
assert conn and self.database.dbapi.conn_is_open(conn)
assert conn and self.database.dbapi.conn_is_alive(conn)
conn1 = self.database.get_conn()
assert conn1 is conn
def test___del__(self):
self.database.config(**configs)
self.database.connect()
conn = self.database.conn
dbapi = self.database.dbapi
del self.database
conn = None
assert not dbapi.conn_is_open(conn)
def test_execute(self):
self.database.config(**configs)
cursor = self.database.execute(
"insert into t_user (name, email) values ('jack', '*****@*****.**')")
assert cursor.lastrowid == 1
def test_execute_sql(self):
pass
def change(self):
self.database.configs(**configs)
self.database.connect()
old_conn = self.database.conn
self.database.execute('create database skylarktests2')
self.database.change('skylarktests2')
assert self.database.configs != configs
if db_type == 'mysql':
self.database.conn is old_conn
else:
self.database.conn is not old_conn
assert not self.database.dbapi.conn_is_open(old_conn)
def test_transaction(self):
db = self.database
db.config(**configs)
db.execute("insert into t_user (name, email) values ('j', '*****@*****.**')")
db.set_autocommit(False)
t = db.transaction()
try:
db.execute("insert into t_user set x;") # syntax error
except Exception:
t.rollback()
else:
raise Exception
cursor = db.execute('select count(*) from t_user;')
db.conn.commit() # !important
assert cursor.fetchone()[0] == 1
with db.transaction() as t:
db.execute(
"insert into t_user (name, email) values ('a', '*****@*****.**')")
db.execute(
"insert into t_user (name, email) values ('a', '*****@*****.**')")
cursor = db.execute('select count(*) from t_user')
db.commit()
assert cursor.fetchone()[0] == 3
