Ejemplo n.º 1
0
class TestFacebookClient(TestCase):
    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_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)
        self.assertDictContainsSubset(
            {
                'method': 'GET',
                'url': 'https://graph.facebook.com/v2.5/me',
                'headers': {
                    'Content-Type': 'application/x-www-form-urlencoded'
                },
                'params': {
                    'access_token': 'fake_token'
                },
                'files': [],
                'timeout': 60,
                'data': None,
            }, self.client.send_kwargs)

    def test_send_request_override_timeout(self):
        self.client = FakeFacebookClient(
            fake_response=self.fake_response,
            request_timeout=10,
        )
        self.client.send_request(self.request)

        self.assertDictEqual(
            {
                'method': 'GET',
                'url': 'https://graph.facebook.com/v2.5/me',
                'headers': {
                    'Content-Type': 'application/x-www-form-urlencoded'
                },
                'params': {
                    'access_token': 'fake_token'
                },
                'files': [],
                'timeout': 10,
                'data': None,
            }, self.client.send_kwargs)

    def test_send_request_with_file(self):
        file_to_upload = FacebookFile(path='{base_path}/foo.txt'.format(
            base_path=os.path.dirname(os.path.abspath(__file__))), )
        request = FacebookRequest(
            endpoint='photos',
            method='POST',
            params={
                'message': 'Awesome Photo',
                'source': file_to_upload,
            },
            access_token='fake_token',
        )
        self.client = FakeFacebookClient(fake_response=self.fake_response, )
        self.client.send_request(request)
        self.assertDictEqual(
            {
                'method': 'POST',
                'url': 'https://graph.facebook.com/v2.5/photos',
                'headers': {},
                'params': {
                    'access_token': 'fake_token'
                },
                'files': [('source',
                           ('foo.txt', file_to_upload, 'text/plain'))],
                'timeout': 60,
                'data': {
                    'message': 'Awesome Photo'
                },
            }, self.client.send_kwargs)

    def test_send_request_error(self):
        self.client = FakeFacebookClient(fake_response=FakeResponse(
            status_code=400, content='{"error": {"code": 100}}'), )
        with 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)
        self.assertDictContainsSubset(
            {
                'method': 'POST',
                'url': 'https://graph.facebook.com/v2.5/',
                'headers': {
                    'Content-Type': 'application/x-www-form-urlencoded'
                },
                'params': {
                    'access_token': 'fake_token'
                },
                'files': [],
                'timeout': 60,
            }, self.client.send_kwargs)

    def test_send_empty_batch_request(self):
        with 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
        with self.assertRaises(FacebookSDKException):
            self.client.send_batch_request(batch_request=FacebookBatchRequest(
                access_token='fake_token',
                requests=requests,
            ), )
Ejemplo n.º 2
0
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,
            ),
        )