def test_timeout(self):
# Check default timeout (5 seconds).
context = FContext()
self.assertEqual(5000, context.timeout)
self.assertEqual("5000", context.get_request_header("_timeout"))
# Set timeout and check expected values.
context.set_timeout(10000)
self.assertEqual(10000, context.timeout)
self.assertEqual("10000", context.get_request_header("_timeout"))
# Check timeout passed to constructor.
context = FContext(timeout=1000)
self.assertEqual(1000, context.timeout)
self.assertEqual("1000", context.get_request_header("_timeout"))
def test_cant_set_opid_public_method(self):
context = FContext(self.correlation_id)
context.set_request_header("_opid", "foo")
self.assertNotEqual(context.get_request_header("_opid"), "foo")
def test_response_header(self):
context = FContext(self.correlation_id)
self.assertEqual(context, context.set_response_header("foo", "bar"))
self.assertEqual("bar", context.get_response_header("foo"))
self.assertEqual(self.correlation_id,
context.get_request_header("_cid"))
def test_op_id(self):
context = FContext(self.correlation_id)
context._set_request_header("_opid", "12345")
self.assertEqual(self.correlation_id, context.correlation_id)
self.assertEqual("12345", context.get_request_header("_opid"))
