Add TestXLSXListField

Author: browniebroke

tests/test_fields.py

@@ -12,9 +12,10 @@
     DateTimeField,
     DateField,
     TimeField,
+    ListField,
 )
 
-from drf_excel.fields import XLSXField, XLSXNumberField, XLSXDateField
+from drf_excel.fields import XLSXField, XLSXNumberField, XLSXDateField, XLSXListField
 from drf_excel.utilities import XLSXStyle
 
 
@@ -304,3 +305,50 @@ def test_cell_date_custom_format(
         assert isinstance(cell, Cell)
         assert cell.value == dt.date(2017, 10, 25)
         assert cell.number_format == "dd/mm/yyyy"
+
+
+class TestXLSXListField:
+    def test_cell_default_separator(self, style: XLSXStyle, worksheet: Worksheet):
+        f = XLSXListField(
+            list_sep=None,
+            key="things",
+            value=["foo", "bar", "baz"],
+            field=ListField(),
+            style=style,
+            mapping=None,
+            cell_style=style,
+        )
+        assert f.original_value == f.value == ["foo", "bar", "baz"]
+        cell = f.cell(worksheet, 1, 1)
+        assert isinstance(cell, Cell)
+        assert cell.value == "foo, bar, baz"
+
+    def test_cell_custom_separator(self, style: XLSXStyle, worksheet: Worksheet):
+        f = XLSXListField(
+            list_sep=";",
+            key="things",
+            value=["john", "doe", "[email protected]"],
+            field=ListField(),
+            style=style,
+            mapping=None,
+            cell_style=style,
+        )
+        assert f.original_value == f.value == ["john", "doe", "[email protected]"]
+        cell = f.cell(worksheet, 1, 1)
+        assert isinstance(cell, Cell)
+        assert cell.value == "john;doe;[email protected]"
+
+    def test_cell_complex_types(self, style: XLSXStyle, worksheet: Worksheet):
+        f = XLSXListField(
+            list_sep=None,
+            key="objs",
+            value=[{"a": 1}, {"b": 2}],
+            field=ListField(),
+            style=style,
+            mapping=None,
+            cell_style=style,
+        )
+        assert f.original_value == f.value == [{"a": 1}, {"b": 2}]
+        cell = f.cell(worksheet, 1, 1)
+        assert isinstance(cell, Cell)
+        assert cell.value == '[{"a": 1}, {"b": 2}]'