Python CustomHookPolicy.on_request 예제들

예제 #1

def test_custom_hook_policy_set_hook():
    def test_callback(response):
        raise ValueError()
    custom_hook_policy = CustomHookPolicy()
    request = HttpRequest('GET', 'http://127.0.0.1/')
    pipeline_request = PipelineRequest(request, PipelineContext(None, raw_response_hook=test_callback))
    custom_hook_policy.on_request(pipeline_request)
    assert custom_hook_policy._callback is test_callback

예제 #2

def test_custom_hook_policy_invoke_hook():
    def test_callback(response):
        raise ValueError()

    custom_hook_policy = CustomHookPolicy()
    request = HttpRequest('GET', 'http://127.0.0.1/')
    pipeline_request = PipelineRequest(request, PipelineContext(None, raw_response_hook=test_callback))
    custom_hook_policy.on_request(pipeline_request)
    with pytest.raises(ValueError):
        custom_hook_policy.on_response(pipeline_request, None)