added tests for 'main', and moved tests for 'Where'

Author: dantownsend

tests/columns/test_combination.py

@@ -0,0 +1,30 @@
+import unittest
+
+from ..example_project.tables import Band
+
+
+class TestWhere(unittest.TestCase):
+    def test_equals(self):
+        _where = Band.name == "Pythonistas"
+        sql = _where.__str__()
+        self.assertEqual(sql, "band.name = 'Pythonistas'")
+
+    def test_not_equal(self):
+        _where = Band.name != "CSharps"
+        sql = _where.__str__()
+        self.assertEqual(sql, "band.name != 'CSharps'")
+
+    def test_like(self):
+        _where = Band.name.like("Python%")
+        sql = _where.__str__()
+        self.assertEqual(sql, "band.name LIKE 'Python%'")
+
+    def test_is_in(self):
+        _where = Band.name.is_in(["Pythonistas", "Rustaceans"])
+        sql = _where.__str__()
+        self.assertEqual(sql, "band.name IN ('Pythonistas', 'Rustaceans')")
+
+    def test_not_in(self):
+        _where = Band.name.not_in(["CSharps"])
+        sql = _where.__str__()
+        self.assertEqual(sql, "band.name NOT IN ('CSharps')")

tests/conftest.py

@@ -15,6 +15,8 @@ async def drop_tables():
         "ticket",
         "poster",
         "migration",
+        "musician",
+        "my_table",
     ]:
         await ENGINE._run_in_new_connection(f"DROP TABLE IF EXISTS {table}")
 

tests/test_field.py

@@ -0,0 +1,9 @@
+from unittest import TestCase
+
+from piccolo.main import main
+
+
+class TestMain(TestCase):
+    def test_main(self):
+        # Just make sure it runs without raising any errors.
+        main()