def test_update_show_attendance(self):
api = PhishNetAPI()
api.uid = os.getenv('UID')
update_attendance_response = api.update_show_attendance(
api.uid, showid=1252691618, update='add')
assert update_attendance_response['error_code'] == 0
assert update_attendance_response[
'error_message'] == "Successfully added 1997-11-30"
assert update_attendance_response['response']['data'][
'action'] == 'add'
assert update_attendance_response['response']['data'][
'showdate'] == '1997-11-30'
assert update_attendance_response['response']['data'][
'showid'] == 1252691618
shows_seen_after_add = int(
update_attendance_response['response']['data']['shows_seen'])
update_attendance_response = api.update_show_attendance(
api.uid, showid=1252691618, update='remove')
assert update_attendance_response['error_code'] == 0
assert update_attendance_response[
'error_message'] == "Successfully removed 1997-11-30"
shows_seen_after_remove = int(
update_attendance_response['response']['data']['shows_seen'])
assert shows_seen_after_add > shows_seen_after_remove
assert update_attendance_response['response']['data'][
'action'] == 'remove'
assert update_attendance_response['response']['data'][
'showdate'] == '1997-11-30'
assert update_attendance_response['response']['data'][
'showid'] == 1252691618
def test_update_show_attendance(self, requests_mock):
api = PhishNetAPI('apikey123456789test1')
api.authkey = 'B6386A2485D94C73DAA'
api.uid = 80
with open('tests/data/add_attendance.json') as f:
add_attendance_json = json.load(f)
requests_mock.post(api.base_url + "attendance/add",
json=add_attendance_json)
update_attendance_response = api.update_show_attendance(
api.uid, showid=1252691618, update='add')
assert update_attendance_response[
'error_message'] == "Successfully added 1997-11-30"
assert update_attendance_response['response']['data'][
'action'] == 'add'
assert update_attendance_response['response']['data'][
'showdate'] == '1997-11-30'
assert update_attendance_response['response']['data'][
'showid'] == 1252691618
shows_seen = update_attendance_response['response']['data'][
'shows_seen']
assert shows_seen == '63'
with open('tests/data/remove_attendance.json') as f:
remove_attendance_json = json.load(f)
requests_mock.post(api.base_url + "attendance/remove",
json=remove_attendance_json)
update_attendance_response = api.update_show_attendance(
api.uid, showid=1252691618, update='remove')
assert update_attendance_response[
'error_message'] == "Successfully removed 1997-11-30"
assert update_attendance_response['response']['data'][
'action'] == 'remove'
assert update_attendance_response['response']['data'][
'showdate'] == '1997-11-30'
assert update_attendance_response['response']['data'][
'showid'] == 1252691618
assert update_attendance_response['response']['data'][
'shows_seen'] == str(int(shows_seen) - 1)