summaryrefslogtreecommitdiff
path: root/lib/python2.7/sqlite3/test/factory.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/python2.7/sqlite3/test/factory.py')
-rw-r--r--lib/python2.7/sqlite3/test/factory.py244
1 files changed, 0 insertions, 244 deletions
diff --git a/lib/python2.7/sqlite3/test/factory.py b/lib/python2.7/sqlite3/test/factory.py
deleted file mode 100644
index 52854be..0000000
--- a/lib/python2.7/sqlite3/test/factory.py
+++ /dev/null
@@ -1,244 +0,0 @@
-#-*- coding: ISO-8859-1 -*-
-# pysqlite2/test/factory.py: tests for the various factories in pysqlite
-#
-# Copyright (C) 2005-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 unittest
-import sqlite3 as sqlite
-
-class MyConnection(sqlite.Connection):
- def __init__(self, *args, **kwargs):
- sqlite.Connection.__init__(self, *args, **kwargs)
-
-def dict_factory(cursor, row):
- d = {}
- for idx, col in enumerate(cursor.description):
- d[col[0]] = row[idx]
- return d
-
-class MyCursor(sqlite.Cursor):
- def __init__(self, *args, **kwargs):
- sqlite.Cursor.__init__(self, *args, **kwargs)
- self.row_factory = dict_factory
-
-class ConnectionFactoryTests(unittest.TestCase):
- def setUp(self):
- self.con = sqlite.connect(":memory:", factory=MyConnection)
-
- def tearDown(self):
- self.con.close()
-
- def CheckIsInstance(self):
- self.assertTrue(isinstance(self.con,
- MyConnection),
- "connection is not instance of MyConnection")
-
-class CursorFactoryTests(unittest.TestCase):
- def setUp(self):
- self.con = sqlite.connect(":memory:")
-
- def tearDown(self):
- self.con.close()
-
- def CheckIsInstance(self):
- cur = self.con.cursor(factory=MyCursor)
- self.assertTrue(isinstance(cur,
- MyCursor),
- "cursor is not instance of MyCursor")
-
-class RowFactoryTestsBackwardsCompat(unittest.TestCase):
- def setUp(self):
- self.con = sqlite.connect(":memory:")
-
- def CheckIsProducedByFactory(self):
- cur = self.con.cursor(factory=MyCursor)
- cur.execute("select 4+5 as foo")
- row = cur.fetchone()
- self.assertTrue(isinstance(row,
- dict),
- "row is not instance of dict")
- cur.close()
-
- def tearDown(self):
- self.con.close()
-
-class RowFactoryTests(unittest.TestCase):
- def setUp(self):
- self.con = sqlite.connect(":memory:")
-
- def CheckCustomFactory(self):
- self.con.row_factory = lambda cur, row: list(row)
- row = self.con.execute("select 1, 2").fetchone()
- self.assertTrue(isinstance(row,
- list),
- "row is not instance of list")
-
- def CheckSqliteRowIndex(self):
- self.con.row_factory = sqlite.Row
- row = self.con.execute("select 1 as a, 2 as b").fetchone()
- self.assertTrue(isinstance(row,
- sqlite.Row),
- "row is not instance of sqlite.Row")
-
- col1, col2 = row["a"], row["b"]
- self.assertTrue(col1 == 1, "by name: wrong result for column 'a'")
- self.assertTrue(col2 == 2, "by name: wrong result for column 'a'")
-
- col1, col2 = row["A"], row["B"]
- self.assertTrue(col1 == 1, "by name: wrong result for column 'A'")
- self.assertTrue(col2 == 2, "by name: wrong result for column 'B'")
-
- col1, col2 = row[0], row[1]
- self.assertTrue(col1 == 1, "by index: wrong result for column 0")
- self.assertTrue(col2 == 2, "by index: wrong result for column 1")
-
- def CheckSqliteRowIter(self):
- """Checks if the row object is iterable"""
- self.con.row_factory = sqlite.Row
- row = self.con.execute("select 1 as a, 2 as b").fetchone()
- for col in row:
- pass
-
- def CheckSqliteRowAsTuple(self):
- """Checks if the row object can be converted to a tuple"""
- self.con.row_factory = sqlite.Row
- row = self.con.execute("select 1 as a, 2 as b").fetchone()
- t = tuple(row)
-
- def CheckSqliteRowAsDict(self):
- """Checks if the row object can be correctly converted to a dictionary"""
- self.con.row_factory = sqlite.Row
- row = self.con.execute("select 1 as a, 2 as b").fetchone()
- d = dict(row)
- self.assertEqual(d["a"], row["a"])
- self.assertEqual(d["b"], row["b"])
-
- def CheckSqliteRowHashCmp(self):
- """Checks if the row object compares and hashes correctly"""
- self.con.row_factory = sqlite.Row
- row_1 = self.con.execute("select 1 as a, 2 as b").fetchone()
- row_2 = self.con.execute("select 1 as a, 2 as b").fetchone()
- row_3 = self.con.execute("select 1 as a, 3 as b").fetchone()
-
- self.assertTrue(row_1 == row_1)
- self.assertTrue(row_1 == row_2)
- self.assertTrue(row_2 != row_3)
-
- self.assertFalse(row_1 != row_1)
- self.assertFalse(row_1 != row_2)
- self.assertFalse(row_2 == row_3)
-
- self.assertEqual(row_1, row_2)
- self.assertEqual(hash(row_1), hash(row_2))
- self.assertNotEqual(row_1, row_3)
- self.assertNotEqual(hash(row_1), hash(row_3))
-
- def tearDown(self):
- self.con.close()
-
-class TextFactoryTests(unittest.TestCase):
- def setUp(self):
- self.con = sqlite.connect(":memory:")
-
- def CheckUnicode(self):
- austria = unicode("Österreich", "latin1")
- row = self.con.execute("select ?", (austria,)).fetchone()
- self.assertTrue(type(row[0]) == unicode, "type of row[0] must be unicode")
-
- def CheckString(self):
- self.con.text_factory = str
- austria = unicode("Österreich", "latin1")
- row = self.con.execute("select ?", (austria,)).fetchone()
- self.assertTrue(type(row[0]) == str, "type of row[0] must be str")
- self.assertTrue(row[0] == austria.encode("utf-8"), "column must equal original data in UTF-8")
-
- def CheckCustom(self):
- self.con.text_factory = lambda x: unicode(x, "utf-8", "ignore")
- austria = unicode("Österreich", "latin1")
- row = self.con.execute("select ?", (austria.encode("latin1"),)).fetchone()
- self.assertTrue(type(row[0]) == unicode, "type of row[0] must be unicode")
- self.assertTrue(row[0].endswith(u"reich"), "column must contain original data")
-
- def CheckOptimizedUnicode(self):
- self.con.text_factory = sqlite.OptimizedUnicode
- austria = unicode("Österreich", "latin1")
- germany = unicode("Deutchland")
- a_row = self.con.execute("select ?", (austria,)).fetchone()
- d_row = self.con.execute("select ?", (germany,)).fetchone()
- self.assertTrue(type(a_row[0]) == unicode, "type of non-ASCII row must be unicode")
- self.assertTrue(type(d_row[0]) == str, "type of ASCII-only row must be str")
-
- def tearDown(self):
- self.con.close()
-
-class TextFactoryTestsWithEmbeddedZeroBytes(unittest.TestCase):
- def setUp(self):
- self.con = sqlite.connect(":memory:")
- self.con.execute("create table test (value text)")
- self.con.execute("insert into test (value) values (?)", ("a\x00b",))
-
- def CheckString(self):
- # text_factory defaults to unicode
- row = self.con.execute("select value from test").fetchone()
- self.assertIs(type(row[0]), unicode)
- self.assertEqual(row[0], "a\x00b")
-
- def CheckCustom(self):
- # A custom factory should receive an str argument
- self.con.text_factory = lambda x: x
- row = self.con.execute("select value from test").fetchone()
- self.assertIs(type(row[0]), str)
- self.assertEqual(row[0], "a\x00b")
-
- def CheckOptimizedUnicodeAsString(self):
- # ASCII -> str argument
- self.con.text_factory = sqlite.OptimizedUnicode
- row = self.con.execute("select value from test").fetchone()
- self.assertIs(type(row[0]), str)
- self.assertEqual(row[0], "a\x00b")
-
- def CheckOptimizedUnicodeAsUnicode(self):
- # Non-ASCII -> unicode argument
- self.con.text_factory = sqlite.OptimizedUnicode
- self.con.execute("delete from test")
- self.con.execute("insert into test (value) values (?)", (u'ä\0ö',))
- row = self.con.execute("select value from test").fetchone()
- self.assertIs(type(row[0]), unicode)
- self.assertEqual(row[0], u"ä\x00ö")
-
- def tearDown(self):
- self.con.close()
-
-def suite():
- connection_suite = unittest.makeSuite(ConnectionFactoryTests, "Check")
- cursor_suite = unittest.makeSuite(CursorFactoryTests, "Check")
- row_suite_compat = unittest.makeSuite(RowFactoryTestsBackwardsCompat, "Check")
- row_suite = unittest.makeSuite(RowFactoryTests, "Check")
- text_suite = unittest.makeSuite(TextFactoryTests, "Check")
- text_zero_bytes_suite = unittest.makeSuite(TextFactoryTestsWithEmbeddedZeroBytes, "Check")
- return unittest.TestSuite((connection_suite, cursor_suite, row_suite_compat, row_suite, text_suite, text_zero_bytes_suite))
-
-def test():
- runner = unittest.TextTestRunner()
- runner.run(suite())
-
-if __name__ == "__main__":
- test()
generated by cgit v1.2.3 (git 2.25.1) at 2024-06-06 18:38:24 +0000