Fix referendum supporting/opposing

Use ballot_item_selection from referendum to find supporting/opposing
contributions.

disclosure/tests/test_api_supporting_opposing.py

@@ -5,7 +5,7 @@
 from finance.models import Beneficiary
 
 
-@with_form460A_data(test_agency='CSD', test_year='2015')
+@with_form460A_data(test_agency='COS', test_year='2015')
 class OpposingTests(APITestCase):
 
     def do_the_thing_for_candidates(self, support):
@@ -18,6 +18,7 @@ def do_the_thing_for_candidates(self, support):
         url = reverse('candidate_%s' % ('supporting' if support else 'opposing'),
                       kwargs={'candidate_id': candidate.id})
         resp = self.client.get(url)
+        self.assertEquals(resp.status_code, 200)
         self.assertTrue(len(resp.data) > 0)
 
         # TODO: replace dummy tests with live data tests.
@@ -41,7 +42,9 @@ def do_the_thing_for_referendums(self, support):
         url = reverse('referendum_%s' % ('supporting' if support else 'opposing'),
                       kwargs={'referendum_id': referendum.id})
         resp = self.client.get(url)
+        self.assertEquals(resp.status_code, 200)
         self.assertTrue(len(resp.data) > 0)
+        print(resp.data)
 
         # TODO: replace dummy tests with live data tests.
         row = resp.data[0]

disclosure/views.py

@@ -11,7 +11,7 @@
 from rest_framework.response import Response
 
 from .serializers import BeneficiaryMoneyReceivedSerializer
-from ballot.models import Ballot, BallotItemSelection
+from ballot.models import Ballot, BallotItemSelection, ReferendumSelection
 from finance.models import Beneficiary, IndependentMoney
 from finance.views import summarize_money
 from locality.models import Locality
@@ -131,18 +131,24 @@ def supporting(self, request, referendum_id):
         """
         List of committees supporting a referendum, and level of benefits given.
         """
-        # ballot_item_selection_id =   # query from referendumresponse model
+        ballot_item_selection = get_object_or_404(
+            ReferendumSelection,
+            in_favor=True,
+            ballot_item__id=referendum_id)
         return super(ReferendumViewSet, self).supporting(
-            request, ballot_item_selection_id=referendum_id)
+            request, ballot_item_selection_id=ballot_item_selection.id)
 
     @list_route(['GET'])
     def opposing(self, request, referendum_id):
         """
         List of committees opposing a referendum, and level of benefits given.
         """
-        # ballot_item_selection_id =   # query from referendumresponse model
+        ballot_item_selection = get_object_or_404(
+            ReferendumSelection,
+            in_favor=False,
+            ballot_item__id=referendum_id)
         return super(ReferendumViewSet, self).opposing(
-            request, ballot_item_selection_id=referendum_id)
+            request, ballot_item_selection_id=ballot_item_selection.id)
 
 
 class CandidateViewSet(BallotItemSelectionViewSet):