def test_read_response_headers(self, mock_read):
headers = {_OPID_HEADER: "0", "_cid": "someid"}
mock_read.return_value = headers
context = FContext("someid")
self.protocol.read_response_headers(context)
self.assertEqual("0", context.get_response_header(_OPID_HEADER))
self.assertEqual("someid", context.get_response_header("_cid"))
def test_read_response_headers(self, mock_read):
headers = {_OPID_HEADER: "0", "_cid": "someid"}
mock_read.return_value = headers
context = FContext("someid")
self.protocol.read_response_headers(context)
# Ensure the opid is not set when the response headers are read in
self.assertIsNone(context.get_response_header(_OPID_HEADER))
self.assertEqual("someid", context.get_response_header("_cid"))
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"))
