@@ -132,6 +132,11 @@ def test_rollback(self):
132132 def test_identifier_case (self ):
133133 self .assertIn ("count" , self .db .execute ("SELECT 1 AS count" )[0 ])
134134
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+
135140 def tearDown (self ):
136141 self .db .execute ("DROP TABLE cs50" )
137142 self .db .execute ("DROP TABLE IF EXISTS foo" )
@@ -166,6 +171,7 @@ def setUp(self):
166171 self .db .execute ("CREATE TABLE IF NOT EXISTS cs50 (id SERIAL PRIMARY KEY, val VARCHAR(16), bin BYTEA)" )
167172 self .db .execute ("DELETE FROM cs50" )
168173
174+
169175 def test_cte (self ):
170176 self .assertEqual (self .db .execute ("WITH foo AS ( SELECT 1 AS bar ) SELECT bar FROM foo" ), [{"bar" : 1 }])
171177
@@ -323,7 +329,7 @@ def test_cte(self):
323329if __name__ == "__main__" :
324330 suite = unittest .TestSuite ([
325331 unittest .TestLoader ().loadTestsFromTestCase (SQLiteTests ),
326- unittest .TestLoader ().loadTestsFromTestCase (MySQLTests ),
332+ # unittest.TestLoader().loadTestsFromTestCase(MySQLTests),
327333 unittest .TestLoader ().loadTestsFromTestCase (PostgresTests )
328334 ])
329335
0 commit comments