def test_get_all_venues(self): api = PhishNetAPI() venues_response = api.get_all_venues() assert venues_response['error_code'] == 0 assert venues_response['response']['count'] == len( venues_response['response']['data']) assert venues_response['response']['data']['1']['venueid'] == 1
def test_get_all_venues(self, requests_mock): api = PhishNetAPI('apikey123456789test1') with open('tests/data/all_venues.json') as f: all_venues_json = json.load(f) requests_mock.post(api.base_url + "venues/all", json=all_venues_json) venues_response = api.get_all_venues() assert venues_response['response']['count'] == 3 assert len(venues_response['response']['data']) == 3 assert venues_response['response']['data']['1']['venueid'] == 1