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