feat(Segment Membership): Seed identities on Beta opt-in

api/tests/unit/segment_membership/test_unit_segment_membership_admin.py

@@ -0,0 +1,29 @@
+from django.contrib.admin.sites import AdminSite
+from django.utils import timezone
+from pytest_mock import MockerFixture
+from segment_membership.admin import SegmentMembershipSeedAdmin
+
+from organisations.models import Organisation
+from segment_membership.models import SegmentMembershipSeed
+
+
+def test_segment_membership_seed_admin_force_reseed__queryset__clears_marker_and_enqueues_seed(
+    organisation: Organisation,
+    mocker: MockerFixture,
+) -> None:
+    # Given
+    seed = SegmentMembershipSeed.objects.create(
+        organisation=organisation, seeded_at=timezone.now()
+    )
+    seed_task = mocker.patch("segment_membership.admin.seed_organisation_identities")
+    admin = SegmentMembershipSeedAdmin(SegmentMembershipSeed, AdminSite())
+
+    # When
+    admin.force_reseed(
+        request=mocker.MagicMock(), queryset=SegmentMembershipSeed.objects.all()
+    )
+
+    # Then
+    seed.refresh_from_db()
+    assert seed.seeded_at is None
+    seed_task.delay.assert_called_once_with(args=(organisation.id,))

api/tests/unit/segment_membership/test_unit_segment_membership_mappers.py

@@ -1,11 +1,14 @@
+from datetime import datetime, timezone
 from decimal import Decimal
 
 import pytest
 from flagsmith_schemas.dynamodb import Identity as DynamoIdentity
 
 from segment_membership.mappers import map_identity_document_to_clickhouse_row
+from segment_membership.types import ClickHouseIdentityRow
 
 UUID_A = "f47ac10b-58cc-4372-a567-0e02b2c3d479"
+INSERTED_AT = datetime(2026, 5, 8, 12, 0, 0, tzinfo=timezone.utc)
 
 
 @pytest.mark.parametrize(
@@ -22,7 +25,7 @@
                     {"trait_key": "plan", "trait_value": "growth"},
                 ],
             },
-            ("env-key", "alice", "env_x_alice", {"plan": "growth"}),
+            ("env-key", "alice", "env_x_alice", {"plan": "growth"}, INSERTED_AT),
             id="single string trait",
         ),
         pytest.param(
@@ -34,7 +37,7 @@
                 "created_date": "2026-05-08T00:00:00Z",
                 "identity_traits": [],
             },
-            ("env-key", "alice", "env_x_alice", None),
+            ("env-key", "alice", "env_x_alice", None, INSERTED_AT),
             id="empty traits collapse to NULL",
         ),
         pytest.param(
@@ -48,7 +51,7 @@
                     {"trait_key": "age", "trait_value": Decimal("18")},
                 ],
             },
-            ("env-key", "alice", "env_x_alice", {"age": 18}),
+            ("env-key", "alice", "env_x_alice", {"age": 18}, INSERTED_AT),
             id="whole-number Decimal narrows to int",
         ),
         pytest.param(
@@ -62,7 +65,7 @@
                     {"trait_key": "score", "trait_value": Decimal("1.5")},
                 ],
             },
-            ("env-key", "alice", "env_x_alice", {"score": 1.5}),
+            ("env-key", "alice", "env_x_alice", {"score": 1.5}, INSERTED_AT),
             id="fractional Decimal narrows to float",
         ),
         pytest.param(
@@ -82,16 +85,17 @@
                 "alice",
                 "env_x_alice",
                 {"plan": "growth", "team": "alpha"},
+                INSERTED_AT,
             ),
             id="multiple traits flatten to a single dict",
         ),
     ],
 )
 def test_map_identity_document_to_clickhouse_row__cases__return_expected(
     doc: DynamoIdentity,
-    expected: tuple[str, str, str, dict[str, object] | None],
+    expected: ClickHouseIdentityRow,
 ) -> None:
-    # Given a Dynamo identity document
-    # When mapped onto an IDENTITIES row
-    # Then it lines up positionally with the IDENTITIES schema
-    assert map_identity_document_to_clickhouse_row("env-key", doc) == expected
+    # Given / When / Then
+    assert (
+        map_identity_document_to_clickhouse_row("env-key", doc, INSERTED_AT) == expected
+    )