breaking down table tests

Author: dantownsend

.gitignore

@@ -1,2 +1,3 @@
 *.pyc
 .pytest_cache/
+.mypy_cache/

.vscode/settings.json

@@ -1,3 +1,7 @@
 {
-    "python.pythonPath": "/Users/dantownsend/.virtualenvs/aragorm/bin/python"
+    "python.pythonPath": "/Users/dantownsend/.virtualenvs/aragorm/bin/python",
+    "python.linting.pylintEnabled": false,
+    "python.linting.flake8Enabled": false,
+    "python.linting.enabled": true,
+    "python.linting.mypyEnabled": true
 }

tests/table/test_create.py

@@ -0,0 +1,28 @@
+import asyncio
+
+from ..base import DBTestCase
+from ..example_project.tables import Pokemon
+
+
+class TestCreate(DBTestCase):
+
+    def setUp(self):
+        """
+        Need to override, otherwise table will be auto created.
+        """
+        pass
+
+    def test_create_table(self):
+        async def create_table():
+            return await Pokemon.create().execute()
+
+        async def count_rows():
+            return await Pokemon.select(
+                'name', 'trainer', 'power'
+            ).count().execute()
+
+        asyncio.run(create_table())
+
+        # Just do a count to make sure the table was created ok.
+        response = asyncio.run(count_rows())
+        self.assertEqual(response[0]['count'], 0)

tests/table/test_delete.py

@@ -0,0 +1,29 @@
+import asyncio
+
+from ..base import DBTestCase
+from ..example_project.tables import Pokemon
+
+
+class TestDelete(DBTestCase):
+
+    def test_delete(self):
+        self.insert_rows()
+
+        async def delete_pokemon():
+            return await Pokemon.delete().where(
+                Pokemon.name == 'weedle'
+            ).execute()
+
+        async def check_pokemon():
+            return await Pokemon.select().where(
+                Pokemon.name == 'weedle'
+            ).count().execute()
+
+        asyncio.run(delete_pokemon())
+        response = asyncio.run(check_pokemon())
+        print(f'response = {response}')
+
+        self.assertEqual(
+            response,
+            [{'count': 0}]
+        )

tests/table/test_metaclass.py

@@ -0,0 +1,9 @@
+from unittest import TestCase
+
+from ..example_project.tables import Pokemon
+
+
+class TestMetaClass(TestCase):
+
+    def test_tablename(self):
+        self.assertEqual(Pokemon.Meta.tablename, 'pokemon')

tests/table/test_select.py

@@ -1,7 +1,4 @@
 import asyncio
-from unittest import TestCase
-
-import asyncpg
 
 from ..base import DBTestCase
 from ..example_project.tables import Pokemon
@@ -282,87 +279,3 @@ async def get_pokemon():
             response,
             [{'count': 1}]
         )
-
-
-class TestUpdate(DBTestCase):
-
-    def test_update(self):
-        self.insert_rows()
-
-        async def update_pokemon():
-            return await Pokemon.update(
-                name='kakuna'
-            ).where(
-                Pokemon.name == 'weedle'
-            ).execute()
-
-        async def check_pokemon():
-            return await Pokemon.select(
-                'name'
-            ).where(
-                Pokemon.name == 'kakuna'
-            ).execute()
-
-        asyncio.run(update_pokemon())
-        response = asyncio.run(check_pokemon())
-        print(f'response = {response}')
-
-        self.assertEqual(
-            response,
-            [{'name': 'kakuna'}]
-        )
-
-
-class TestDelete(DBTestCase):
-
-    def test_delete(self):
-        self.insert_rows()
-
-        async def delete_pokemon():
-            return await Pokemon.delete().where(
-                Pokemon.name == 'weedle'
-            ).execute()
-
-        async def check_pokemon():
-            return await Pokemon.select().where(
-                Pokemon.name == 'weedle'
-            ).count().execute()
-
-        asyncio.run(delete_pokemon())
-        response = asyncio.run(check_pokemon())
-        print(f'response = {response}')
-
-        self.assertEqual(
-            response,
-            [{'count': 0}]
-        )
-
-
-class TestCreate(DBTestCase):
-
-    def setUp(self):
-        """
-        Need to override, otherwise table will be auto created.
-        """
-        pass
-
-    def test_create_table(self):
-        async def create_table():
-            return await Pokemon.create().execute()
-
-        async def count_rows():
-            return await Pokemon.select(
-                'name', 'trainer', 'power'
-            ).count().execute()
-
-        asyncio.run(create_table())
-
-        # Just do a count to make sure the table was created ok.
-        response = asyncio.run(count_rows())
-        self.assertEqual(response[0]['count'], 0)
-
-
-class TestMetaClass(TestCase):
-
-    def test_tablename(self):
-        self.assertEqual(Pokemon.Meta.tablename, 'pokemon')

tests/table/test_update.py

@@ -0,0 +1,35 @@
+import asyncio
+
+import asyncpg
+
+from ..base import DBTestCase
+from ..example_project.tables import Pokemon
+
+
+class TestUpdate(DBTestCase):
+
+    def test_update(self):
+        self.insert_rows()
+
+        async def update_pokemon():
+            return await Pokemon.update(
+                name='kakuna'
+            ).where(
+                Pokemon.name == 'weedle'
+            ).execute()
+
+        async def check_pokemon():
+            return await Pokemon.select(
+                'name'
+            ).where(
+                Pokemon.name == 'kakuna'
+            ).execute()
+
+        asyncio.run(update_pokemon())
+        response = asyncio.run(check_pokemon())
+        print(f'response = {response}')
+
+        self.assertEqual(
+            response,
+            [{'name': 'kakuna'}]
+        )