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