added test for serialising LazyTableReference in migrations

Author: dantownsend

tests/apps/migrations/auto/test_serialisation.py

@@ -1,8 +1,8 @@
 from unittest import TestCase
 
 from piccolo.apps.migrations.auto.serialisation import serialise_params
-
 from piccolo.columns.defaults import DateNow, TimeNow, TimestampNow, UUID4
+from piccolo.columns.reference import LazyTableReference
 
 
 class TestSerialiseParams(TestCase):
@@ -28,3 +28,33 @@ def test_timestamp(self):
     def test_uuid(self):
         serialised = serialise_params(params={"default": UUID4()})
         self.assertTrue(serialised.params["default"].__repr__() == "UUID4()")
+
+    def test_lazy_table_reference(self):
+        # These are equivalent:
+        references_list = [
+            LazyTableReference(
+                table_class_name="Manager", app_name="example_app"
+            ),
+            LazyTableReference(
+                table_class_name="Manager",
+                module_path="tests.example_app.tables",
+            ),
+        ]
+
+        for references in references_list:
+            serialised = serialise_params(params={"references": references})
+            self.assertTrue(
+                serialised.params["references"].__repr__() == "Manager"
+            )
+
+            self.assertTrue(len(serialised.extra_imports) == 1)
+            self.assertEqual(
+                serialised.extra_imports[0].__str__(),
+                "from piccolo.table import Table",
+            )
+
+            self.assertTrue(len(serialised.extra_definitions) == 1)
+            self.assertEqual(
+                serialised.extra_definitions[0].__str__(),
+                'class Manager(Table, tablename="manager"): pass',
+            )