google_apis/gcm/base/mcs_util_unittest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "google_apis/gcm/base/mcs_util.h"
   6
   7 #include "base/bind.h"
   8 #include "base/memory/scoped_ptr.h"
   9 #include "base/run_loop.h"
  10 #include "base/strings/string_number_conversions.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 namespace gcm {
  14 namespace {
  15
  16 const uint64 kAuthId = 4421448356646222460;
  17 const uint64 kAuthToken = 12345;
  18
  19 // Build a login request protobuf.
  20 TEST(MCSUtilTest, BuildLoginRequest) {
  21   scoped_ptr<mcs_proto::LoginRequest> login_request =
  22       BuildLoginRequest(kAuthId, kAuthToken, "1.0");
  23   ASSERT_EQ("chrome-1.0", login_request->id());
  24   ASSERT_EQ(base::Uint64ToString(kAuthToken), login_request->auth_token());
  25   ASSERT_EQ(base::Uint64ToString(kAuthId), login_request->user());
  26   ASSERT_EQ("android-3d5c23dac2a1fa7c", login_request->device_id());
  27   ASSERT_EQ("new_vc", login_request->setting(0).name());
  28   ASSERT_EQ("1", login_request->setting(0).value());
  29   // TODO(zea): test the other fields once they have valid values.
  30 }
  31
  32 // Test building a protobuf and extracting the tag from a protobuf.
  33 TEST(MCSUtilTest, ProtobufToTag) {
  34   for (size_t i = 0; i < kNumProtoTypes; ++i) {
  35     scoped_ptr<google::protobuf::MessageLite> protobuf =
  36         BuildProtobufFromTag(i);
  37     if (!protobuf.get())  // Not all tags have protobuf definitions.
  38       continue;
  39     ASSERT_EQ((int)i, GetMCSProtoTag(*protobuf)) << "Type " << i;
  40   }
  41 }
  42
  43 // Test getting and setting persistent ids.
  44 TEST(MCSUtilTest, PersistentIds) {
  45   static_assert(kNumProtoTypes == 16U, "Update Persistent Ids");
  46   const int kTagsWithPersistentIds[] = {
  47     kIqStanzaTag,
  48     kDataMessageStanzaTag
  49   };
  50   for (size_t i = 0; i < arraysize(kTagsWithPersistentIds); ++i) {
  51     int tag = kTagsWithPersistentIds[i];
  52     scoped_ptr<google::protobuf::MessageLite> protobuf =
  53         BuildProtobufFromTag(tag);
  54     ASSERT_TRUE(protobuf.get());
  55     SetPersistentId(base::IntToString(tag), protobuf.get());
  56     int get_val = 0;
  57     base::StringToInt(GetPersistentId(*protobuf), &get_val);
  58     ASSERT_EQ(tag, get_val);
  59   }
  60 }
  61
  62 // Test getting and setting stream ids.
  63 TEST(MCSUtilTest, StreamIds) {
  64   static_assert(kNumProtoTypes == 16U, "Update Stream Ids");
  65   const int kTagsWithStreamIds[] = {
  66     kIqStanzaTag,
  67     kDataMessageStanzaTag,
  68     kHeartbeatPingTag,
  69     kHeartbeatAckTag,
  70     kLoginResponseTag,
  71   };
  72   for (size_t i = 0; i < arraysize(kTagsWithStreamIds); ++i) {
  73     int tag = kTagsWithStreamIds[i];
  74     scoped_ptr<google::protobuf::MessageLite> protobuf =
  75         BuildProtobufFromTag(tag);
  76     ASSERT_TRUE(protobuf.get());
  77     SetLastStreamIdReceived(tag, protobuf.get());
  78     int get_id = GetLastStreamIdReceived(*protobuf);
  79     ASSERT_EQ(tag, get_id);
  80   }
  81 }
  82
  83 }  // namespace
  84 }  // namespace gcm