当前位置: 首页>>代码示例>>Python>>正文


Python ClientNetwork._check_response方法代码示例

本文整理汇总了Python中acme.client.ClientNetwork._check_response方法的典型用法代码示例。如果您正苦于以下问题:Python ClientNetwork._check_response方法的具体用法?Python ClientNetwork._check_response怎么用?Python ClientNetwork._check_response使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在acme.client.ClientNetwork的用法示例。


在下文中一共展示了ClientNetwork._check_response方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。

示例1: ClientNetworkTest

# 需要导入模块: from acme.client import ClientNetwork [as 别名]
# 或者: from acme.client.ClientNetwork import _check_response [as 别名]
class ClientNetworkTest(unittest.TestCase):
    """Tests for acme.client.ClientNetwork."""

    def setUp(self):
        self.verify_ssl = mock.MagicMock()
        self.wrap_in_jws = mock.MagicMock(return_value=mock.sentinel.wrapped)

        from acme.client import ClientNetwork
        self.net = ClientNetwork(
            key=KEY, alg=jose.RS256, verify_ssl=self.verify_ssl,
            user_agent='acme-python-test')

        self.response = mock.MagicMock(ok=True, status_code=http_client.OK)
        self.response.headers = {}
        self.response.links = {}

    def test_init(self):
        self.assertTrue(self.net.verify_ssl is self.verify_ssl)

    def test_wrap_in_jws(self):
        class MockJSONDeSerializable(jose.JSONDeSerializable):
            # pylint: disable=missing-docstring
            def __init__(self, value):
                self.value = value

            def to_partial_json(self):
                return {'foo': self.value}

            @classmethod
            def from_json(cls, value):
                pass  # pragma: no cover

        # pylint: disable=protected-access
        jws_dump = self.net._wrap_in_jws(
            MockJSONDeSerializable('foo'), nonce=b'Tg')
        jws = acme_jws.JWS.json_loads(jws_dump)
        self.assertEqual(json.loads(jws.payload.decode()), {'foo': 'foo'})
        self.assertEqual(jws.signature.combined.nonce, b'Tg')

    def test_check_response_not_ok_jobj_no_error(self):
        self.response.ok = False
        self.response.json.return_value = {}
        with mock.patch('acme.client.messages.Error.from_json') as from_json:
            from_json.side_effect = jose.DeserializationError
            # pylint: disable=protected-access
            self.assertRaises(
                errors.ClientError, self.net._check_response, self.response)

    def test_check_response_not_ok_jobj_error(self):
        self.response.ok = False
        self.response.json.return_value = messages.Error(
            detail='foo', typ='serverInternal', title='some title').to_json()
        # pylint: disable=protected-access
        self.assertRaises(
            messages.Error, self.net._check_response, self.response)

    def test_check_response_not_ok_no_jobj(self):
        self.response.ok = False
        self.response.json.side_effect = ValueError
        # pylint: disable=protected-access
        self.assertRaises(
            errors.ClientError, self.net._check_response, self.response)

    def test_check_response_ok_no_jobj_ct_required(self):
        self.response.json.side_effect = ValueError
        for response_ct in [self.net.JSON_CONTENT_TYPE, 'foo']:
            self.response.headers['Content-Type'] = response_ct
            # pylint: disable=protected-access
            self.assertRaises(
                errors.ClientError, self.net._check_response, self.response,
                content_type=self.net.JSON_CONTENT_TYPE)

    def test_check_response_ok_no_jobj_no_ct(self):
        self.response.json.side_effect = ValueError
        for response_ct in [self.net.JSON_CONTENT_TYPE, 'foo']:
            self.response.headers['Content-Type'] = response_ct
            # pylint: disable=protected-access,no-value-for-parameter
            self.assertEqual(
                self.response, self.net._check_response(self.response))

    def test_check_response_jobj(self):
        self.response.json.return_value = {}
        for response_ct in [self.net.JSON_CONTENT_TYPE, 'foo']:
            self.response.headers['Content-Type'] = response_ct
            # pylint: disable=protected-access,no-value-for-parameter
            self.assertEqual(
                self.response, self.net._check_response(self.response))

    @mock.patch('acme.client.requests')
    def test_send_request(self, mock_requests):
        mock_requests.request.return_value = self.response
        # pylint: disable=protected-access
        self.assertEqual(self.response, self.net._send_request(
            'HEAD', 'url', 'foo', bar='baz'))
        mock_requests.request.assert_called_once_with(
            'HEAD', 'url', 'foo', verify=mock.ANY, bar='baz', headers=mock.ANY)

    @mock.patch('acme.client.requests')
    def test_send_request_verify_ssl(self, mock_requests):
        # pylint: disable=protected-access
#.........这里部分代码省略.........
开发者ID:bad194,项目名称:certbot,代码行数:103,代码来源:client_test.py

示例2: ClientNetworkTest

# 需要导入模块: from acme.client import ClientNetwork [as 别名]
# 或者: from acme.client.ClientNetwork import _check_response [as 别名]
class ClientNetworkTest(unittest.TestCase):
    """Tests for acme.client.ClientNetwork."""

    def setUp(self):
        self.verify_ssl = mock.MagicMock()
        self.wrap_in_jws = mock.MagicMock(return_value=mock.sentinel.wrapped)

        from acme.client import ClientNetwork

        self.net = ClientNetwork(key=KEY, alg=jose.RS256, verify_ssl=self.verify_ssl)

        self.response = mock.MagicMock(ok=True, status_code=http_client.OK)
        self.response.headers = {}
        self.response.links = {}

    def test_init(self):
        self.assertTrue(self.net.verify_ssl is self.verify_ssl)

    def test_wrap_in_jws(self):
        class MockClientRequestableResource(jose.JSONDeSerializable):
            # pylint: disable=missing-docstring
            resource_type = "mock"

            def __init__(self, value):
                self.value = value

            def to_partial_json(self):
                return {"foo": self.value}

            @classmethod
            def from_json(cls, value):
                pass  # pragma: no cover

        # pylint: disable=protected-access
        jws_dump = self.net._wrap_in_jws(MockClientRequestableResource("foo"), nonce=b"Tg")
        jws = acme_jws.JWS.json_loads(jws_dump)
        self.assertEqual(json.loads(jws.payload.decode()), {"foo": "foo", "resource": "mock"})
        self.assertEqual(jws.signature.combined.nonce, b"Tg")

    def test_check_response_not_ok_jobj_no_error(self):
        self.response.ok = False
        self.response.json.return_value = {}
        # pylint: disable=protected-access
        self.assertRaises(errors.ClientError, self.net._check_response, self.response)

    def test_check_response_not_ok_jobj_error(self):
        self.response.ok = False
        self.response.json.return_value = messages.Error(
            detail="foo", typ="serverInternal", title="some title"
        ).to_json()
        # pylint: disable=protected-access
        self.assertRaises(messages.Error, self.net._check_response, self.response)

    def test_check_response_not_ok_no_jobj(self):
        self.response.ok = False
        self.response.json.side_effect = ValueError
        # pylint: disable=protected-access
        self.assertRaises(errors.ClientError, self.net._check_response, self.response)

    def test_check_response_ok_no_jobj_ct_required(self):
        self.response.json.side_effect = ValueError
        for response_ct in [self.net.JSON_CONTENT_TYPE, "foo"]:
            self.response.headers["Content-Type"] = response_ct
            # pylint: disable=protected-access
            self.assertRaises(
                errors.ClientError, self.net._check_response, self.response, content_type=self.net.JSON_CONTENT_TYPE
            )

    def test_check_response_ok_no_jobj_no_ct(self):
        self.response.json.side_effect = ValueError
        for response_ct in [self.net.JSON_CONTENT_TYPE, "foo"]:
            self.response.headers["Content-Type"] = response_ct
            # pylint: disable=protected-access,no-value-for-parameter
            self.assertEqual(self.response, self.net._check_response(self.response))

    def test_check_response_jobj(self):
        self.response.json.return_value = {}
        for response_ct in [self.net.JSON_CONTENT_TYPE, "foo"]:
            self.response.headers["Content-Type"] = response_ct
            # pylint: disable=protected-access,no-value-for-parameter
            self.assertEqual(self.response, self.net._check_response(self.response))

    @mock.patch("acme.client.requests")
    def test_send_request(self, mock_requests):
        mock_requests.request.return_value = self.response
        # pylint: disable=protected-access
        self.assertEqual(self.response, self.net._send_request("HEAD", "url", "foo", bar="baz"))
        mock_requests.request.assert_called_once_with("HEAD", "url", "foo", verify=mock.ANY, bar="baz")

    @mock.patch("acme.client.requests")
    def test_send_request_verify_ssl(self, mock_requests):
        # pylint: disable=protected-access
        for verify in True, False:
            mock_requests.request.reset_mock()
            mock_requests.request.return_value = self.response
            self.net.verify_ssl = verify
            # pylint: disable=protected-access
            self.assertEqual(self.response, self.net._send_request("GET", "url"))
            mock_requests.request.assert_called_once_with("GET", "url", verify=verify)

#.........这里部分代码省略.........
开发者ID:Jonadabe,项目名称:letsencrypt,代码行数:103,代码来源:client_test.py


注:本文中的acme.client.ClientNetwork._check_response方法示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。