Index: blimp/common/create_blimp_message_unittest.cc |
diff --git a/blimp/common/create_blimp_message_unittest.cc b/blimp/common/create_blimp_message_unittest.cc |
index d2b57dc5488ad63b68cb4830e1867dd8735a1174..315b9d295dca0c6cfb45d2f230396c2a74db1f21 100644 |
--- a/blimp/common/create_blimp_message_unittest.cc |
+++ b/blimp/common/create_blimp_message_unittest.cc |
@@ -82,9 +82,9 @@ TEST(CreateBlimpMessageTest, StartConnectionMessage) { |
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, |