Index: chrome/browser/sync/notifier/cache_invalidation_packet_handler_unittest.cc |
=================================================================== |
--- chrome/browser/sync/notifier/cache_invalidation_packet_handler_unittest.cc (revision 117278) |
+++ chrome/browser/sync/notifier/cache_invalidation_packet_handler_unittest.cc (working copy) |
@@ -8,8 +8,10 @@ |
#include "base/memory/weak_ptr.h" |
#include "base/message_loop.h" |
#include "google/cacheinvalidation/callback.h" |
+#include "google/cacheinvalidation/v2/client_gateway.pb.h" |
#include "google/cacheinvalidation/v2/system-resources.h" |
#include "jingle/notifier/base/fake_base_task.h" |
+#include "jingle/notifier/listener/notification_defines.h" |
#include "testing/gmock/include/gmock/gmock.h" |
#include "testing/gtest/include/gtest/gtest.h" |
#include "talk/base/task.h" |
@@ -32,6 +34,12 @@ |
class CacheInvalidationPacketHandlerTest : public testing::Test { |
public: |
virtual ~CacheInvalidationPacketHandlerTest() {} |
+ |
+ void InitNotification( |
akalin
2012/01/12 23:06:44
seems cleaner to just have InitNotification return
ghc
2012/01/13 01:23:07
Done.
|
+ const std::string& data, notifier::Notification* notification) { |
+ notification->channel = "tango_raw"; |
+ notification->data = data; |
+ } |
}; |
TEST_F(CacheInvalidationPacketHandlerTest, Basic) { |
@@ -46,6 +54,10 @@ |
&callback, &MockMessageCallback::StoreMessage); |
const char kInboundMessage[] = "non-bogus"; |
+ ipc::invalidation::ClientGatewayMessage envelope; |
+ envelope.set_network_message(kInboundMessage); |
+ std::string serialized; |
+ envelope.SerializeToString(&serialized); |
{ |
CacheInvalidationPacketHandler handler(fake_base_task.AsWeakPtr()); |
handler.SetMessageReceiver(mock_message_callback); |
@@ -54,11 +66,11 @@ |
message_loop.RunAllPending(); |
{ |
- handler.HandleInboundPacket("bogus"); |
- std::string inbound_message_encoded; |
- EXPECT_TRUE( |
- base::Base64Encode(kInboundMessage, &inbound_message_encoded)); |
- handler.HandleInboundPacket(inbound_message_encoded); |
+ notifier::Notification notification; |
+ InitNotification("bogus", ¬ification); |
+ handler.OnNotificationReceived(notification); |
+ InitNotification(serialized, ¬ification); |
+ handler.OnNotificationReceived(notification); |
} |
// Take care of any tasks posted by HandleOutboundPacket(). |