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
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
