Esempio n. 1
0
    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
Esempio n. 2
0
    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