def test_fields(self):
factories.CandidateHistoryFactory(candidate_id='P01',
two_year_period=2014,
candidate_election_year=2016),
factories.CandidateHistoryFactory(candidate_id='P01',
two_year_period=2016,
candidate_election_year=2016),
factories.ScheduleEByCandidateFactory(
candidate_id='P01',
total=100,
cycle=2012,
committee_id='C01',
support_oppose_indicator='S',
),
factories.ScheduleEByCandidateFactory(
candidate_id='P01',
total=200,
cycle=2014,
committee_id='C02',
support_oppose_indicator='S',
),
factories.ScheduleEByCandidateFactory(
candidate_id='P01',
total=300,
cycle=2014,
committee_id='C02',
support_oppose_indicator='O',
),
factories.ScheduleEByCandidateFactory(
candidate_id='P01',
total=400,
cycle=2016,
committee_id='C03',
support_oppose_indicator='S',
),
factories.ScheduleEByCandidateFactory(
candidate_id='P01',
total=500,
cycle=2016,
committee_id='C03',
support_oppose_indicator='O',
),
results = self._results(
api.url_for(IETotalsByCandidateView,
candidate_id='P01',
election_full=False))
assert len(results) == 4
results = self._results(
api.url_for(IETotalsByCandidateView,
candidate_id='P01',
election_full=True))
assert len(results) == 2
assert results[0]['total'] == 800
assert results[1]['total'] == 600
def test_sort_by_committee_name(self):
factories.CommitteeHistoryFactory(name='Warner for America',
cycle=2010,
committee_id='C005')
factories.CommitteeHistoryFactory(name='Ritche for America',
cycle=2010,
committee_id='C006')
factories.CandidateHistoryFactory(
candidate_id='S005',
name='WARNER, MARK',
election_years=[2010],
two_year_period=2010,
office='S',
state='NY',
)
factories.CandidateHistoryFactory(
candidate_id='S006',
name='BALDWIN, ALISSA',
election_years=[2010],
two_year_period=2010,
office='S',
state='NY',
)
factories.CandidateElectionFactory(candidate_id='S005',
cand_election_year=2010)
factories.CandidateElectionFactory(candidate_id='S006',
cand_election_year=2010)
factories.ScheduleEByCandidateFactory(
total=50000,
count=10,
cycle=2010,
candidate_id='S005',
support_oppose_indicator='S',
committee_id='C005',
),
factories.ScheduleEByCandidateFactory(
total=10000,
count=5,
cycle=2010,
candidate_id='S005',
support_oppose_indicator='S',
committee_id='C006',
),
response = self._results(
api.url_for(ScheduleEByCandidateView,
sort='-committee_name',
office='senate',
cycle=2010,
state='NY'))
self.assertEqual(len(response), 2)
self.assertEqual(response[0]['committee_name'], 'Warner for America')
self.assertEqual(response[1]['committee_name'], 'Ritche for America')
def test_sort_by_candidate_name_descending(self):
factories.CandidateHistoryFactory(
candidate_id='S005',
name='WARNER, MARK',
two_year_period=2010,
office='S',
state='NY',
)
factories.CandidateHistoryFactory(
candidate_id='S006',
name='BALDWIN, ALISSA',
two_year_period=2010,
office='S',
state='NY',
)
factories.CandidateElectionFactory(candidate_id='S005',
cand_election_year=2010)
factories.CandidateElectionFactory(candidate_id='S006',
cand_election_year=2010)
factories.ScheduleEByCandidateFactory(
total=50000,
count=10,
cycle=2010,
candidate_id='S005',
support_oppose_indicator='S',
),
factories.ScheduleEByCandidateFactory(
total=10000,
count=5,
cycle=2010,
candidate_id='S006',
support_oppose_indicator='S',
),
response = self._results(
api.url_for(ScheduleEByCandidateView,
cycle=2010,
office='senate',
state='NY',
sort='-candidate_name'))
self.assertEqual(len(response), 2)
self.assertEqual(response[0]['candidate_name'], 'WARNER, MARK')
self.assertEqual(response[1]['candidate_name'], 'BALDWIN, ALISSA')
def test_fields(self):
candidate_id = 'S001'
support_oppose_indicator = 'S'
factories.ScheduleEByCandidateFactory(
candidate_id=candidate_id,
support_oppose_indicator=support_oppose_indicator,
cycle=2018)
factories.CandidateElectionFactory(candidate_id=candidate_id,
cand_election_year=2018)
results = self._results(
api.url_for(ScheduleEByCandidateView, candidate_id='S001'))
assert len(results) == 1
assert results[0].keys() == ScheduleEByCandidateSchema().fields.keys()
def test_sched_e_filter_match(self):
factories.CandidateElectionFactory(candidate_id='S002',
cand_election_year=2014)
factories.ScheduleEByCandidateFactory(
total=50000,
count=10,
cycle=2008,
candidate_id='S001',
support_oppose_indicator='O',
),
factories.ScheduleEByCandidateFactory(
total=10000,
count=5,
cycle=2010,
candidate_id='S002',
support_oppose_indicator='O',
),
results = self._results(
api.url_for(ScheduleEByCandidateView,
candidate_id='S002',
cycle=2014,
support_oppose_indicator='S'))
self.assertEqual(len(results), 1)
def test_sort_by_candidate_id(self):
factories.CandidateHistoryFactory(
candidate_id='S001',
name='Robert Ritchie',
two_year_period=2012,
office='S',
state='NY',
)
factories.CandidateHistoryFactory(
candidate_id='S002',
name='FARLEY Ritchie',
two_year_period=2012,
office='S',
state='NY',
)
factories.CandidateHistoryFactory(
candidate_id='S003',
name='Robert Ritchie',
election_years=[2012],
two_year_period=2012,
office='S',
state='NY',
)
factories.CandidateElectionFactory(candidate_id='S001',
cand_election_year=2012)
factories.CandidateElectionFactory(candidate_id='S002',
cand_election_year=2012)
factories.CandidateElectionFactory(candidate_id='S003',
cand_election_year=2012)
factories.ScheduleEByCandidateFactory(
cycle=2012,
candidate_id='S001',
support_oppose_indicator='O',
total=700,
count=5,
),
factories.ScheduleEByCandidateFactory(
cycle=2012,
candidate_id='S002',
support_oppose_indicator='O',
total=500,
count=3,
),
factories.ScheduleEByCandidateFactory(
cycle=2012,
candidate_id='S003',
support_oppose_indicator='S',
total=100,
count=1,
),
response = self._results(
api.url_for(ScheduleEByCandidateView,
sort='-candidate_id',
office='senate',
state='NY',
cycle=2012))
self.assertEqual(len(response), 3)
self.assertEqual(response[0]['candidate_id'], 'S003')
self.assertEqual(response[0]['support_oppose_indicator'], 'S')
self.assertEqual(response[1]['candidate_id'], 'S002')
self.assertEqual(response[1]['support_oppose_indicator'], 'O')
self.assertEqual(response[2]['candidate_id'], 'S001')
self.assertEqual(response[2]['support_oppose_indicator'], 'O')
