added a new Table to the example app, for testing JSON columns

Author: dantownsend

tests/conf/test_apps.py

@@ -70,6 +70,7 @@ def test_table_finder(self):
                 "Concert",
                 "Manager",
                 "Poster",
+                "RecordingStudio",
                 "Shirt",
                 "Ticket",
                 "Venue",
@@ -95,6 +96,7 @@ def test_table_finder_coercion(self):
                 "Concert",
                 "Manager",
                 "Poster",
+                "RecordingStudio",
                 "Shirt",
                 "Ticket",
                 "Venue",
@@ -129,5 +131,13 @@ def test_exclude_tags(self):
 
         self.assertEqual(
             table_class_names,
-            ["Band", "Concert", "Manager", "Shirt", "Ticket", "Venue"],
+            [
+                "Band",
+                "Concert",
+                "Manager",
+                "RecordingStudio",
+                "Shirt",
+                "Ticket",
+                "Venue",
+            ],
         )

tests/conftest.py

@@ -17,6 +17,7 @@ async def drop_tables():
         "migration",
         "musician",
         "my_table",
+        "recording_studio",
         "shirt",
     ]:
         await ENGINE._run_in_new_connection(f"DROP TABLE IF EXISTS {table}")

tests/example_app/tables.py

@@ -1,7 +1,15 @@
 from enum import Enum
 
 from piccolo.table import Table
-from piccolo.columns import Varchar, ForeignKey, Integer, Numeric, Text
+from piccolo.columns import (
+    ForeignKey,
+    Integer,
+    JSON,
+    JSONB,
+    Numeric,
+    Text,
+    Varchar,
+)
 from piccolo.columns.readable import Readable
 
 ###############################################################################
@@ -60,3 +68,12 @@ class Size(str, Enum):
         large = "l"
 
     size = Varchar(length=1, choices=Size, default=Size.large)
+
+
+class RecordingStudio(Table):
+    """
+    Used for testing JSON and JSONB columns.
+    """
+
+    facilities = JSON()
+    facilities_b = JSONB()