Used oneof in blimp_message.proto

blimp/net/blimp_connection_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 "blimp/net/blimp_connection.h"
 
 #include <stddef.h>
 
 #include <string>
 
 #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"
 #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"
@@ skipping 17 matching lines @@
     connection_->AddConnectionErrorObserver(&error_observer1_);
     connection_->AddConnectionErrorObserver(&error_observer2_);
     connection_->AddConnectionErrorObserver(&error_observer3_);
     connection_->RemoveConnectionErrorObserver(&error_observer3_);
   }
 
   ~BlimpConnectionTest() override {}
 
  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_;
   testing::StrictMock<MockPacketWriter>* writer_;
   testing::StrictMock<MockConnectionErrorObserver> error_observer1_;
   testing::StrictMock<MockConnectionErrorObserver> error_observer2_;
 
   // This error observer is Removed() immediately after it's added;
   // it should never be called.
   testing::StrictMock<MockConnectionErrorObserver> error_observer3_;
@@ skipping 60 matching lines @@
 
   net::TestCompletionCallback complete_cb_2;
   sender->ProcessMessage(CreateControlMessage(),
                          complete_cb_2.callback());
   base::ResetAndReturn(&write_packet_cb).Run(net::ERR_FAILED);
   EXPECT_EQ(net::ERR_FAILED, complete_cb_2.WaitForResult());
 }
 
 }  // namespace
 }  // namespace blimp