Skip to content

Commit cb03140

Browse files
committed
updated tests
1 parent d827119 commit cb03140

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ s = cs50.get_string();
3030
1. In `/etc/profile.d/cli.sh`, remove `valgrind` function for now.
3131
1. Run `service mysql start`.
3232
1. Run `mysql -e 'CREATE DATABASE IF NOT EXISTS test;'`.
33-
1. In `/etc/postgresql/10/main/pg_hba.conf, change:
33+
1. In `/etc/postgresql/12/main/pg_hba.conf, change:
3434
```
3535
local all postgres peer
3636
host all all 127.0.0.1/32 md5

tests/sql.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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):
323329
if __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

Comments
 (0)