Used oneof in blimp_message.proto

blimp/net/browser_connection_handler_unittest.cc

Index: blimp/net/browser_connection_handler_unittest.cc
diff --git a/blimp/net/browser_connection_handler_unittest.cc b/blimp/net/browser_connection_handler_unittest.cc
index 2c539492ba8b3f0127321e7925721c2f8a0aff1f..2f0c809162c99348e47f1a0f5c86cf64a773d668 100644
--- a/blimp/net/browser_connection_handler_unittest.cc
+++ b/blimp/net/browser_connection_handler_unittest.cc
@@ -48,10 +48,10 @@ MATCHER_P(EqualsMessageIgnoringId, message, "") {
 
 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() {}
@@ -113,14 +113,14 @@ class FakeBlimpConnection : public BlimpConnection,
 
 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;
 }
@@ -132,14 +132,14 @@ class BrowserConnectionHandlerTest : public testing::Test {
         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 {}