Used oneof in blimp_message.proto

blimp/net/blimp_message_checkpointer_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_checkpointer.h"
 
 #include "base/memory/ptr_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/test/test_mock_time_task_runner.h"
 #include "base/thread_task_runner_handle.h"
@@ skipping 16 matching lines @@
   MockCheckpointObserver() {}
   ~MockCheckpointObserver() override {}
 
   MOCK_METHOD1(OnMessageCheckpoint, void(int64_t));
 };
 
 }  // namespace
 
 static std::unique_ptr<BlimpMessage> CreateExpectedAckMessage(int64_t id) {
   std::unique_ptr<BlimpMessage> message = base::WrapUnique(new BlimpMessage);
-  message->set_type(BlimpMessage::PROTOCOL_CONTROL);
+  message->mutable_protocol_control();
   ProtocolControlMessage* protocol_control =
       message->mutable_protocol_control();
   protocol_control->set_type(ProtocolControlMessage::CHECKPOINT_ACK);
   CheckpointAckMessage* checkpoint_ack =
       protocol_control->mutable_checkpoint_ack();
   checkpoint_ack->set_checkpoint_id(id);
   return message;
 }
 
 class BlimpMessageCheckpointerTest : public testing::Test {
  public:
   BlimpMessageCheckpointerTest()
       : runner_(new base::TestMockTimeTaskRunner), runner_handle_(runner_) {}
 
   ~BlimpMessageCheckpointerTest() override {}
 
   int64_t SimulateIncomingMessage() {
     std::unique_ptr<BlimpMessage> message(new BlimpMessage);
     message->set_message_id(++message_id_);
-    message->set_type(BlimpMessage::INPUT);
+    message->mutable_input();
     checkpointer_->ProcessMessage(
         std::move(message),
         base::Bind(&BlimpMessageCheckpointerTest::IncomingCompletionCallback,
                    base::Unretained(this)));
     return message_id_;
   }
 
   void SetUp() override {
     checkpointer_ = base::WrapUnique(new BlimpMessageCheckpointer(
         &incoming_processor_, &outgoing_processor_, &checkpoint_observer_));
@@ skipping 106 matching lines @@
 
   // Simulate an incoming message.
   std::unique_ptr<BlimpMessage> ack_message = CreateCheckpointAckMessage(10);
   checkpointer_->ProcessMessage(
       std::move(ack_message),
       base::Bind(&BlimpMessageCheckpointerTest::IncomingCompletionCallback,
                  base::Unretained(this)));
 }
 
 }  // namespace blimp