Add serializer tests

Author: datamik

src/country_workspace/migrations/0019_dataserializer_program_serializer.py

@@ -0,0 +1,34 @@
+# Generated by Django 5.2.3 on 2025-07-04 13:18
+
+import concurrency.fields
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("country_workspace", "0018_rdp_hope_rdi_id_alter_rdp_status"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="DataSerializer",
+            fields=[
+                ("id", models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
+                ("last_modified", models.DateTimeField(auto_now=True)),
+                ("version", concurrency.fields.IntegerVersionField(default=0, help_text="record revision number")),
+                ("name", models.SlugField(max_length=64, unique=True)),
+                ("code", models.TextField()),
+            ],
+            options={
+                "abstract": False,
+            },
+        ),
+        migrations.AddField(
+            model_name="program",
+            name="serializer",
+            field=models.ForeignKey(
+                null=True, on_delete=django.db.models.deletion.SET_NULL, to="country_workspace.dataserializer"
+            ),
+        ),
+    ]

src/country_workspace/models/program.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Iterable
 
 from django.db import models
 from django.utils.translation import gettext as _
@@ -123,3 +123,8 @@ def get_checker_for(self, m: type[Validable] | Validable) -> DataChecker:
         if isinstance(m, (Individual | CountryIndividual)) or m in (Individual, CountryIndividual):
             return self.individual_checker
         raise ValueError(m)
+
+    def serialize(self, data: list[dict]) -> Iterable:
+        if self.serializer:
+            return self.serializer.serialize(data)
+        return data

tests/extras/testutils/factories/program.py

@@ -5,6 +5,7 @@
 
 from .base import AutoRegisterModelFactory
 from .office import OfficeFactory
+from .serializer import DataSerializerFactory
 from .smart_import import DataCheckerFactory
 
 
@@ -29,6 +30,7 @@ class ProgramFactory(AutoRegisterModelFactory):
     household_checker = factory.SubFactory(DataCheckerFactory)
     individual_checker = factory.SubFactory(DataCheckerFactory)
     beneficiary_group = factory.SubFactory(BeneficiaryGroupFactory)
+    serializer = factory.SubFactory(DataSerializerFactory)
 
     class Meta:
         model = Program

tests/extras/testutils/factories/serializer.py

@@ -0,0 +1,12 @@
+import factory
+from country_workspace.models.serializer import DataSerializer
+from .base import AutoRegisterModelFactory
+
+
+class DataSerializerFactory(AutoRegisterModelFactory):
+    name = factory.Sequence(lambda n: f"serializer-{n}")
+    code = "function test(data) { return data; }"
+
+    class Meta:
+        model = DataSerializer
+        django_get_or_create = ("name",)

tests/models/test_m_program.py

@@ -0,0 +1,22 @@
+import pytest
+
+from country_workspace.models import Program
+from testutils.factories import ProgramFactory
+
+
+@pytest.fixture
+def program():
+    return ProgramFactory()
+
+
+def test_program_serialize(program: Program):
+    data = [{"foo": "bar"}]
+    result = program.serialize(data)
+    assert result == data
+
+
+def test_program_no_serializer(program: Program):
+    program.serializer = None
+    data = [{"foo": "bar"}]
+    result = program.serialize(data)
+    assert result == data

tests/models/test_m_serializer.py

@@ -0,0 +1,31 @@
+import pytest
+from django.core.exceptions import ValidationError
+
+from country_workspace.models.serializer import DataSerializer
+from testutils.factories.serializer import DataSerializerFactory
+
+
+@pytest.fixture
+def data_serializer():
+    return DataSerializerFactory()
+
+
+def test_serialize(data_serializer: DataSerializer):
+    data = [{"foo": "bar"}]
+    result = data_serializer.serialize(data)
+    assert result == data
+
+
+def test_str(data_serializer: DataSerializer):
+    assert str(data_serializer) == f"DataSerializer: {data_serializer.name}"
+
+
+def test_clean_valid(data_serializer: DataSerializer):
+    data_serializer.clean()
+
+
+@pytest.mark.parametrize("code", ["", "function {"])
+def test_clean_invalid_js(data_serializer: DataSerializer, code: str):
+    data_serializer.code = code
+    with pytest.raises(ValidationError, match="Invalid JavaScript code"):
+        data_serializer.clean()