示例#1
0
    def test_get_upcoming_shows(self):
        api = PhishNetAPI()
        upcoming_shows_response = api.get_upcoming_shows()

        assert upcoming_shows_response['error_code'] == 0
        assert upcoming_shows_response['response']['count'] == len(
            upcoming_shows_response['response']['data'])
示例#2
0
    def test_get_upcoming_shows(self, requests_mock):
        api = PhishNetAPI('apikey123456789test1')
        with open('tests/data/get_upcoming_shows.json') as f:
            get_upcoming_shows_json = json.load(f)
        requests_mock.post(api.base_url + "shows/upcoming",
                           json=get_upcoming_shows_json)

        upcoming_shows_response = api.get_upcoming_shows()
        assert upcoming_shows_response['response']['count'] == 17
        assert len(upcoming_shows_response['response']['data']) == 17
        assert upcoming_shows_response['response']['data'][0][
            'showid'] == 1470182793