Used oneof in blimp_message.proto

blimp/net/browser_connection_handler_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/browser_connection_handler.h"
 
 #include <stddef.h>
 
 #include <string>
 
@@ skipping 30 matching lines @@
 
   std::string expected_serialized;
   std::string actual_serialized;
   expected_message.SerializeToString(&expected_serialized);
   actual_message.SerializeToString(&actual_serialized);
   return expected_serialized == actual_serialized;
 }
 
 class FakeFeature {
  public:
-  FakeFeature(BlimpMessage::Type type,
+  FakeFeature(BlimpMessage::FeatureCase feature_case,
               BrowserConnectionHandler* connection_handler) {
-    outgoing_message_processor_ =
-        connection_handler->RegisterFeature(type, &incoming_message_processor_);
+    outgoing_message_processor_ = connection_handler->RegisterFeature(
+        feature_case, &incoming_message_processor_);
   }
 
   ~FakeFeature() {}
 
   BlimpMessageProcessor* outgoing_message_processor() {
     return outgoing_message_processor_.get();
   }
 
   MockBlimpMessageProcessor* incoming_message_processor() {
     return &incoming_message_processor_;
@@ skipping 41 matching lines @@
     callback.Run(net::OK);
   }
 
   FakeBlimpConnection* other_end_ = nullptr;
   ConnectionErrorObserver* error_observer_ = nullptr;
   BlimpMessageProcessor* incoming_message_processor_ = nullptr;
 };
 
 std::unique_ptr<BlimpMessage> CreateInputMessage(int tab_id) {
   std::unique_ptr<BlimpMessage> output(new BlimpMessage);
-  output->set_type(BlimpMessage::INPUT);
+  output->mutable_input();
   output->set_target_tab_id(tab_id);
   return output;
 }
 
 std::unique_ptr<BlimpMessage> CreateControlMessage(int tab_id) {
   std::unique_ptr<BlimpMessage> output(new BlimpMessage);
-  output->set_type(BlimpMessage::TAB_CONTROL);
+  output->mutable_tab_control();
   output->set_target_tab_id(tab_id);
   return output;
 }
 
 class BrowserConnectionHandlerTest : public testing::Test {
  public:
   BrowserConnectionHandlerTest()
       : client_connection_handler_(new BrowserConnectionHandler),
         engine_connection_handler_(new BrowserConnectionHandler) {
     SetupConnections();
 
-    client_input_feature_.reset(
-        new FakeFeature(BlimpMessage::INPUT, client_connection_handler_.get()));
-    engine_input_feature_.reset(
-        new FakeFeature(BlimpMessage::INPUT, engine_connection_handler_.get()));
+    client_input_feature_.reset(new FakeFeature(
+        BlimpMessage::kInput, client_connection_handler_.get()));
+    engine_input_feature_.reset(new FakeFeature(
+        BlimpMessage::kInput, engine_connection_handler_.get()));
     client_control_feature_.reset(new FakeFeature(
-        BlimpMessage::TAB_CONTROL, client_connection_handler_.get()));
+        BlimpMessage::kTabControl, client_connection_handler_.get()));
     engine_control_feature_.reset(new FakeFeature(
-        BlimpMessage::TAB_CONTROL, engine_connection_handler_.get()));
+        BlimpMessage::kTabControl, engine_connection_handler_.get()));
   }
 
   ~BrowserConnectionHandlerTest() override {}
   void TearDown() override { base::RunLoop().RunUntilIdle(); }
 
  protected:
   void SetupConnections() {
     client_connection_ = new FakeBlimpConnection();
     engine_connection_ = new FakeBlimpConnection();
     client_connection_->set_other_end(engine_connection_);
@@ skipping 69 matching lines @@
   // re-established.
   client_input_feature_->outgoing_message_processor()->ProcessMessage(
       std::move(client_input_message), net::CompletionCallback());
 
   // Simulates reconnection.
   SetupConnections();
 }
 
 }  // namespace
 }  // namespace blimp