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