/Lib/sqlite3/test/regression.py
http://unladen-swallow.googlecode.com/ · Python · 180 lines · 110 code · 16 blank · 54 comment · 12 complexity · 0a38ea77d9d2425a209ba6ebc2863f25 MD5 · raw file
- #-*- coding: ISO-8859-1 -*-
- # pysqlite2/test/regression.py: pysqlite regression tests
- #
- # Copyright (C) 2006-2007 Gerhard Häring <gh@ghaering.de>
- #
- # This file is part of pysqlite.
- #
- # This software is provided 'as-is', without any express or implied
- # warranty. In no event will the authors be held liable for any damages
- # arising from the use of this software.
- #
- # Permission is granted to anyone to use this software for any purpose,
- # including commercial applications, and to alter it and redistribute it
- # freely, subject to the following restrictions:
- #
- # 1. The origin of this software must not be misrepresented; you must not
- # claim that you wrote the original software. If you use this software
- # in a product, an acknowledgment in the product documentation would be
- # appreciated but is not required.
- # 2. Altered source versions must be plainly marked as such, and must not be
- # misrepresented as being the original software.
- # 3. This notice may not be removed or altered from any source distribution.
- import datetime
- import unittest
- import sqlite3 as sqlite
- class RegressionTests(unittest.TestCase):
- def setUp(self):
- self.con = sqlite.connect(":memory:")
- def tearDown(self):
- self.con.close()
- def CheckPragmaUserVersion(self):
- # This used to crash pysqlite because this pragma command returns NULL for the column name
- cur = self.con.cursor()
- cur.execute("pragma user_version")
- def CheckPragmaSchemaVersion(self):
- # This still crashed pysqlite <= 2.2.1
- con = sqlite.connect(":memory:", detect_types=sqlite.PARSE_COLNAMES)
- try:
- cur = self.con.cursor()
- cur.execute("pragma schema_version")
- finally:
- cur.close()
- con.close()
- def CheckStatementReset(self):
- # pysqlite 2.1.0 to 2.2.0 have the problem that not all statements are
- # reset before a rollback, but only those that are still in the
- # statement cache. The others are not accessible from the connection object.
- con = sqlite.connect(":memory:", cached_statements=5)
- cursors = [con.cursor() for x in xrange(5)]
- cursors[0].execute("create table test(x)")
- for i in range(10):
- cursors[0].executemany("insert into test(x) values (?)", [(x,) for x in xrange(10)])
- for i in range(5):
- cursors[i].execute(" " * i + "select x from test")
- con.rollback()
- def CheckColumnNameWithSpaces(self):
- cur = self.con.cursor()
- cur.execute('select 1 as "foo bar [datetime]"')
- self.failUnlessEqual(cur.description[0][0], "foo bar")
- cur.execute('select 1 as "foo baz"')
- self.failUnlessEqual(cur.description[0][0], "foo baz")
- def CheckStatementAvailable(self):
- # pysqlite up to 2.3.2 crashed on this, because the active statement handle was not checked
- # before trying to fetch data from it. close() destroys the active statement ...
- con = sqlite.connect(":memory:", detect_types=sqlite.PARSE_DECLTYPES)
- cur = con.cursor()
- cur.execute("select 4 union select 5")
- cur.close()
- cur.fetchone()
- cur.fetchone()
- def CheckStatementFinalizationOnCloseDb(self):
- # pysqlite versions <= 2.3.3 only finalized statements in the statement
- # cache when closing the database. statements that were still
- # referenced in cursors weren't closed an could provoke "
- # "OperationalError: Unable to close due to unfinalised statements".
- con = sqlite.connect(":memory:")
- cursors = []
- # default statement cache size is 100
- for i in range(105):
- cur = con.cursor()
- cursors.append(cur)
- cur.execute("select 1 x union select " + str(i))
- con.close()
- def CheckOnConflictRollback(self):
- if sqlite.sqlite_version_info < (3, 2, 2):
- return
- con = sqlite.connect(":memory:")
- con.execute("create table foo(x, unique(x) on conflict rollback)")
- con.execute("insert into foo(x) values (1)")
- try:
- con.execute("insert into foo(x) values (1)")
- except sqlite.DatabaseError:
- pass
- con.execute("insert into foo(x) values (2)")
- try:
- con.commit()
- except sqlite.OperationalError:
- self.fail("pysqlite knew nothing about the implicit ROLLBACK")
- def CheckWorkaroundForBuggySqliteTransferBindings(self):
- """
- pysqlite would crash with older SQLite versions unless
- a workaround is implemented.
- """
- self.con.execute("create table foo(bar)")
- self.con.execute("drop table foo")
- self.con.execute("create table foo(bar)")
- def CheckEmptyStatement(self):
- """
- pysqlite used to segfault with SQLite versions 3.5.x. These return NULL
- for "no-operation" statements
- """
- self.con.execute("")
- def CheckUnicodeConnect(self):
- """
- With pysqlite 2.4.0 you needed to use a string or a APSW connection
- object for opening database connections.
- Formerly, both bytestrings and unicode strings used to work.
- Let's make sure unicode strings work in the future.
- """
- con = sqlite.connect(u":memory:")
- con.close()
- def CheckTypeMapUsage(self):
- """
- pysqlite until 2.4.1 did not rebuild the row_cast_map when recompiling
- a statement. This test exhibits the problem.
- """
- SELECT = "select * from foo"
- con = sqlite.connect(":memory:",detect_types=sqlite.PARSE_DECLTYPES)
- con.execute("create table foo(bar timestamp)")
- con.execute("insert into foo(bar) values (?)", (datetime.datetime.now(),))
- con.execute(SELECT)
- con.execute("drop table foo")
- con.execute("create table foo(bar integer)")
- con.execute("insert into foo(bar) values (5)")
- con.execute(SELECT)
- def CheckRegisterAdapter(self):
- """
- See issue 3312.
- """
- self.assertRaises(TypeError, sqlite.register_adapter, {}, None)
- def CheckSetIsolationLevel(self):
- """
- See issue 3312.
- """
- con = sqlite.connect(":memory:")
- self.assertRaises(UnicodeEncodeError, setattr, con,
- "isolation_level", u"\xe9")
- def suite():
- regression_suite = unittest.makeSuite(RegressionTests, "Check")
- return unittest.TestSuite((regression_suite,))
- def test():
- runner = unittest.TextTestRunner()
- runner.run(suite())
- if __name__ == "__main__":
- test()