Mercurial > cpython-lru_cache-weakref

annotate Lib/sqlite3/test/dump.py @ 83447:678564fc0962 3.3

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
#17789: test_random now works with unittest test discovery. Patch by Zachary Ware.
author Ezio Melotti <ezio.melotti@gmail.com>
date Fri, 19 Apr 2013 05:45:27 +0300
parents 2cdb599172ab
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
46176
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
1 # Author: Paul Kippes <kippesp@gmail.com>
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
2
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
3 import unittest
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
4 import sqlite3 as sqlite
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
5
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
6 class DumpTests(unittest.TestCase):
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
7 def setUp(self):
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
8 self.cx = sqlite.connect(":memory:")
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
9 self.cu = self.cx.cursor()
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
10
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
11 def tearDown(self):
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
12 self.cx.close()
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
13
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
14 def CheckTableDump(self):
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
15 expected_sqls = [
74892
4b105d328fe7 Fix sqlite3.Connection.iterdump on tables/fields with reserved names or quotes
Petri Lehtinen <petri@digip.org>
parents: 46176
diff changeset
16 """CREATE TABLE "index"("index" blob);"""
4b105d328fe7 Fix sqlite3.Connection.iterdump on tables/fields with reserved names or quotes
Petri Lehtinen <petri@digip.org>
parents: 46176
diff changeset
17 ,
4b105d328fe7 Fix sqlite3.Connection.iterdump on tables/fields with reserved names or quotes
Petri Lehtinen <petri@digip.org>
parents: 46176
diff changeset
18 """INSERT INTO "index" VALUES(X'01');"""
4b105d328fe7 Fix sqlite3.Connection.iterdump on tables/fields with reserved names or quotes
Petri Lehtinen <petri@digip.org>
parents: 46176
diff changeset
19 ,
4b105d328fe7 Fix sqlite3.Connection.iterdump on tables/fields with reserved names or quotes
Petri Lehtinen <petri@digip.org>
parents: 46176
diff changeset
20 """CREATE TABLE "quoted""table"("quoted""field" text);"""
4b105d328fe7 Fix sqlite3.Connection.iterdump on tables/fields with reserved names or quotes
Petri Lehtinen <petri@digip.org>
parents: 46176
diff changeset
21 ,
4b105d328fe7 Fix sqlite3.Connection.iterdump on tables/fields with reserved names or quotes
Petri Lehtinen <petri@digip.org>
parents: 46176
diff changeset
22 """INSERT INTO "quoted""table" VALUES('quoted''value');"""
4b105d328fe7 Fix sqlite3.Connection.iterdump on tables/fields with reserved names or quotes
Petri Lehtinen <petri@digip.org>
parents: 46176
diff changeset
23 ,
46176
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
24 "CREATE TABLE t1(id integer primary key, s1 text, " \
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
25 "t1_i1 integer not null, i2 integer, unique (s1), " \
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
26 "constraint t1_idx1 unique (i2));"
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
27 ,
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
28 "INSERT INTO \"t1\" VALUES(1,'foo',10,20);"
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
29 ,
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
30 "INSERT INTO \"t1\" VALUES(2,'foo2',30,30);"
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
31 ,
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
32 "CREATE TABLE t2(id integer, t2_i1 integer, " \
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
33 "t2_i2 integer, primary key (id)," \
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
34 "foreign key(t2_i1) references t1(t1_i1));"
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
35 ,
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
36 "CREATE TRIGGER trigger_1 update of t1_i1 on t1 " \
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
37 "begin " \
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
38 "update t2 set t2_i1 = new.t1_i1 where t2_i1 = old.t1_i1; " \
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
39 "end;"
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
40 ,
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
41 "CREATE VIEW v1 as select * from t1 left join t2 " \
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
42 "using (id);"
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
43 ]
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
44 [self.cu.execute(s) for s in expected_sqls]
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
45 i = self.cx.iterdump()
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
46 actual_sqls = [s for s in i]
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
47 expected_sqls = ['BEGIN TRANSACTION;'] + expected_sqls + \
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
48 ['COMMIT;']
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
49 [self.assertEqual(expected_sqls[i], actual_sqls[i])
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
50 for i in range(len(expected_sqls))]
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
51
81367
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
52 def CheckUnorderableRow(self):
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
53 # iterdump() should be able to cope with unorderable row types (issue #15545)
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
54 class UnorderableRow:
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
55 def __init__(self, cursor, row):
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
56 self.row = row
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
57 def __getitem__(self, index):
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
58 return self.row[index]
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
59 self.cx.row_factory = UnorderableRow
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
60 CREATE_ALPHA = """CREATE TABLE "alpha" ("one");"""
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
61 CREATE_BETA = """CREATE TABLE "beta" ("two");"""
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
62 expected = [
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
63 "BEGIN TRANSACTION;",
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
64 CREATE_ALPHA,
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
65 CREATE_BETA,
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
66 "COMMIT;"
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
67 ]
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
68 self.cu.execute(CREATE_BETA)
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
69 self.cu.execute(CREATE_ALPHA)
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
70 got = list(self.cx.iterdump())
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
71 self.assertEqual(expected, got)
2cdb599172ab #15545: fix sqlite3.iterdump regression on unsortable row_factory objects.
R David Murray <rdmurray@bitdance.com>
parents: 74892
diff changeset
72
46176
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
73 def suite():
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
74 return unittest.TestSuite(unittest.makeSuite(DumpTests, "Check"))
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
75
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
76 def test():
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
77 runner = unittest.TextTestRunner()
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
78 runner.run(suite())
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
79
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
80 if __name__ == "__main__":
872abed7d069 Added missing files for new iterdump method.
Gerhard Häring <gh@ghaering.de>
parents:
diff changeset
81 test()