Esempio n. 1
0
    def test_call_stream_pass(self, m):
        """Ensure that successful HTTP streaming response codes pass."""
        with mock.patch.object(Client, 'get_credentials', return_value=True):
            c = Client(("http://mydomain.com", "http://mystreamingdomain.com"),
                       "my_account", "my_token")
            c.session = requests.Session()

        for status_code in [200, 201, 202, 301, 302]:
            m.get(requests_mock.ANY, status_code=status_code)
            c._Client__call_stream(uri="http://example.com",
                                   params={"test": "test"},
                                   method="get")

            m.post(requests_mock.ANY, status_code=status_code)
            c._Client__call_stream(uri="http://example.com",
                                   params={"test": "test"},
                                   method="post")
Esempio n. 2
0
    def test_call_stream_fail(self, m):
        """Ensure that failure HTTP streaming response codes fail."""
        with mock.patch.object(Client, 'get_credentials', return_value=True):
            c = Client(("http://mydomain.com", "http://mystreamingdomain.com"),
                       "my_account", "my_token")
            c.session = requests.Session()

        status_codes = [400, 401, 403, 404, 500]
        caught = 0
        for status_code in status_codes:
            m.get(requests_mock.ANY,
                  text=json.dumps({'message': 'test'}),
                  status_code=400)
            try:
                c._Client__call_stream(uri="http://example.com",
                                       params={"test": "test"},
                                       method="get")
            except BadRequest:
                caught += 1
        self.assertEqual(len(status_codes), caught)