Home | History | Annotate | Download | only in test 
      1 #-*- coding: ISO-8859-1 -*-
      2 # pysqlite2/test/regression.py: pysqlite regression tests
      3 #
      4 # Copyright (C) 2007 Gerhard Hring <gh (at] ghaering.de>
      5 #
      6 # This file is part of pysqlite.
      7 #
      8 # This software is provided 'as-is', without any express or implied
      9 # warranty.  In no event will the authors be held liable for any damages
     10 # arising from the use of this software.
     11 #
     12 # Permission is granted to anyone to use this software for any purpose,
     13 # including commercial applications, and to alter it and redistribute it
     14 # freely, subject to the following restrictions:
     15 #
     16 # 1. The origin of this software must not be misrepresented; you must not
     17 #    claim that you wrote the original software. If you use this software
     18 #    in a product, an acknowledgment in the product documentation would be
     19 #    appreciated but is not required.
     20 # 2. Altered source versions must be plainly marked as such, and must not be
     21 #    misrepresented as being the original software.
     22 # 3. This notice may not be removed or altered from any source distribution.
     23 
     24 from __future__ import with_statement
     25 import unittest
     26 import sqlite3 as sqlite
     27 
     28 did_rollback = False
     29 
     30 class MyConnection(sqlite.Connection):
     31     def rollback(self):
     32         global did_rollback
     33         did_rollback = True
     34         sqlite.Connection.rollback(self)
     35 
     36 class ContextTests(unittest.TestCase):
     37     def setUp(self):
     38         global did_rollback
     39         self.con = sqlite.connect(":memory:", factory=MyConnection)
     40         self.con.execute("create table test(c unique)")
     41         did_rollback = False
     42 
     43     def tearDown(self):
     44         self.con.close()
     45 
     46     def CheckContextManager(self):
     47         """Can the connection be used as a context manager at all?"""
     48         with self.con:
     49             pass
     50 
     51     def CheckContextManagerCommit(self):
     52         """Is a commit called in the context manager?"""
     53         with self.con:
     54             self.con.execute("insert into test(c) values ('foo')")
     55         self.con.rollback()
     56         count = self.con.execute("select count(*) from test").fetchone()[0]
     57         self.assertEqual(count, 1)
     58 
     59     def CheckContextManagerRollback(self):
     60         """Is a rollback called in the context manager?"""
     61         global did_rollback
     62         self.assertEqual(did_rollback, False)
     63         try:
     64             with self.con:
     65                 self.con.execute("insert into test(c) values (4)")
     66                 self.con.execute("insert into test(c) values (4)")
     67         except sqlite.IntegrityError:
     68             pass
     69         self.assertEqual(did_rollback, True)
     70 
     71 def suite():
     72     ctx_suite = unittest.makeSuite(ContextTests, "Check")
     73     return unittest.TestSuite((ctx_suite,))
     74 
     75 def test():
     76     runner = unittest.TextTestRunner()
     77     runner.run(suite())
     78 
     79 if __name__ == "__main__":
     80     test()
     81