Used oneof in blimp_message.proto

blimp/common/create_blimp_message_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/common/create_blimp_message.h"
 #include "blimp/common/proto/blimp_message.pb.h"
 #include "blimp/common/proto/compositor.pb.h"
 #include "blimp/common/proto/input.pb.h"
 #include "blimp/common/proto/navigation.pb.h"
 #include "blimp/common/proto/render_widget.pb.h"
@@ skipping 46 matching lines @@
   EXPECT_NE(nullptr, message);
   EXPECT_EQ(details, message->mutable_render_widget());
   EXPECT_EQ(kTabId, message->target_tab_id());
 }
 
 TEST(CreateBlimpMessageTest, SizeMessage) {
   SizeMessage* details = nullptr;
   std::unique_ptr<BlimpMessage> message = CreateBlimpMessage(&details);
   EXPECT_NE(nullptr, details);
   EXPECT_NE(nullptr, message);
-  EXPECT_EQ(TabControlMessage::SIZE, message->mutable_tab_control()->type());
+  EXPECT_EQ(TabControlMessage::kSize,
+            message->mutable_tab_control()->tab_control_case());
   EXPECT_EQ(details, message->mutable_tab_control()->mutable_size());
 }
 
 TEST(CreateBlimpMessageTest, EngineSettingsMessage) {
   EngineSettingsMessage* details;
   std::unique_ptr<BlimpMessage> message = CreateBlimpMessage(&details);
   EXPECT_NE(nullptr, details);
   EXPECT_NE(nullptr, message);
   EXPECT_EQ(details, message->mutable_settings()->mutable_engine_settings());
 }
 
 TEST(CreateBlimpMessageTest, StartConnectionMessage) {
   const char* client_token = "token";
   const int protocol_version = 1;
   std::unique_ptr<BlimpMessage> message =
       CreateStartConnectionMessage(client_token, protocol_version);
   EXPECT_NE(nullptr, message);
-  EXPECT_EQ(BlimpMessage::PROTOCOL_CONTROL, message->type());
-  EXPECT_EQ(ProtocolControlMessage::START_CONNECTION,
-            message->protocol_control().type());
+  EXPECT_EQ(BlimpMessage::kProtocolControl, message->feature_case());
+  EXPECT_EQ(ProtocolControlMessage::kStartConnection,
+            message->protocol_control().connection_message_case());
   EXPECT_EQ(client_token,
             message->protocol_control().start_connection().client_token());
   EXPECT_EQ(protocol_version,
             message->protocol_control().start_connection().protocol_version());
 }
 
 }  // namespace
 }  // namespace blimp