Python sqlite3和单元测试

import os

import unittest            # 包含单元测试模块

import sqlite3 as sqlite   # 包含sqlite3模块

def get_db_path():

return "sqlite_testdb"

class TransactionTests(unittest.TestCase):  # 单元测试第一步:　由TestCase派生类

def setUp(self): # 单元测试环境配置

try:

os.remove(get_db_path())

except:

pass

self.con1 = sqlite.connect(get_db_path(), timeout=0.1) # 连接数据库

self.cur1 = self.con1.cursor() # 获取游标

self.con2 = sqlite.connect(get_db_path(), timeout=0.1)

self.cur2 = self.con2.cursor()

def tearDown(self): # 单元测试环境清除

self.cur1.close() # 关闭游标

self.con1.close() # 关闭连接

self.cur2.close()

self.con2.close()

os.unlink(get_db_path())

def CheckDMLdoesAutoCommitBefore(self):

self.cur1.execute("create table test(i)") # 执行SQL查询

self.cur1.execute("insert into test(i) values (5)")

self.cur1.execute("create table test2(j)")

self.cur2.execute("select i from test")

res = self.cur2.fetchall()

self.failUnlessEqual(len(res), 1) # 测试

def CheckInsertStartsTransaction(self):

self.cur1.execute("create table test(i)")

self.cur1.execute("insert into test(i) values (5)")

self.cur2.execute("select i from test")

res = self.cur2.fetchall()

self.failUnlessEqual(len(res), 0)

def CheckUpdateStartsTransaction(self):

self.cur1.execute("create table test(i)")

self.cur1.execute("insert into test(i) values (5)")

self.con1.commit()

self.cur1.execute("update test set i=6")

self.cur2.execute("select i from test")

res = self.cur2.fetchone()[0]

self.failUnlessEqual(res, 5)

def CheckDeleteStartsTransaction(self):

self.cur1.execute("create table test(i)")

self.cur1.execute("insert into test(i) values (5)")

self.con1.commit()

self.cur1.execute("delete from test")

self.cur2.execute("select i from test")

res = self.cur2.fetchall()

self.failUnlessEqual(len(res), 1)

def CheckReplaceStartsTransaction(self):

self.cur1.execute("create table test(i)")

self.cur1.execute("insert into test(i) values (5)")

self.con1.commit()

self.cur1.execute("replace into test(i) values (6)")

self.cur2.execute("select i from test")

res = self.cur2.fetchall()

self.failUnlessEqual(len(res), 1)

self.failUnlessEqual(res[0][0], 5)

def CheckToggleAutoCommit(self):

self.cur1.execute("create table test(i)")

self.cur1.execute("insert into test(i) values (5)")

self.con1.isolation_level = None

self.failUnlessEqual(self.con1.isolation_level, None)

self.cur2.execute("select i from test")

res = self.cur2.fetchall()

self.failUnlessEqual(len(res), 1)

self.con1.isolation_level = "DEFERRED"

self.failUnlessEqual(self.con1.isolation_level , "DEFERRED")

self.cur1.execute("insert into test(i) values (5)")

self.cur2.execute("select i from test")

res = self.cur2.fetchall()

self.failUnlessEqual(len(res), 1)

def CheckRaiseTimeout(self):

self.cur1.execute("create table test(i)")

self.cur1.execute("insert into test(i) values (5)")

try:

self.cur2.execute("insert into test(i) values (5)")

self.fail("should have raised an OperationalError")

except sqlite.OperationalError:

pass

except:

self.fail("should have raised an OperationalError")

class SpecialCommandTests(unittest.TestCase):

def setUp(self):

self.con = sqlite.connect(":memory:")

self.cur = self.con.cursor()

def CheckVacuum(self):

self.cur.execute("create table test(i)")

self.cur.execute("insert into test(i) values (5)")

self.cur.execute("vacuum")

def CheckDropTable(self):

self.cur.execute("create table test(i)")

self.cur.execute("insert into test(i) values (5)")

self.cur.execute("drop table test")

def CheckPragma(self):

self.cur.execute("create table test(i)")

self.cur.execute("insert into test(i) values (5)")

self.cur.execute("pragma count_changes=1")

def tearDown(self):

self.cur.close()

self.con.close()

def suite(): # 单元测试第二步

default_suite = unittest.makeSuite(TransactionTests, "Check")

special_command_suite = unittest.makeSuite(SpecialCommandTests, "Check")

return unittest.TestSuite((default_suite, special_command_suite)) # 用例

def test(): # 单元测试第三步

runner = unittest.TextTestRunner()

runner.run(suite()) # 运行单元测试

if __name__ == "__main__":

test()