def setUp(self): super(TestFacebookClient, self).setUp() self.request = FacebookRequest(endpoint='me', method='GET') self.batch_request = FacebookBatchRequest(access_token='fake_token', requests=[self.request]) self.fake_response = FakeResponse( status_code=200, content= '{"data":[{"id":"123","name":"Foo"},{"id":"1337","name":"Bar"}]}', ) self.fake_batch_response = FakeResponse( status_code=200, content='[{"code":"123","body":"Foo"}]', ) self.client = FakeFacebookClient(fake_response=self.fake_response)
def test_send_batch_request(self): self.facebook.client = FakeFacebookClient(fake_response=FakeResponse( status_code=200, content='[{"code":"200","body":"Foo"}]', headers= '{"Content-Type": "application/json; charset=UTF-8","x-fb-trace-id": "1234"}' )) batches = [ self.facebook.request( method='GET', endpoint='some_endpoint', ) ] response = self.facebook.send_batch_request( requests=batches, access_token='foo_token', graph_version='v2.7', timeout=182, ) self.assertEqual(response.request.method, 'POST') self.assertEqual(response.request.access_token, 'foo_token') self.assertEqual(response.request.graph_version, 'v2.7') self.assertEqual(response.request.timeout, 182) self.assertDictEqual(response.request.params, {'access_token': 'foo_token'})
def test_delete_facebook_get_method(self): self.facebook.client = FakeFacebookClient(fake_response=FakeResponse( status_code=200, content='', )) response = self.facebook.delete(endpoint='/test', ) self.assertEqual(response.request.method, 'DELETE')
def test_post_facebook_get_method(self): self.facebook.client = FakeFacebookClient(fake_response=FakeResponse( status_code=200, content='', headers='', )) response = self.facebook.post(endpoint='/test', ) self.assertEqual(response.request.method, 'POST')
def test_send_request_error(self): self.client = FakeFacebookClient(fake_response=FakeResponse( status_code=400, content='{"error": {"code": 100}}'), ) self.assertRaises( FacebookResponseException, self.client.send_request, self.request, )
def setUp(self): super(TestFacebookClient, self).setUp() self.request = FacebookRequest(endpoint='me', method='GET') self.batch_request = FacebookBatchRequest( access_token='fake_token', requests=[self.request] ) self.fake_response = FakeResponse( status_code=200, content='{"data":[{"id":"123","name":"Foo"},{"id":"1337","name":"Bar"}]}', headers='{"Content-Type": "application/json; charset=UTF-8","x-fb-trace-id": "1234"}', ) self.fake_batch_response = FakeResponse( status_code=200, content='[{"code":"123","body":"Foo"}]', headers='{"Content-Type": "application/json; charset=UTF-8","x-fb-trace-id": "1234"}', ) self.client = FakeFacebookClient(fake_response=self.fake_response)
def test_send_request_error(self): self.client = FakeFacebookClient( fake_response=FakeResponse( status_code=400, content='{"error": {"code": 100}}', headers='{"Content-Type": "application/json; charset=UTF-8","x-fb-trace-id": "1234"}', ) ) with self.assertRaises(FacebookResponseException): self.client.send_request(self.request)
def test_send_request_default_access_token_and_version(self): self.facebook.client = FakeFacebookClient(fake_response=FakeResponse( status_code=200, content='', )) response = self.facebook.send_request( method='GET', endpoint='some_endpoint', ) self.assertEqual(response.request.method, 'GET') self.assertEqual(response.request.endpoint, 'some_endpoint') self.assertEqual(response.request.access_token, 'my_token') self.assertEqual(response.request.graph_version, 'v2.6') self.assertDictEqual(response.request.params, {'access_token': 'my_token'})
def test_send_request(self): self.facebook.client = FakeFacebookClient(fake_response=FakeResponse( status_code=200, content='', )) response = self.facebook.send_request(method='GET', endpoint='some_endpoint', access_token='foo_token', graph_version='v2.7') self.assertIsInstance(response, FacebookResponse) self.assertEqual(response.request.method, 'GET') self.assertEqual(response.request.endpoint, 'some_endpoint') self.assertEqual(response.request.access_token, 'foo_token') self.assertEqual(response.request.graph_version, 'v2.7') self.assertDictEqual(response.request.params, {'access_token': 'foo_token'})
def test_send_batch_request_default_access_token_and_version(self): self.facebook.client = FakeFacebookClient(fake_response=FakeResponse( status_code=200, content='[{"code":"200","body":"Foo"}]', )) batches = [ self.facebook.request( method='GET', endpoint='some_endpoint', ) ] response = self.facebook.send_batch_request(requests=batches, ) self.assertEqual(response.request.method, 'POST') self.assertEqual(response.request.access_token, 'my_token') self.assertEqual(response.request.graph_version, 'v2.6') self.assertDictEqual(response.request.params, {'access_token': 'my_token'})
def test_get_movies_empty_list(self, request): mock_data = [] request.return_value = FakeResponse(mock_data, HTTPStatus.OK) result = MoviesPeopleGetter._get_movies() self.assertEqual(result, {})
def test_get_movies_bad_request(self, request): request.return_value = FakeResponse(None, HTTPStatus.BAD_REQUEST) with self.assertRaises(InternalServerError, msg='Something went wrong') as e: MoviesPeopleGetter._get_movies()
def test_get_movies(self, request): request.return_value = FakeResponse(MOVIES_DATA, HTTPStatus.OK) result = MoviesPeopleGetter._get_movies() self.assertEqual(result, {m['id']: m for m in MOVIES_DATA})
def test_get_people(self, request): request.return_value = FakeResponse(PEOPLE_DATA, HTTPStatus.OK) result = MoviesPeopleGetter._get_people() self.assertEqual(result, PEOPLE_DATA)