def test_get_relationships(self):
api = PhishNetAPI()
relationships_response = api.get_relationships(1)
assert relationships_response['error_code'] == 0
assert relationships_response['response']['count'] == len(
relationships_response['response']['data'].keys())
def test_get_relationships(self, requests_mock):
api = PhishNetAPI('apikey123456789test1')
with open('tests/data/get_relationships.json') as f:
get_relationships_json = json.load(f)
requests_mock.post(api.base_url + "relationships/get",
json=get_relationships_json)
relationships_response = api.get_relationships(1)
assert relationships_response['response']['count'] == 2
assert len(relationships_response['response']['data'].keys()) == 2
assert len(
relationships_response['response']['data']["friends"].keys()) == 13
assert len(
relationships_response['response']['data']["fans"].keys()) == 13
