@@ -132,6 +132,11 @@ def test_rollback(self):
132
132
def test_identifier_case (self ):
133
133
self .assertIn ("count" , self .db .execute ("SELECT 1 AS count" )[0 ])
134
134
135
+ def test_lastrowid (self ):
136
+ self .db .execute ("CREATE TABLE foo(id SERIAL PRIMARY KEY, firstname TEXT, lastname TEXT)" )
137
+ self .assertEqual (self .db .execute ("INSERT INTO foo (firstname, lastname) VALUES('firstname', 'lastname')" ), 1 )
138
+ self .assertRaises (ValueError , self .db .execute , "INSERT INTO foo (id, firstname, lastname) VALUES(1, 'firstname', 'lastname')" )
139
+
135
140
def tearDown (self ):
136
141
self .db .execute ("DROP TABLE cs50" )
137
142
self .db .execute ("DROP TABLE IF EXISTS foo" )
@@ -166,6 +171,7 @@ def setUp(self):
166
171
self .db .execute ("CREATE TABLE IF NOT EXISTS cs50 (id SERIAL PRIMARY KEY, val VARCHAR(16), bin BYTEA)" )
167
172
self .db .execute ("DELETE FROM cs50" )
168
173
174
+
169
175
def test_cte (self ):
170
176
self .assertEqual (self .db .execute ("WITH foo AS ( SELECT 1 AS bar ) SELECT bar FROM foo" ), [{"bar" : 1 }])
171
177
@@ -323,7 +329,7 @@ def test_cte(self):
323
329
if __name__ == "__main__" :
324
330
suite = unittest .TestSuite ([
325
331
unittest .TestLoader ().loadTestsFromTestCase (SQLiteTests ),
326
- unittest .TestLoader ().loadTestsFromTestCase (MySQLTests ),
332
+ # unittest.TestLoader().loadTestsFromTestCase(MySQLTests),
327
333
unittest .TestLoader ().loadTestsFromTestCase (PostgresTests )
328
334
])
329
335
0 commit comments