Exemple #1
0
 def setUp(self):
     super(TestMockClient, self).setUp()
     self.mock_client = MockClient(self.io_loop)
Exemple #2
0
 def setUp(self):
     super(TestMockClient, self).setUp()
     self.mock_client = MockClient(self.io_loop)
Exemple #3
0
class TestMockClient(AsyncTestCase):
    def setUp(self):
        super(TestMockClient, self).setUp()
        self.mock_client = MockClient(self.io_loop)

    @gen_test
    def test_patches_known_hosts(self):
        self.mock_client.mock_url("http://foo.com/bar")
        with self.mock_client.patch():
            client = AsyncHTTPClient()
            response = yield client.fetch("http://foo.com/bar")

        self.assertEqual(200, response.code)
        self.assertEqual("", response.body)
        self.assertEqual(0, response.headers["Content-Length"])

    @gen_test
    def test_patches_allow_overwrite(self):
        response = self.mock_client.mock_url("http://foo.com/bar")
        response.code = 204
        response.body = "BODY"
        response.headers["X-Test"] = "foobar"

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            fetched = yield client.fetch("http://foo.com/bar")

        self.assertEqual(204, fetched.code)
        self.assertEqual("BODY", fetched.body)
        self.assertEqual("foobar", fetched.headers["x-test"])

    @gen_test
    def test_patch_respects_method(self):
        self.mock_client.mock_url("http://foo.com/bar", method="POST")

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            get_response = yield client.fetch("http://foo.com/bar",
                                              method="GET",
                                              raise_error=False)

            post_response = yield client.fetch("http://foo.com/bar",
                                               method="POST")

        self.assertEqual(405, get_response.code)
        self.assertEqual(200, post_response.code)

    @gen_test
    def test_patches_raises_error(self):
        response = self.mock_client.mock_url("http://foo.com/bar")
        response.code = 500
        response.body = "ERROR"

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            with self.assertRaises(HTTPError) as exc_context:
                yield client.fetch("http://foo.com/bar")

        exception = exc_context.exception
        self.assertEqual(500, exception.code)
        self.assertEqual(5, exception.response.headers["Content-length"])

    @gen_test
    def test_patches_respects_raise_error_flag(self):
        response = self.mock_client.mock_url("http://foo.com/bar")
        response.code = 500

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            fetched_response = yield client.fetch("http://foo.com/bar",
                                                  raise_error=False)

        self.assertEqual(500, fetched_response.code)

    @gen_test
    def test_patch_passes_through_unknown_requests(self):
        class Handler(RequestHandler):
            def get(self):
                self.finish("REAL RESPONSE")

        socket, port = bind_unused_port()
        app = Application([("/", Handler)])
        server = HTTPServer(app)
        server.add_sockets([socket])
        with self.mock_client.patch():
            client = AsyncHTTPClient()
            response = yield client.fetch("http://localhost:{}/".format(port))

        self.assertEqual(200, response.code)
        self.assertEqual("REAL RESPONSE", response.body.decode("utf-8"))

    @gen_test
    def test_patch_ignores_query_parameters(self):
        response = self.mock_client.mock_url("http://foo.com/bar")

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            fetched_response = yield client.fetch("http://foo.com/bar?q=baz")

        self.assertEqual(fetched_response, response)

    @gen_test
    def test_patch_requires_multiple_mocks_for_multiple_requests(self):
        self.mock_client.mock_url("http://foo.com/one")
        # put two requests in for a second path
        self.mock_client.mock_url("http://foo.com/two")
        self.mock_client.mock_url("http://foo.com/two")

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            yield client.fetch("http://foo.com/one")
            with self.assertRaises(MissingMockResponse):
                yield client.fetch("http://foo.com/one")

            yield client.fetch("http://foo.com/two")
            yield client.fetch("http://foo.com/two")
Exemple #4
0
class TestMockClient(AsyncTestCase):

    def setUp(self):
        super(TestMockClient, self).setUp()
        self.mock_client = MockClient(self.io_loop)

    @gen_test
    def test_patches_known_hosts(self):
        self.mock_client.mock_url("http://foo.com/bar")
        with self.mock_client.patch():
            client = AsyncHTTPClient()
            response = yield client.fetch("http://foo.com/bar")

        self.assertEqual(200, response.code)
        self.assertEqual("", response.body)
        self.assertEqual(0, response.headers["Content-Length"])

    @gen_test
    def test_patches_allow_overwrite(self):
        response = self.mock_client.mock_url("http://foo.com/bar")
        response.code = 204
        response.body = "BODY"
        response.headers["X-Test"] = "foobar"

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            fetched = yield client.fetch("http://foo.com/bar")

        self.assertEqual(204, fetched.code)
        self.assertEqual("BODY", fetched.body)
        self.assertEqual("foobar", fetched.headers["x-test"])

    @gen_test
    def test_patch_respects_method(self):
        self.mock_client.mock_url("http://foo.com/bar", method="POST")

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            get_response = yield client.fetch(
                "http://foo.com/bar", method="GET", raise_error=False)

            post_response = yield client.fetch(
                "http://foo.com/bar", method="POST")

        self.assertEqual(405, get_response.code)
        self.assertEqual(200, post_response.code)

    @gen_test
    def test_patches_raises_error(self):
        response = self.mock_client.mock_url("http://foo.com/bar")
        response.code = 500
        response.body = "ERROR"

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            with self.assertRaises(HTTPError) as exc_context:
                yield client.fetch("http://foo.com/bar")

        exception = exc_context.exception
        self.assertEqual(500, exception.code)
        self.assertEqual(5, exception.response.headers["Content-length"])

    @gen_test
    def test_patches_respects_raise_error_flag(self):
        response = self.mock_client.mock_url("http://foo.com/bar")
        response.code = 500

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            fetched_response = yield client.fetch(
                "http://foo.com/bar", raise_error=False)

        self.assertEqual(500, fetched_response.code)

    @gen_test
    def test_patch_passes_through_unknown_requests(self):

        class Handler(RequestHandler):
            def get(self):
                self.finish("REAL RESPONSE")

        port = get_unused_port()
        app = Application([("/", Handler)])
        app.listen(port)
        with self.mock_client.patch():
            client = AsyncHTTPClient()
            response = yield client.fetch("http://localhost:{}/".format(port))

        self.assertEqual(200, response.code)
        self.assertEqual("REAL RESPONSE", response.body.decode("utf-8"))

    @gen_test
    def test_patch_ignores_query_parameters(self):
        response = self.mock_client.mock_url("http://foo.com/bar")

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            fetched_response = yield client.fetch("http://foo.com/bar?q=baz")

        self.assertEqual(fetched_response, response)

    @gen_test
    def test_patch_requires_multiple_mocks_for_multiple_requests(self):
        self.mock_client.mock_url("http://foo.com/one")
        # put two requests in for a second path
        self.mock_client.mock_url("http://foo.com/two")
        self.mock_client.mock_url("http://foo.com/two")

        with self.mock_client.patch():
            client = AsyncHTTPClient()
            yield client.fetch("http://foo.com/one")
            with self.assertRaises(MissingMockResponse):
                yield client.fetch("http://foo.com/one")

            yield client.fetch("http://foo.com/two")
            yield client.fetch("http://foo.com/two")