WIP add result_set to ballot serializer

ynr/apps/elections/api/next/serializers.py

@@ -9,6 +9,8 @@
     MinimalPostSerializer,
 )
 from rest_framework import serializers
+from uk_results import models as uk_results_models
+from uk_results.api.next.serializers import MinimalResultSerializer
 
 
 class MinimalElectionSerializer(serializers.HyperlinkedModelSerializer):
@@ -102,6 +104,7 @@ class Meta:
             "replaces",
             "replaced_by",
             "uncontested",
+            "results",
         )
 
     replaces = serializers.SlugRelatedField(
@@ -126,6 +129,7 @@ class Meta:
     post = MinimalPostSerializer(read_only=True)
     sopn = serializers.SerializerMethodField()
     candidacies = serializers.SerializerMethodField()
+    results = MinimalResultSerializer(read_only=True, source="resultset")
 
     results_url = serializers.HyperlinkedIdentityField(
         view_name="resultset-detail",
@@ -182,3 +186,15 @@ def get_cancelled(self, instance):
         if instance.cancelled:
             return True
         return instance.uncontested
+
+    @swagger_serializer_method(serializer_or_field=MinimalResultSerializer)
+    def get_result_set(self, instance):
+        """
+        If the ballot has a result set, return it
+        """
+        try:
+            return MinimalResultSerializer(
+                instance.result_set, read_only=True
+            ).data
+        except uk_results_models.MinimalResultsSerializer.DoesNotExist:
+            return None

ynr/apps/elections/tests/test_serializers.py

@@ -43,3 +43,15 @@ def test_when_uncontested_and_cancelled(self):
             serializer = BallotSerializer(instance=ballot)
             cancelled = serializer.get_cancelled(instance=ballot)
             self.assertTrue(cancelled)
+
+    def test_get_results_set_when_none(self):
+        ballot = BallotPaperFactory()
+        serializer = BallotSerializer(instance=ballot)
+        result_set = serializer.get_result_set(instance=ballot)
+        self.assertIsNone(result_set)
+
+    def test_get_results_set_with_results(self):
+        ballot = BallotPaperFactory()
+        serializer = BallotSerializer(instance=ballot)
+        results_set = serializer.get_result_set(instance=ballot)
+        self.assertIsNotNone(results_set)

ynr/apps/uk_results/api/next/serializers.py

@@ -68,6 +68,18 @@ def get_url(self, obj):
     candidate_results = CandidateResultSerializer(many=True)
 
 
+class MinimalResultSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ResultSet
+        fields = (
+            "num_turnout_reported",
+            "turnout_percentage",
+            "num_spoilt_ballots",
+            "source",
+            "total_electorate",
+        )
+
+
 class ElectedSerializer(CandidacyOnBallotSerializer):
     class Meta:
         model = Membership