Used oneof in blimp_message.proto

blimp/net/blimp_connection_unittest.cc

Index: blimp/net/blimp_connection_unittest.cc
diff --git a/blimp/net/blimp_connection_unittest.cc b/blimp/net/blimp_connection_unittest.cc
index 59c7a39a370f218e2a17c58283f87dd14d19822b..ee04e92c86508d450bb281e42448bb6f5963dc17 100644
--- a/blimp/net/blimp_connection_unittest.cc
+++ b/blimp/net/blimp_connection_unittest.cc
@@ -11,6 +11,7 @@
 #include "base/callback_helpers.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
+#include "blimp/common/create_blimp_message.h"
 #include "blimp/common/proto/blimp_message.pb.h"
 #include "blimp/net/common.h"
 #include "blimp/net/connection_error_observer.h"
@@ -48,15 +49,13 @@ class BlimpConnectionTest : public testing::Test {
 
  protected:
   std::unique_ptr<BlimpMessage> CreateInputMessage() {
-    std::unique_ptr<BlimpMessage> msg(new BlimpMessage);
-    msg->set_type(BlimpMessage::INPUT);
-    return msg;
+    InputMessage* input;
+    return CreateBlimpMessage(&input);
   }
 
   std::unique_ptr<BlimpMessage> CreateControlMessage() {
-    std::unique_ptr<BlimpMessage> msg(new BlimpMessage);
-    msg->set_type(BlimpMessage::TAB_CONTROL);
-    return msg;
+    TabControlMessage* control;
+    return CreateBlimpMessage(&control);
   }
 
   base::MessageLoop message_loop_;