def test_create_registration_report(self, mocker):
report_id = '123'
partner_id = '1'
partner_api_key = 'abcd'
mocker.post('https://vr.rockthevote.com/api/v4/registrant_reports.json',
json={'report_id': report_id})
rtv = RockTheVote(partner_id=partner_id, partner_api_key=partner_api_key)
result = rtv.create_registration_report()
self.assertEqual(result, report_id)
def test_get_registration_report(self, mocker):
partner_id = '1'
partner_api_key = 'abcd'
mocker.get('https://vr.rockthevote.com/api/v4/registrant_reports/1',
json={'download_url': 'https://vr.rockthevote.com/download/whatever'})
mocker.get('https://vr.rockthevote.com/download/whatever',
text=open(f'{_dir}/sample.csv').read())
rtv = RockTheVote(partner_id=partner_id, partner_api_key=partner_api_key)
result = rtv.get_registration_report(report_id=1)
self.assertEqual(result.num_rows, 1)
self.assertEqual(result[0]['first_name'], 'Carol')
self.assertEqual(result[0]['last_name'], 'King')
def test_get_state_requirements(self, mocker):
partner_id = '1'
partner_api_key = 'abcd'
with open(f'{_dir}/sample.json', 'r') as j:
expected_json = json.load(j)
mocker.get(
'https://register.rockthevote.com/api/v4/state_requirements.json',
json=expected_json)
rtv = RockTheVote(partner_id=partner_id,
partner_api_key=partner_api_key)
result = rtv.get_state_requirements('en', 'fl', '33314')
print(result.columns)
self.assertEqual(result.num_rows, 1)
self.assertEqual(result[0]['requires_party'], True)
self.assertEqual(result[0]['requires_race'], True)