class TestTimeout(ArgumentTestCase, FuncDecoratorTestCase):
type_cls = arguments.Timeout
expected_converter_key = keys.Identity()
def test_modify_request(self, request_builder):
arguments.Timeout().modify_request(request_builder, 10)
assert request_builder.info["timeout"] == 10
class TestContext(ArgumentTestCase, FuncDecoratorTestCase):
type_cls = arguments.Context
expected_converter_key = keys.Identity()
def test_modify_request(self, request_builder):
arguments.Context("key").modify_request(request_builder, "value")
assert request_builder.context["key"] == "value"
class TestContextMap(ArgumentTestCase, FuncDecoratorTestCase):
type_cls = arguments.ContextMap
expected_converter_key = keys.Identity()
def test_modify_request(self, request_builder):
arguments.ContextMap().modify_request(request_builder,
{"key": "value"})
assert request_builder.context == {"key": "value"}
def test_modify_request_not_mapping(self, request_builder):
with pytest.raises(TypeError):
arguments.ContextMap().modify_request(request_builder, "value")
def converter_key(self):
"""Do not convert passed argument."""
return keys.Identity()
