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)
participants = zoom.get_past_meeting_participants(ZOOM_MEETING_ID)
filtered_participants = participants.select_rows(lambda row: row.duration > MINIMUM_DURATION)
# Coalesce the columns into something VAN expects
column_map = {"first_name": ["fn", "first", "firstname", "first name"],
"last_name": ["ln", "last", "lastname", "last name"],
"date_of_birth": ["dob", "date of birth", "birthday"],
"email": ["email address", "email_address"],
"street_number": ["street number", "street no.", "street no", "street #"],
"street_name": ["street name", "street"],
"phone": ["phone_number", "phone #", "phone_#", "phone no.", "phone no"],
"zip": ["zip5", "zipcode", "zip code"]}
filtered_participants.map_and_coalesce_columns(column_map)
# Gets activist code id given code name
activist_code_id = None
for code in van.get_activist_codes():
if code['name'] == ACTIVIST_CODE_NAME:
activist_code_id = code['activistCodeId']
for participant in filtered_participants:
# generates list of parameters from matched columns, only inlcudes if row has data for column
params = {col: participant[col] for col in column_map.keys() if participant[col]}
van_person = van.upsert_person(**params) # updates if it finds a match, or inserts new user
if activist_code_id:
van.apply_activist_code(id=van_person['vanId'], activist_code_id=activist_code_id,
id_type="vanid")