Index: remoting/protocol/message_decoder_unittest.cc |
diff --git a/remoting/protocol/message_decoder_unittest.cc b/remoting/protocol/message_decoder_unittest.cc |
index a2d4b203590175d77635f781a4155694a7ac5b12..90c66802830b4dd844aac824f4e825a31a35cece 100644 |
--- a/remoting/protocol/message_decoder_unittest.cc |
+++ b/remoting/protocol/message_decoder_unittest.cc |
@@ -41,18 +41,8 @@ static void PrepareData(uint8** buffer, int* size) { |
// Then append 10 update sequences to the data. |
for (int i = 0; i < 10; ++i) { |
- msg.mutable_begin_update_stream(); |
- AppendMessage(msg, &encoded_data); |
- msg.Clear(); |
- |
- msg.mutable_update_stream_packet()->mutable_rect_data()-> |
- set_sequence_number(0); |
- msg.mutable_update_stream_packet()->mutable_rect_data()-> |
- set_data(kTestData); |
- AppendMessage(msg, &encoded_data); |
- msg.Clear(); |
- |
- msg.mutable_end_update_stream(); |
+ msg.mutable_video_packet()->set_sequence_number(0); |
+ msg.mutable_video_packet()->set_data(kTestData); |
AppendMessage(msg, &encoded_data); |
msg.Clear(); |
} |
@@ -90,30 +80,19 @@ void SimulateReadSequence(const int read_sequence[], int sequence_size) { |
} |
// Then verify the decoded messages. |
- EXPECT_EQ(31u, message_list.size()); |
+ EXPECT_EQ(11u, message_list.size()); |
EXPECT_TRUE(message_list.front()->has_init_client()); |
delete message_list.front(); |
message_list.pop_front(); |
- int index = 0; |
for (std::list<ChromotingHostMessage*>::iterator it = |
message_list.begin(); |
it != message_list.end(); ++it) { |
ChromotingHostMessage* message = *it; |
- int type = index % 3; |
- ++index; |
- if (type == 0) { |
- // Begin update stream. |
- EXPECT_TRUE(message->has_begin_update_stream()); |
- } else if (type == 1) { |
- // Partial update stream. |
- EXPECT_TRUE(message->has_update_stream_packet()); |
- EXPECT_EQ(kTestData, |
- message->update_stream_packet().rect_data().data()); |
- } else if (type == 2) { |
- // End update stream. |
- EXPECT_TRUE(message->has_end_update_stream()); |
- } |
+ // Partial update stream. |
+ EXPECT_TRUE(message->has_video_packet()); |
+ EXPECT_EQ(kTestData, |
+ message->video_packet().data().data()); |
} |
STLDeleteElements(&message_list); |
} |