Index: remoting/protocol/message_decoder_unittest.cc |
=================================================================== |
--- remoting/protocol/message_decoder_unittest.cc (revision 64676) |
+++ remoting/protocol/message_decoder_unittest.cc (working copy) |
@@ -41,10 +41,20 @@ |
// Then append 10 update sequences to the data. |
for (int i = 0; i < 10; ++i) { |
- msg.mutable_video_packet()->set_sequence_number(0); |
- msg.mutable_video_packet()->set_data(kTestData); |
+ 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(); |
+ AppendMessage(msg, &encoded_data); |
+ msg.Clear(); |
} |
*size = encoded_data.length(); |
@@ -80,19 +90,30 @@ |
} |
// Then verify the decoded messages. |
- EXPECT_EQ(11u, message_list.size()); |
+ EXPECT_EQ(31u, 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; |
- // Partial update stream. |
- EXPECT_TRUE(message->has_video_packet()); |
- EXPECT_EQ(kTestData, |
- message->video_packet().data().data()); |
+ 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()); |
+ } |
} |
STLDeleteElements(&message_list); |
} |