def test_get_all_people_types(self):
api = PhishNetAPI()
people_types_response = api.get_all_people_types()
assert people_types_response['error_code'] == 0
assert people_types_response['response']['count'] == len(
people_types_response['response']['data'].keys())
assert people_types_response['response']['data']["1"] == "The Band"
def test_get_all_people_types(self, requests_mock):
api = PhishNetAPI('apikey123456789test1')
with open('tests/data/all_people_types.json') as f:
all_people_types_json = json.load(f)
requests_mock.post(api.base_url + "people/get",
json=all_people_types_json)
people_types_response = api.get_all_people_types()
assert people_types_response['response']['count'] == 9
assert len(people_types_response['response']['data'].keys()) == 9
assert people_types_response['response']['data']["1"] == "The Band"
