iframe-proxy | Sunbelt Computer Software

History

81 lines (73 loc) · 2.77 KB

Raw

# Author: Paul Kippes <kippesp@gmail.com>

import unittest

import sqlite3 as sqlite

class DumpTests(unittest.TestCase):

def setUp(self):

self.cx = sqlite.connect(":memory:")

self.cu = self.cx.cursor()

def tearDown(self):

self.cx.close()

def CheckTableDump(self):

expected_sqls = [

"""CREATE TABLE "index"("index" blob);"""

"""INSERT INTO "index" VALUES(X'01');"""

"""CREATE TABLE "quoted""table"("quoted""field" text);"""

"""INSERT INTO "quoted""table" VALUES('quoted''value');"""

"CREATE TABLE t1(id integer primary key, s1 text, " \

"t1_i1 integer not null, i2 integer, unique (s1), " \

"constraint t1_idx1 unique (i2));"

"INSERT INTO \"t1\" VALUES(1,'foo',10,20);"

"INSERT INTO \"t1\" VALUES(2,'foo2',30,30);"

"CREATE TABLE t2(id integer, t2_i1 integer, " \

"t2_i2 integer, primary key (id)," \

"foreign key(t2_i1) references t1(t1_i1));"

"CREATE TRIGGER trigger_1 update of t1_i1 on t1 " \

"begin " \

"update t2 set t2_i1 = new.t1_i1 where t2_i1 = old.t1_i1; " \

"end;"

"CREATE VIEW v1 as select * from t1 left join t2 " \

"using (id);"

]

[self.cu.execute(s) for s in expected_sqls]

i = self.cx.iterdump()

actual_sqls = [s for s in i]

expected_sqls = ['BEGIN TRANSACTION;'] + expected_sqls + \

['COMMIT;']

[self.assertEqual(expected_sqls[i], actual_sqls[i])

for i in range(len(expected_sqls))]

def CheckUnorderableRow(self):

# iterdump() should be able to cope with unorderable row types (issue #15545)

class UnorderableRow:

def __init__(self, cursor, row):

self.row = row

def __getitem__(self, index):

return self.row[index]

self.cx.row_factory = UnorderableRow

CREATE_ALPHA = """CREATE TABLE "alpha" ("one");"""

CREATE_BETA = """CREATE TABLE "beta" ("two");"""

expected = [

"BEGIN TRANSACTION;",

CREATE_ALPHA,

CREATE_BETA,

"COMMIT;"

]

self.cu.execute(CREATE_BETA)

self.cu.execute(CREATE_ALPHA)

got = list(self.cx.iterdump())

self.assertEqual(expected, got)

def suite():

return unittest.TestSuite(unittest.makeSuite(DumpTests, "Check"))

def test():

runner = unittest.TextTestRunner()

runner.run(suite())

if __name__ == "__main__":

test()

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

dump.py

Latest commit

History

dump.py

File metadata and controls