def setUp(self):
self.van = VAN(os.environ['VAN_API_KEY'],
db="MyVoters",
raise_for_status=False)
class TestActivistCodes(unittest.TestCase):
def setUp(self):
self.van = VAN(os.environ['VAN_API_KEY'],
db="MyVoters",
raise_for_status=False)
def tearDown(self):
pass
@requests_mock.Mocker()
def test_get_activist_codes(self, m):
# Create response
json = {
u'count':
43,
u'items': [{
u'status': u'Active',
u'scriptQuestion': None,
u'name': u'TEST CODE',
u'mediumName': u'TEST CODE',
u'activistCodeId': 4388538,
u'shortName': u'TC',
u'type': u'Action',
u'description': None
}],
u'nextPageLink':
None
}
m.get(self.van.connection.uri + 'activistCodes', json=json)
# Expected Structure
expected = [
'status', 'scriptQuestion', 'name', 'mediumName', 'activistCodeId',
'shortName', 'type', 'description'
]
# Assert response is expected structure
self.assertTrue(validate_list(expected, self.van.get_activist_codes()))
# To Do: Test what happens when it doesn't find any ACs
@requests_mock.Mocker()
def test_get_activist_code(self, m):
# Create response
json = {
"status": "Active",
"scriptQuestion": "null",
"name": "Anti-Choice",
"mediumName": "Anti",
"activistCodeId": 4135099,
"shortName": "AC",
"type": "Constituency",
"description": "A person who has been flagged as anti-choice."
}
m.get(self.van.connection.uri + 'activistCodes/4388538', json=json)
self.assertEqual(json, self.van.get_activist_code(4388538))
@requests_mock.Mocker()
def test_toggle_activist_code(self, m):
# Test apply activist code
m.post(self.van.connection.uri + 'people/2335282/canvassResponses',
status_code=204)
self.assertTrue(
self.van.toggle_activist_code(2335282, 4429154, 'apply'), 204)
# Test remove activist code
m.post(self.van.connection.uri + 'people/2335282/canvassResponses',
status_code=204)
self.assertTrue(
self.van.toggle_activist_code(2335282, 4429154, 'remove'), 204)
@requests_mock.Mocker()
def test_apply_activist_code(self, m):
# Test apply activist code
m.post(self.van.connection.uri + 'people/2335282/canvassResponses',
status_code=204)
self.assertEqual(self.van.apply_activist_code(2335282, 4429154), 204)
@requests_mock.Mocker()
def test_remove_activist_code(self, m):
# Test remove activist code
m.post(self.van.connection.uri + 'people/2335282/canvassResponses',
status_code=204)
self.assertEqual(self.van.remove_activist_code(2335282, 4429154), 204)
class TestTargets(unittest.TestCase):
mock_data = mock_data = '12827,Volunteer Recruitment Tiers,Tier,109957740\n12827,Volunteer Recruitment Tiers,Tier,109957754'
mock_result = Table([
('12827', 'Volunteer Recruitment Tiers', 'Tier', '109957740'),
('12827', 'Volunteer Recruitment Tiers', 'Tier', '109957754')
])
def setUp(self):
self.van = VAN(os.environ['VAN_API_KEY'],
db="MyVoters",
raise_for_status=False)
def tearDown(self):
pass
@requests_mock.Mocker()
def test_get_targets(self, m):
# Create response
json = {
u'count':
2,
u'items': [{
u'targetId': 12827,
u'type': u'TEST CODE',
u'name': u'TEST CODE',
u'description': None,
u'points': 20,
u'areSubgroupsSticky': False,
u'status': u'Active',
u'subgroups': None,
u'markedSubgroup': None
}],
u'nextPageLink':
None
}
m.get(self.van.connection.uri + 'targets', json=json)
# Expected Structure
expected = [
'targetId', 'type', 'name', 'description', 'points',
'areSubgroupsSticky', 'status', 'subgroups', 'markedSubgroup'
]
# Assert response is expected structure
self.assertTrue(validate_list(expected, self.van.get_targets()))
# To Do: Test what happens when it doesn't find any targets
@requests_mock.Mocker()
def test_get_target(self, m):
# Create response
json = {
u'targetId':
15723,
u'name':
u'Mail_VR_Chase',
u'type':
u'Dynamic',
u'description':
None,
u'points':
15,
u'areSubgroupsSticky':
False,
u'status':
u'Active',
u'subgroups': [{
u'targetId': 12827,
u'fullName': u'April_VR_Chase Calls',
u'name': u'April_Chase_20',
u'subgroupId': 46803,
u'isAssociatedWithBadges': True
}],
u'markedSubgroup':
None
}
m.get(self.van.connection.uri + 'targets/15723', json=json)
self.assertEqual(json, self.van.get_target(15723))
@requests_mock.Mocker()
def test_create_target_export(self, m):
export_job_id = '{"exportJobId": "455961790"}'
target_id = 12827
m.post(self.van.connection.uri + 'targetExportJobs',
json=export_job_id,
status_code=204)
# Test that it doesn't throw and error
r = self.van.create_target_export(target_id, webhook_url=None)
self.assertEqual(r, export_job_id)
@requests_mock.Mocker()
def test_get_target_export(self, m):
export_job_id = 455961790
json = [{
"targetId": 12827,
"file": {
"downloadUrl":
"https://ngpvan.blob.core.windows.net/target-export-files/TargetExport_455961790.csv",
"dateExpired": "null",
"recordCount": 1016883
},
"webhookUrl": "null",
"exportJobId": 455961790,
"jobStatus": "Complete"
}]
download_url = 'https://ngpvan.blob.core.windows.net/target-export-files/TargetExport_455961790.csv'
m.post(self.van.connection.uri + 'targetExportJobs',
json=export_job_id,
status_code=204)
m.get(self.van.connection.uri + f'targetExportJobs/455961790',
json=json)
m.get(download_url, text=self.mock_data)
assert_matching_tables(self.van.get_target_export(export_job_id),
self.mock_result)