Used oneof in blimp_message.proto

blimp/net/blimp_message_pump_unittest.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <stddef.h>
 #include <string>
 
 #include "base/callback_helpers.h"
+#include "blimp/common/create_blimp_message.h"
 #include "blimp/common/proto/blimp_message.pb.h"
 #include "blimp/net/blimp_message_pump.h"
 #include "blimp/net/common.h"
 #include "blimp/net/connection_error_observer.h"
 #include "blimp/net/test_common.h"
 #include "net/base/completion_callback.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 #include "net/base/test_completion_callback.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::_;
 using testing::DoAll;
 using testing::InSequence;
 using testing::NotNull;
 using testing::Return;
 using testing::SaveArg;
 
 namespace blimp {
 namespace {
 
 class BlimpMessagePumpTest : public testing::Test {
  public:
-  BlimpMessagePumpTest()
-      : message1_(new BlimpMessage), message2_(new BlimpMessage) {
-    message1_->set_type(BlimpMessage::INPUT);
-    message2_->set_type(BlimpMessage::TAB_CONTROL);
+  BlimpMessagePumpTest() {
+    TabControlMessage* tab_control = nullptr;
+    message1_ = CreateBlimpMessage(&tab_control);
+    InputMessage* input = nullptr;
+    message2_ = CreateBlimpMessage(&input);
     message_pump_.reset(new BlimpMessagePump(&reader_));
     message_pump_->set_error_observer(&error_observer_);
   }
 
   ~BlimpMessagePumpTest() override {}
 
   void NullMessageProcessor() { message_pump_->SetMessageProcessor(nullptr); }
 
  protected:
   std::unique_ptr<BlimpMessage> message1_;
@@ skipping 107 matching lines @@
   message_pump_->SetMessageProcessor(&receiver_);
   ASSERT_FALSE(read_packet_cb.is_null());
   base::ResetAndReturn(&read_packet_cb).Run(message1_->ByteSize());
   process_msg_cb.Run(net::OK);
   // Running |process_msg_cb| should NOT trigger another ReadPacket call.
 }
 
 }  // namespace
 
 }  // namespace blimp