| Index: remoting/protocol/message_decoder_unittest.cc
|
| diff --git a/remoting/protocol/message_decoder_unittest.cc b/remoting/protocol/message_decoder_unittest.cc
|
| index d85c37b46304b8db4b97ade38ac958ed9415e5b4..f2ecfcc2f4e2007bc65ab294566a0c4b8986bf1e 100644
|
| --- a/remoting/protocol/message_decoder_unittest.cc
|
| +++ b/remoting/protocol/message_decoder_unittest.cc
|
| @@ -6,7 +6,7 @@
|
|
|
| #include "base/scoped_ptr.h"
|
| #include "base/stl_util-inl.h"
|
| -#include "remoting/proto/internal.pb.h"
|
| +#include "remoting/proto/event.pb.h"
|
| #include "remoting/protocol/message_decoder.h"
|
| #include "remoting/protocol/util.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| @@ -16,7 +16,7 @@ namespace protocol {
|
|
|
| static const int kTestKey = 142;
|
|
|
| -static void AppendMessage(const ChromotingClientMessage& msg,
|
| +static void AppendMessage(const EventMessage& msg,
|
| std::string* buffer) {
|
| // Contains one encoded message.
|
| scoped_refptr<net::IOBufferWithSize> encoded_msg;
|
| @@ -29,12 +29,14 @@ static void PrepareData(uint8** buffer, int* size) {
|
| // Contains all encoded messages.
|
| std::string encoded_data;
|
|
|
| - ChromotingClientMessage msg;
|
| + EventMessage msg;
|
|
|
| // Then append 10 update sequences to the data.
|
| for (int i = 0; i < 10; ++i) {
|
| - msg.mutable_key_event()->set_key(kTestKey + i);
|
| - msg.mutable_key_event()->set_pressed((i % 2) != 0);
|
| + Event* event = msg.add_event();
|
| + event->set_timestamp(i);
|
| + event->mutable_key()->set_keycode(kTestKey + i);
|
| + event->mutable_key()->set_pressed((i % 2) != 0);
|
| AppendMessage(msg, &encoded_data);
|
| msg.Clear();
|
| }
|
| @@ -59,7 +61,7 @@ void SimulateReadSequence(const int read_sequence[], int sequence_size) {
|
|
|
| // Then feed the protocol decoder using the above generated data and the
|
| // read pattern.
|
| - std::list<ChromotingClientMessage*> message_list;
|
| + std::list<EventMessage*> message_list;
|
| for (int i = 0; i < size;) {
|
| // First generate the amount to feed the decoder.
|
| int read = std::min(size - i, read_sequence[i % sequence_size]);
|
| @@ -75,17 +77,18 @@ void SimulateReadSequence(const int read_sequence[], int sequence_size) {
|
| EXPECT_EQ(10u, message_list.size());
|
|
|
| int index = 0;
|
| - for (std::list<ChromotingClientMessage*>::iterator it =
|
| + for (std::list<EventMessage*>::iterator it =
|
| message_list.begin();
|
| it != message_list.end(); ++it) {
|
| - ChromotingClientMessage* message = *it;
|
| + EventMessage* message = *it;
|
| // Partial update stream.
|
| - EXPECT_TRUE(message->has_key_event());
|
| + EXPECT_EQ(message->event_size(), 1);
|
| + EXPECT_TRUE(message->event(0).has_key());
|
|
|
| // TODO(sergeyu): Don't use index here. Instead store the expected values
|
| // in an array.
|
| - EXPECT_EQ(kTestKey + index, message->key_event().key());
|
| - EXPECT_EQ((index % 2) != 0, message->key_event().pressed());
|
| + EXPECT_EQ(kTestKey + index, message->event(0).key().keycode());
|
| + EXPECT_EQ((index % 2) != 0, message->event(0).key().pressed());
|
| ++index;
|
| }
|
| STLDeleteElements(&message_list);
|
|
|