ipc/ipc_sync_message_unittest.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4 //
   5 // Unit test to make sure that the serialization of synchronous IPC messages
   6 // works.  This ensures that the macros and templates were defined correctly.
   7 // Doesn't test the IPC channel mechanism.
   8
   9 #include <string.h>
  10
  11 #include "base/basictypes.h"
  12 #include "ipc/ipc_message.h"
  13 #include "ipc/ipc_message_utils.h"
  14 #include "base/logging.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16
  17 #define IPC_MESSAGE_IMPL
  18 #include "ipc/ipc_sync_message_unittest.h"
  19
  20 namespace {
  21
  22 static IPC::Message* g_reply;
  23
  24 class TestMessageReceiver {
  25  public:
  26
  27   void On_0_1(bool* out1) {
  28     *out1 = false;
  29   }
  30
  31   void On_0_2(bool* out1, int* out2) {
  32     *out1 = true;
  33     *out2 = 2;
  34   }
  35
  36   void On_0_3(bool* out1, int* out2, std::string* out3) {
  37     *out1 = false;
  38     *out2 = 3;
  39     *out3 = "0_3";
  40   }
  41
  42   void On_1_1(int in1, bool* out1) {
  43     DCHECK_EQ(1, in1);
  44     *out1 = true;
  45   }
  46
  47   void On_1_2(bool in1, bool* out1, int* out2) {
  48     DCHECK(!in1);
  49     *out1 = true;
  50     *out2 = 12;
  51   }
  52
  53   void On_1_3(int in1, std::string* out1, int* out2, bool* out3) {
  54     DCHECK_EQ(3, in1);
  55     *out1 = "1_3";
  56     *out2 = 13;
  57     *out3 = false;
  58   }
  59
  60   void On_2_1(int in1, bool in2, bool* out1) {
  61     DCHECK_EQ(1, in1);
  62     DCHECK(!in2);
  63     *out1 = true;
  64   }
  65
  66   void On_2_2(bool in1, int in2, bool* out1, int* out2) {
  67     DCHECK(!in1);
  68     DCHECK_EQ(2, in2);
  69     *out1 = true;
  70     *out2 = 22;
  71   }
  72
  73   void On_2_3(int in1, bool in2, std::string* out1, int* out2, bool* out3) {
  74     DCHECK_EQ(3, in1);
  75     DCHECK(in2);
  76     *out1 = "2_3";
  77     *out2 = 23;
  78     *out3 = false;
  79   }
  80
  81   void On_3_1(int in1, bool in2, std::string in3, bool* out1) {
  82     DCHECK_EQ(1, in1);
  83     DCHECK(!in2);
  84     DCHECK_EQ("3_1", in3);
  85     *out1 = true;
  86   }
  87
  88   void On_3_2(std::string in1, bool in2, int in3, bool* out1, int* out2) {
  89     DCHECK_EQ("3_2", in1);
  90     DCHECK(!in2);
  91     DCHECK_EQ(2, in3);
  92     *out1 = true;
  93     *out2 = 32;
  94   }
  95
  96   void On_3_3(int in1, std::string in2, bool in3, std::string* out1, int* out2,
  97               bool* out3) {
  98     DCHECK_EQ(3, in1);
  99     DCHECK_EQ("3_3", in2);
 100     DCHECK(in3);
 101     *out1 = "3_3";
 102     *out2 = 33;
 103     *out3 = false;
 104   }
 105
 106   void On_3_4(bool in1, int in2, std::string in3, int* out1, bool* out2,
 107               std::string* out3, bool* out4) {
 108     DCHECK(in1);
 109     DCHECK_EQ(3, in2);
 110     DCHECK_EQ("3_4", in3);
 111     *out1 = 34;
 112     *out2 = true;
 113     *out3 = "3_4";
 114     *out4 = false;
 115   }
 116
 117   bool Send(IPC::Message* message) {
 118     // gets the reply message, stash in global
 119     DCHECK(g_reply == NULL);
 120     g_reply = message;
 121     return true;
 122   }
 123
 124   bool OnMessageReceived(const IPC::Message& msg) {
 125     IPC_BEGIN_MESSAGE_MAP(TestMessageReceiver, msg)
 126       IPC_MESSAGE_HANDLER(Msg_C_0_1, On_0_1)
 127       IPC_MESSAGE_HANDLER(Msg_C_0_2, On_0_2)
 128       IPC_MESSAGE_HANDLER(Msg_C_0_3, On_0_3)
 129       IPC_MESSAGE_HANDLER(Msg_C_1_1, On_1_1)
 130       IPC_MESSAGE_HANDLER(Msg_C_1_2, On_1_2)
 131       IPC_MESSAGE_HANDLER(Msg_C_1_3, On_1_3)
 132       IPC_MESSAGE_HANDLER(Msg_C_2_1, On_2_1)
 133       IPC_MESSAGE_HANDLER(Msg_C_2_2, On_2_2)
 134       IPC_MESSAGE_HANDLER(Msg_C_2_3, On_2_3)
 135       IPC_MESSAGE_HANDLER(Msg_C_3_1, On_3_1)
 136       IPC_MESSAGE_HANDLER(Msg_C_3_2, On_3_2)
 137       IPC_MESSAGE_HANDLER(Msg_C_3_3, On_3_3)
 138       IPC_MESSAGE_HANDLER(Msg_C_3_4, On_3_4)
 139       IPC_MESSAGE_HANDLER(Msg_R_0_1, On_0_1)
 140       IPC_MESSAGE_HANDLER(Msg_R_0_2, On_0_2)
 141       IPC_MESSAGE_HANDLER(Msg_R_0_3, On_0_3)
 142       IPC_MESSAGE_HANDLER(Msg_R_1_1, On_1_1)
 143       IPC_MESSAGE_HANDLER(Msg_R_1_2, On_1_2)
 144       IPC_MESSAGE_HANDLER(Msg_R_1_3, On_1_3)
 145       IPC_MESSAGE_HANDLER(Msg_R_2_1, On_2_1)
 146       IPC_MESSAGE_HANDLER(Msg_R_2_2, On_2_2)
 147       IPC_MESSAGE_HANDLER(Msg_R_2_3, On_2_3)
 148       IPC_MESSAGE_HANDLER(Msg_R_3_1, On_3_1)
 149       IPC_MESSAGE_HANDLER(Msg_R_3_2, On_3_2)
 150       IPC_MESSAGE_HANDLER(Msg_R_3_3, On_3_3)
 151       IPC_MESSAGE_HANDLER(Msg_R_3_4, On_3_4)
 152     IPC_END_MESSAGE_MAP()
 153     return true;
 154   }
 155
 156 };
 157
 158 void Send(IPC::SyncMessage* msg) {
 159   static TestMessageReceiver receiver;
 160
 161   IPC::MessageReplyDeserializer* reply_serializer = msg->GetReplyDeserializer();
 162   DCHECK(reply_serializer != NULL);
 163
 164   // "send" the message
 165   receiver.OnMessageReceived(*msg);
 166   delete msg;
 167
 168   // get the reply message from the global, and deserialize the output
 169   // parameters into the output pointers.
 170   DCHECK(g_reply != NULL);
 171   bool result = reply_serializer->SerializeOutputParameters(*g_reply);
 172   DCHECK(result);
 173   delete g_reply;
 174   g_reply = NULL;
 175   delete reply_serializer;
 176 }
 177
 178 TEST(IPCSyncMessageTest, Main) {
 179   bool bool1 = true;
 180   int int1 = 0;
 181   std::string string1;
 182
 183   Send(new Msg_C_0_1(&bool1));
 184   DCHECK(!bool1);
 185
 186   Send(new Msg_C_0_2(&bool1, &int1));
 187   DCHECK(bool1);
 188   DCHECK_EQ(2, int1);
 189
 190   Send(new Msg_C_0_3(&bool1, &int1, &string1));
 191   DCHECK(!bool1);
 192   DCHECK_EQ(3, int1);
 193   DCHECK_EQ("0_3", string1);
 194
 195   bool1 = false;
 196   Send(new Msg_C_1_1(1, &bool1));
 197   DCHECK(bool1);
 198
 199   bool1 = false;
 200   Send(new Msg_C_1_2(false, &bool1, &int1));
 201   DCHECK(bool1);
 202   DCHECK_EQ(12, int1);
 203
 204   bool1 = true;
 205   Send(new Msg_C_1_3(3, &string1, &int1, &bool1));
 206   DCHECK_EQ("1_3", string1);
 207   DCHECK_EQ(13, int1);
 208   DCHECK(!bool1);
 209
 210   bool1 = false;
 211   Send(new Msg_C_2_1(1, false, &bool1));
 212   DCHECK(bool1);
 213
 214   bool1 = false;
 215   Send(new Msg_C_2_2(false, 2, &bool1, &int1));
 216   DCHECK(bool1);
 217   DCHECK_EQ(22, int1);
 218
 219   bool1 = true;
 220   Send(new Msg_C_2_3(3, true, &string1, &int1, &bool1));
 221   DCHECK_EQ("2_3", string1);
 222   DCHECK_EQ(23, int1);
 223   DCHECK(!bool1);
 224
 225   bool1 = false;
 226   Send(new Msg_C_3_1(1, false, "3_1", &bool1));
 227   DCHECK(bool1);
 228
 229   bool1 = false;
 230   Send(new Msg_C_3_2("3_2", false, 2, &bool1, &int1));
 231   DCHECK(bool1);
 232   DCHECK_EQ(32, int1);
 233
 234   bool1 = true;
 235   Send(new Msg_C_3_3(3, "3_3", true, &string1, &int1, &bool1));
 236   DCHECK_EQ("3_3", string1);
 237   DCHECK_EQ(33, int1);
 238   DCHECK(!bool1);
 239
 240   bool1 = false;
 241   bool bool2 = true;
 242   Send(new Msg_C_3_4(true, 3, "3_4", &int1, &bool1, &string1, &bool2));
 243   DCHECK_EQ(34, int1);
 244   DCHECK(bool1);
 245   DCHECK_EQ("3_4", string1);
 246   DCHECK(!bool2);
 247
 248   // Routed messages, just a copy of the above but with extra routing paramater
 249   Send(new Msg_R_0_1(0, &bool1));
 250   DCHECK(!bool1);
 251
 252   Send(new Msg_R_0_2(0, &bool1, &int1));
 253   DCHECK(bool1);
 254   DCHECK_EQ(2, int1);
 255
 256   Send(new Msg_R_0_3(0, &bool1, &int1, &string1));
 257   DCHECK(!bool1);
 258   DCHECK_EQ(3, int1);
 259   DCHECK_EQ("0_3", string1);
 260
 261   bool1 = false;
 262   Send(new Msg_R_1_1(0, 1, &bool1));
 263   DCHECK(bool1);
 264
 265   bool1 = false;
 266   Send(new Msg_R_1_2(0, false, &bool1, &int1));
 267   DCHECK(bool1);
 268   DCHECK_EQ(12, int1);
 269
 270   bool1 = true;
 271   Send(new Msg_R_1_3(0, 3, &string1, &int1, &bool1));
 272   DCHECK_EQ("1_3", string1);
 273   DCHECK_EQ(13, int1);
 274   DCHECK(!bool1);
 275
 276   bool1 = false;
 277   Send(new Msg_R_2_1(0, 1, false, &bool1));
 278   DCHECK(bool1);
 279
 280   bool1 = false;
 281   Send(new Msg_R_2_2(0, false, 2, &bool1, &int1));
 282   DCHECK(bool1);
 283   DCHECK_EQ(22, int1);
 284
 285   bool1 = true;
 286   Send(new Msg_R_2_3(0, 3, true, &string1, &int1, &bool1));
 287   DCHECK(!bool1);
 288   DCHECK_EQ("2_3", string1);
 289   DCHECK_EQ(23, int1);
 290
 291   bool1 = false;
 292   Send(new Msg_R_3_1(0, 1, false, "3_1", &bool1));
 293   DCHECK(bool1);
 294
 295   bool1 = false;
 296   Send(new Msg_R_3_2(0, "3_2", false, 2, &bool1, &int1));
 297   DCHECK(bool1);
 298   DCHECK_EQ(32, int1);
 299
 300   bool1 = true;
 301   Send(new Msg_R_3_3(0, 3, "3_3", true, &string1, &int1, &bool1));
 302   DCHECK_EQ("3_3", string1);
 303   DCHECK_EQ(33, int1);
 304   DCHECK(!bool1);
 305 }
 306
 307 }  // namespace