Move common tests to SQLTests

Author: Kareem Zidane

tests/sql.py

@@ -27,6 +27,7 @@ def test_delete_returns_affected_rows(self):
     def test_insert_returns_last_row_id(self):
         self.assertEqual(self.db.execute("INSERT INTO cs50(val) VALUES('foo')"), 1)
         self.assertEqual(self.db.execute("INSERT INTO cs50(val) VALUES('bar')"), 2)
+        self.assertEqual(self.db.execute("INSERT INTO cs50(val) VALUES('qux')"), 3)
 
     def test_select_all(self):
         self.assertEqual(self.db.execute("SELECT * FROM cs50"), [])
@@ -131,55 +132,13 @@ def test_rollback(self):
     def test_identifier_case(self):
         self.assertIn("count", self.db.execute("SELECT 1 AS count")[0])
 
-    def tearDown(self):
-        self.db.execute("DROP TABLE IF EXISTS cs50")
-        self.db.execute("DROP TABLE IF EXISTS foo")
-        self.db.execute("DROP TABLE IF EXISTS bar")
-
-class MySQLTests(SQLTests):
-    @classmethod
-    def setUpClass(self):
-        self.db = SQL("mysql://root@127.0.0.1/test")
-
-    def setUp(self):
-        self.db.execute("CREATE TABLE IF NOT EXISTS cs50 (id INTEGER NOT NULL AUTO_INCREMENT, val VARCHAR(16), bin BLOB, PRIMARY KEY (id))")
-        self.db.execute("DELETE FROM cs50")
-
-class PostgresTests(SQLTests):
-    @classmethod
-    def setUpClass(self):
-        self.db = SQL("postgresql://postgres:postgres@127.0.0.1/test")
-
-    def setUp(self):
-        self.db.execute("CREATE TABLE IF NOT EXISTS cs50 (id SERIAL PRIMARY KEY, val VARCHAR(16), bin BYTEA)")
-        self.db.execute("DELETE FROM cs50")
-
-    def test_cte(self):
-        self.assertEqual(self.db.execute("WITH foo AS ( SELECT 1 AS bar ) SELECT bar FROM foo"), [{"bar": 1}])
-
-    def test_postgres_scheme(self):
-        db = SQL("postgres://postgres:postgres@127.0.0.1/test")
-        db.execute("SELECT 1")
-
-class SQLiteTests(SQLTests):
-    @classmethod
-    def setUpClass(self):
-        open("test.db", "w").close()
-        self.db = SQL("sqlite:///test.db")
-
-    def setUp(self):
-        self.db.execute("CREATE TABLE IF NOT EXISTS cs50 (id INTEGER PRIMARY KEY, val TEXT, bin BLOB)")
-        self.db.execute("DELETE FROM cs50")
-
-    def test_lastrowid(self):
-        self.db.execute("CREATE TABLE foo(id INTEGER PRIMARY KEY AUTOINCREMENT, firstname TEXT, lastname TEXT)")
-        self.assertEqual(self.db.execute("INSERT INTO foo (firstname, lastname) VALUES('firstname', 'lastname')"), 1)
-        self.assertRaises(ValueError, self.db.execute, "INSERT INTO foo (id, firstname, lastname) VALUES(1, 'firstname', 'lastname')")
-        self.assertEqual(self.db.execute("INSERT OR IGNORE INTO foo (id, firstname, lastname) VALUES(1, 'firstname', 'lastname')"), None)
+    def test_none(self):
+        self.db.execute("CREATE TABLE foo (val INTEGER)")
+        self.db.execute("SELECT * FROM foo WHERE val = ?", None)
 
     def test_integrity_constraints(self):
         self.db.execute("CREATE TABLE foo(id INTEGER PRIMARY KEY)")
-        self.assertEqual(self.db.execute("INSERT INTO foo VALUES(1)"), 1)
+        self.db.execute("INSERT INTO foo VALUES(1)")
         self.assertRaises(ValueError, self.db.execute, "INSERT INTO foo VALUES(1)")
 
     def test_foreign_key_support(self):
@@ -188,7 +147,7 @@ def test_foreign_key_support(self):
         self.assertRaises(ValueError, self.db.execute, "INSERT INTO bar VALUES(50)")
 
     def test_qmark(self):
-        self.db.execute("CREATE TABLE foo (firstname STRING, lastname STRING)")
+        self.db.execute("CREATE TABLE foo (firstname VARCHAR(255), lastname VARCHAR(255))")
 
         self.db.execute("INSERT INTO foo VALUES (?, 'bar')", "baz")
         self.assertEqual(self.db.execute("SELECT * FROM foo"), [{"firstname": "baz", "lastname": "bar"}])
@@ -218,7 +177,7 @@ def test_qmark(self):
         self.assertEqual(self.db.execute("SELECT * FROM foo"), [{"firstname": "bar", "lastname": "baz"}])
         self.db.execute("DELETE FROM foo")
 
-        self.db.execute("CREATE TABLE bar (firstname STRING)")
+        self.db.execute("CREATE TABLE bar (firstname VARCHAR(255))")
 
         self.db.execute("INSERT INTO bar VALUES (?)", "baz")
         self.assertEqual(self.db.execute("SELECT * FROM bar"), [{"firstname": "baz"}])
@@ -242,7 +201,7 @@ def test_qmark(self):
         self.assertRaises(RuntimeError, self.db.execute, "INSERT INTO foo VALUES (?, ?)", 'bar', baz='baz')
 
     def test_named(self):
-        self.db.execute("CREATE TABLE foo (firstname STRING, lastname STRING)")
+        self.db.execute("CREATE TABLE foo (firstname VARCHAR(255), lastname VARCHAR(255))")
 
         self.db.execute("INSERT INTO foo VALUES (:baz, 'bar')", baz="baz")
         self.assertEqual(self.db.execute("SELECT * FROM foo"), [{"firstname": "baz", "lastname": "bar"}])
@@ -264,7 +223,7 @@ def test_named(self):
         self.assertEqual(self.db.execute("SELECT * FROM foo"), [{"firstname": "bar", "lastname": "baz"}])
         self.db.execute("DELETE FROM foo")
 
-        self.db.execute("CREATE TABLE bar (firstname STRING)")
+        self.db.execute("CREATE TABLE bar (firstname VARCHAR(255))")
         self.db.execute("INSERT INTO bar VALUES (:baz)", baz="baz")
         self.assertEqual(self.db.execute("SELECT * FROM bar"), [{"firstname": "baz"}])
 
@@ -274,7 +233,7 @@ def test_named(self):
         self.assertRaises(RuntimeError, self.db.execute, "INSERT INTO foo VALUES (:bar, :baz)", 'baz', bar='bar')
 
     def test_numeric(self):
-        self.db.execute("CREATE TABLE foo (firstname STRING, lastname STRING)")
+        self.db.execute("CREATE TABLE foo (firstname VARCHAR(255), lastname VARCHAR(255))")
 
         self.db.execute("INSERT INTO foo VALUES (:1, 'bar')", "baz")
         self.assertEqual(self.db.execute("SELECT * FROM foo"), [{"firstname": "baz", "lastname": "bar"}])
@@ -296,7 +255,7 @@ def test_numeric(self):
         self.assertEqual(self.db.execute("SELECT * FROM foo"), [{"firstname": "bar", "lastname": "baz"}])
         self.db.execute("DELETE FROM foo")
 
-        self.db.execute("CREATE TABLE bar (firstname STRING)")
+        self.db.execute("CREATE TABLE bar (firstname VARCHAR(255))")
         self.db.execute("INSERT INTO bar VALUES (:1)", "baz")
         self.assertEqual(self.db.execute("SELECT * FROM bar"), [{"firstname": "baz"}])
 
@@ -308,9 +267,51 @@ def test_numeric(self):
     def test_cte(self):
         self.assertEqual(self.db.execute("WITH foo AS ( SELECT 1 AS bar ) SELECT bar FROM foo"), [{"bar": 1}])
 
-    def test_none(self):
-        self.db.execute("CREATE TABLE foo (val INTEGER)")
-        self.db.execute("SELECT * FROM foo WHERE val = ?", None)
+    def tearDown(self):
+        self.db.execute("DROP TABLE IF EXISTS cs50")
+        self.db.execute("DROP TABLE IF EXISTS bar")
+        self.db.execute("DROP TABLE IF EXISTS foo")
+
+class MySQLTests(SQLTests):
+    @classmethod
+    def setUpClass(self):
+        self.db = SQL("mysql://root@127.0.0.1/test")
+
+    def setUp(self):
+        self.db.execute("CREATE TABLE IF NOT EXISTS cs50 (id INTEGER NOT NULL AUTO_INCREMENT, val VARCHAR(16), bin BLOB, PRIMARY KEY (id))")
+        self.db.execute("DELETE FROM cs50")
+
+class PostgresTests(SQLTests):
+    @classmethod
+    def setUpClass(self):
+        self.db = SQL("postgresql://postgres:postgres@127.0.0.1/test")
+
+    def setUp(self):
+        self.db.execute("CREATE TABLE IF NOT EXISTS cs50 (id SERIAL PRIMARY KEY, val VARCHAR(16), bin BYTEA)")
+        self.db.execute("DELETE FROM cs50")
+
+    def test_cte(self):
+        self.assertEqual(self.db.execute("WITH foo AS ( SELECT 1 AS bar ) SELECT bar FROM foo"), [{"bar": 1}])
+
+    def test_postgres_scheme(self):
+        db = SQL("postgres://postgres:postgres@127.0.0.1/test")
+        db.execute("SELECT 1")
+
+class SQLiteTests(SQLTests):
+    @classmethod
+    def setUpClass(self):
+        open("test.db", "w").close()
+        self.db = SQL("sqlite:///test.db")
+
+    def setUp(self):
+        self.db.execute("CREATE TABLE IF NOT EXISTS cs50 (id INTEGER PRIMARY KEY, val TEXT, bin BLOB)")
+        self.db.execute("DELETE FROM cs50")
+
+    def test_lastrowid(self):
+        self.db.execute("CREATE TABLE foo(id INTEGER PRIMARY KEY AUTOINCREMENT, firstname TEXT, lastname TEXT)")
+        self.assertEqual(self.db.execute("INSERT INTO foo (firstname, lastname) VALUES('firstname', 'lastname')"), 1)
+        self.assertRaises(ValueError, self.db.execute, "INSERT INTO foo (id, firstname, lastname) VALUES(1, 'firstname', 'lastname')")
+        self.assertEqual(self.db.execute("INSERT OR IGNORE INTO foo (id, firstname, lastname) VALUES(1, 'firstname', 'lastname')"), None)
 
 if __name__ == "__main__":
     suite = unittest.TestSuite([