Used oneof in blimp_message.proto

blimp/net/blimp_message_output_buffer_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_message_output_buffer.h"
 
 #include "base/callback_helpers.h"
 #include "base/logging.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 11 matching lines @@
 using testing::SaveArg;
 
 namespace blimp {
 namespace {
 
 class BlimpMessageOutputBufferTest : public testing::Test {
  public:
   BlimpMessageOutputBufferTest() {}
 
   void SetUp() override {
-    input_msg_.set_type(BlimpMessage::INPUT);
+    input_msg_.mutable_input();
     input_msg_.set_message_id(1);
-    compositor_msg_.set_type(BlimpMessage::COMPOSITOR);
+    compositor_msg_.mutable_compositor();
     compositor_msg_.set_message_id(2);
 
     // Buffer should only have space for two unacknowledged messages
     // (with message IDs).
     ASSERT_EQ(input_msg_.ByteSize(), compositor_msg_.ByteSize());
     buffer_.reset(new BlimpMessageOutputBuffer(2 * input_msg_.GetCachedSize()));
   }
 
  protected:
   void AddOutputExpectation(const BlimpMessage& msg) {
@@ skipping 209 matching lines @@
 
   // Remote end acknowledges #1, buffer should be empty.
   buffer_->OnMessageCheckpoint(2);
   ASSERT_EQ(0, buffer_->GetBufferByteSizeForTest());
   ASSERT_EQ(0, buffer_->GetUnacknowledgedMessageCountForTest());
   EXPECT_EQ(net::OK, complete_cb_2.WaitForResult());
 }
 
 }  // namespace
 }  // namespace blimp