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()