added test, tolerating non-DDL, non-DML keywords like BEGIN

dmalan · dmalan · commit a130c1b17a2e · 2020-01-30T20:33:28.000-05:00
diff --git a/setup.py b/setup.py
@@ -16,5 +16,5 @@
     package_dir={"": "src"},
     packages=["cs50"],
     url="https://github.com/cs50/python-cs50",
-    version="5.0.2"
+    version="5.0.3"
 )
diff --git a/src/cs50/sql.py b/src/cs50/sql.py
@@ -126,7 +126,7 @@ def execute(self, sql, *args, **kwargs):
                 command = token.value.upper()
                 break
         else:
-            raise RuntimeError("unrecognized command")
+            command = None
 
         # Flatten statement
         tokens = list(statements[0].flatten())
diff --git a/tests/sql.py b/tests/sql.py
@@ -128,6 +128,9 @@ def test_rollback(self):
         self.db.execute("ROLLBACK")
         self.assertEqual(self.db.execute("SELECT val FROM cs50"), [])
 
+    def test_cte(self):
+        self.assertEqual(self.db.execute("WITH foo AS ( SELECT 1 ) SELECT * FROM foo"), [{"1": 1}])
+
     def tearDown(self):
         self.db.execute("DROP TABLE cs50")
         self.db.execute("DROP TABLE IF EXISTS foo")

Original file line number	Diff line number	Diff line change
`@@ -16,5 +16,5 @@`
`16`	`16`	`package_dir={"": "src"},`
`17`	`17`	`packages=["cs50"],`
`18`	`18`	`url="https://github.com/cs50/python-cs50",`
`19`		`- version="5.0.2"`
	`19`	`+ version="5.0.3"`
`20`	`20`	`)`