def test_write_response_headers(self, mock_write):
context = FContext("foo")
mock_write.return_value = "bar"
mock_trans = mock.Mock()
self.protocol.trans = mock_trans
self.protocol.write_response_headers(context)
mock_write.assert_called_with(context.get_response_headers())
mock_trans.write.assert_called_with("bar")
def test_response_headers(self):
context = FContext(self.correlation_id)
context.set_response_header("foo", "bar")
headers = context.get_response_headers()
self.assertEqual("bar", headers.get('foo'))
