fleshing out tests

Author: dantownsend

.gitignore

@@ -0,0 +1,2 @@
+*.pyc
+.pytest_cache/

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python.pythonPath": "/Users/dantownsend/.virtualenvs/aragorm/bin/python"
+}

aragorm/model.py

@@ -22,7 +22,11 @@ class Value(object):
 
 class Where(object):
     """
-    Example use case -
+    Example use case - id = 5
+
+    Can potentially simplify things dramatically here by just accepting
+    postgres where clauses directly, avoiding the need for equal, greater than
+    etc syntax.
     """
 
     def __init__(self, field: str, value: Value):

aragorm/tests/test_model.py

@@ -0,0 +1,54 @@
+# Need a test runner ...
+import asyncio
+from unittest import TestCase
+
+import asyncpg
+
+
+TEST_CREDENTIALS = {
+    'host': 'localhost',
+    'database': 'aragorm',
+    'user': 'aragorm',
+    'password': 'aragorm'
+}
+
+
+class TestQuery(TestCase):
+
+    async def create_table(self):
+        conn = await asyncpg.connect(**TEST_CREDENTIALS)
+        await conn.execute('''
+            CREATE TABLE pokemon (
+                name VARCHAR(50),
+                power SMALLINT
+            );'''
+        )
+        await conn.close()
+
+    async def insert_rows(self):
+        conn = await asyncpg.connect(**TEST_CREDENTIALS)
+        await conn.execute('''
+            INSERT INTO pokemon (
+                name,
+                power
+            ) VALUES (
+                'pikachu',
+                1000
+            );'''
+        )
+
+    async def drop_table(self):
+        conn = await asyncpg.connect(**TEST_CREDENTIALS)
+        await conn.execute('DROP TABLE pokemon;')
+        await conn.close()
+
+    def setUp(self):
+        asyncio.run(self.create_table())
+
+    def test_query(self):
+        print('hello there ...')
+        asyncio.run(self.insert_rows())
+        breakpoint()
+
+    def tearDown(self):
+        asyncio.run(self.drop_table())

setup.py

@@ -0,0 +1,3 @@
+# asyncpg==0.17.0
+# pytest==3.7.2
+# uvloop==0.11.2