summaryrefslogtreecommitdiff
path: root/examples/18-database/output
diff options
context:
space:
mode:
Diffstat (limited to 'examples/18-database/output')
-rw-r--r--examples/18-database/output/sqlite3/test.py24
1 files changed, 24 insertions, 0 deletions
diff --git a/examples/18-database/output/sqlite3/test.py b/examples/18-database/output/sqlite3/test.py
new file mode 100644
index 0000000..885212b
--- /dev/null
+++ b/examples/18-database/output/sqlite3/test.py
@@ -0,0 +1,24 @@
+import os.path
+import sqlite3
+import unittest
+
+
+class DataTestCase(unittest.TestCase):
+ def test_create(self):
+ sql_script_path = os.path.join(os.path.dirname(__file__), "scripts/data/Data.sql")
+ with open(sql_script_path, "rt", encoding="utf-8") as sql_script_file:
+ sql_script = sql_script_file.read()
+
+ connection = sqlite3.connect(":memory:", autocommit=False)
+ cursor = connection.cursor()
+ cursor.executescript(sql_script)
+ connection.commit()
+
+ cursor.execute("INSERT INTO authors VALUES (1, 'Alice')")
+ cursor.execute("INSERT INTO authors VALUES (2, 'Bob')")
+ cursor.execute("INSERT INTO authors VALUES (3, 'Carol')")
+
+ cursor.execute("SELECT id, name FROM authors WHERE id > 1 ORDER BY id")
+ rows = cursor.fetchall()
+
+ self.assertEqual(rows, [(2, "Bob"), (3, "Carol")])