class TestFacebookClient(TestCase):
def setUp(self):
super(TestFacebookClient, self).setUp()
self.request = FacebookRequest(endpoint='')
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_prepare_request(self):
request_params = self.client._prepareRequest(request=self.request)
self.assertEqual(request_params.get('method'), self.request.method)
self.assertEqual(request_params.get('url'),
'%s%s' % (BASE_GRAPH_URL, self.request.url))
self.assertEqual(request_params.get('params'), self.request.params)
self.assertIn('Content-Type', request_params.get('headers'))
self.assertEqual(
request_params.get('headers').get('Content-Type'),
'application/x-www-form-urlencoded')
def test_send_request(self):
response = self.client.send_request(self.request)
self.assertIsInstance(response, FacebookResponse)
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 test_send_batch_request(self):
self.client = FakeFacebookClient(
fake_response=self.fake_batch_response, )
response = self.client.send_batch_request(
batch_request=self.batch_request, )
self.assertIsInstance(response, FacebookBatchResponse)
def test_send_empty_batch_request(self):
self.assertRaises(
FacebookSDKException,
self.client.send_batch_request,
batch_request=FacebookBatchRequest(access_token='fake_token', ),
)
def test_send_over_limit_batch_request(self):
requests = [self.request] * 51
self.assertRaises(
FacebookSDKException,
self.client.send_batch_request,
batch_request=FacebookBatchRequest(
access_token='fake_token',
requests=requests,
),
)
