Index: google_apis/gcm/engine/gcm_checkin_request_unittest.cc |
diff --git a/google_apis/gcm/engine/gcm_checkin_request_unittest.cc b/google_apis/gcm/engine/gcm_checkin_request_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..1dbbec069bfa0fc1d48de8cffa231234adb8ab61 |
--- /dev/null |
+++ b/google_apis/gcm/engine/gcm_checkin_request_unittest.cc |
@@ -0,0 +1,318 @@ |
+// Copyright (c) 2013 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include <string> |
+ |
+#include "google_apis/gcm/engine/gcm_checkin_request.h" |
+#include "google_apis/gcm/protocol/checkin.pb.h" |
+#include "net/url_request/test_url_fetcher_factory.h" |
+#include "net/url_request/url_request_test_util.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace gcm { |
+ |
+const int64 kAndroidId = 42; |
+const int64 kBlankAndroidId = -1; |
+const uint64 kBlankSecurityToken = 999999; |
+const char kChromeVersion[] = "Version String"; |
+const uint64 kSecurityToken = 77; |
+const int64 kUserSerialNumber = 7; |
+ |
+class GCMCheckinRequestTest : public testing::Test { |
+ public: |
+ enum MissingValue { |
+ MISSING_ANDROID_ID, |
+ MISSING_SECURITY_TOKEN |
+ }; |
+ |
+ GCMCheckinRequestTest(); |
+ virtual ~GCMCheckinRequestTest(); |
+ |
+ checkin_proto::ChromeBuildProto CreateChromeBuildProto(); |
+ |
+ void FetcherCallback(uint64 android_id, uint64 security_token); |
+ |
+ void SetResponse(net::TestURLFetcher* fetcher, |
+ uint64 android_id, |
+ uint64 security_token); |
+ |
+ void SetResponseMissingValue(net::TestURLFetcher* fetcher, |
+ MissingValue missing_value); |
+ |
+ protected: |
+ int64 android_id_; |
+ uint64 security_token_; |
+ int checkin_device_type_; |
+ GCMClient::CheckInInfo user_checkin_info_; |
+ base::MessageLoop message_loop_; |
+ net::TestURLFetcherFactory factory_; |
jianli
2014/01/06 22:26:33
nit: url_factory_factory_
fgorski
2014/01/06 23:15:54
Done.
|
+ scoped_refptr<net::TestURLRequestContextGetter> getter_; |
jianli
2014/01/06 22:26:33
nit: getter_ is too short to convey the meaning. P
fgorski
2014/01/06 23:15:54
Done.
|
+}; |
+ |
+GCMCheckinRequestTest::GCMCheckinRequestTest() |
+ : android_id_(kBlankAndroidId), |
+ security_token_(kBlankSecurityToken), |
+ checkin_device_type_(0), |
+ getter_(new net::TestURLRequestContextGetter( |
+ message_loop_.message_loop_proxy())) {} |
+ |
+GCMCheckinRequestTest::~GCMCheckinRequestTest() {} |
+ |
+checkin_proto::ChromeBuildProto |
+GCMCheckinRequestTest::CreateChromeBuildProto() { |
+ checkin_proto::ChromeBuildProto chrome_build_proto; |
+ chrome_build_proto.set_platform( |
+ checkin_proto::ChromeBuildProto::PLATFORM_LINUX); |
+ chrome_build_proto.set_channel( |
+ checkin_proto::ChromeBuildProto::CHANNEL_CANARY); |
+ chrome_build_proto.set_chrome_version(kChromeVersion); |
+ return chrome_build_proto; |
+} |
+ |
+void GCMCheckinRequestTest::FetcherCallback(uint64 android_id, |
+ uint64 security_token) { |
+ android_id_ = android_id; |
+ security_token_ = security_token; |
+} |
+ |
+void GCMCheckinRequestTest::SetResponse(net::TestURLFetcher* fetcher, |
+ uint64 android_id, |
+ uint64 security_token) { |
+ checkin_proto::AndroidCheckinResponse response; |
+ response.set_stats_ok(true); |
+ response.set_android_id(android_id); |
+ response.set_security_token(security_token); |
+ |
+ std::string response_string; |
+ response.SerializeToString(&response_string); |
+ fetcher->SetResponseString(response_string); |
+} |
+ |
+void GCMCheckinRequestTest::SetResponseMissingValue( |
+ net::TestURLFetcher* fetcher, |
+ MissingValue missing_value) { |
+ checkin_proto::AndroidCheckinResponse response; |
+ response.set_stats_ok(true); |
+ if (missing_value != MISSING_ANDROID_ID) |
+ response.set_android_id(kAndroidId); |
+ else |
+ response.set_security_token(kSecurityToken); |
+ |
+ std::string response_string; |
+ response.SerializeToString(&response_string); |
+ fetcher->SetResponseString(response_string); |
+} |
+ |
+TEST_F(GCMCheckinRequestTest, FetcherData) { |
+ checkin_proto::ChromeBuildProto chrome_build_proto = CreateChromeBuildProto(); |
+ user_checkin_info_.android_id = kAndroidId; |
+ user_checkin_info_.secret = kSecurityToken; |
+ GCMCheckinRequest request(base::Bind(&GCMCheckinRequestTest::FetcherCallback, |
+ base::Unretained(this)), |
+ chrome_build_proto, |
+ kUserSerialNumber, |
+ user_checkin_info_, |
+ getter_.get()); |
+ request.Start(); |
+ |
+ net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0); |
+ ASSERT_TRUE(fetcher); |
+ fetcher->set_response_code(net::HTTP_OK); |
+ checkin_proto::AndroidCheckinRequest request_proto; |
+ request_proto.ParseFromString(fetcher->upload_data()); |
+ EXPECT_EQ(kAndroidId, request_proto.id()); |
+ EXPECT_EQ(kSecurityToken, request_proto.security_token()); |
+ EXPECT_EQ(kUserSerialNumber, request_proto.user_serial_number()); |
+ EXPECT_EQ(chrome_build_proto.platform(), |
+ request_proto.checkin().chrome_build().platform()); |
+ EXPECT_EQ(chrome_build_proto.chrome_version(), |
+ request_proto.checkin().chrome_build().chrome_version()); |
+ EXPECT_EQ(chrome_build_proto.channel(), |
+ request_proto.checkin().chrome_build().channel()); |
+ |
+#if defined(CHROME_OS) |
+ EXPECT_EQ(checkin_proto::DEVICE_CHROME_OS, request_proto.checkin().type()); |
+#else |
+ EXPECT_EQ(checkin_proto::DEVICE_CHROME_BROWSER, |
+ request_proto.checkin().type()); |
+#endif |
+} |
+ |
+TEST_F(GCMCheckinRequestTest, ResponseBodyEmpty) { |
+ checkin_proto::ChromeBuildProto chrome_build_proto = CreateChromeBuildProto(); |
+ GCMCheckinRequest request(base::Bind(&GCMCheckinRequestTest::FetcherCallback, |
+ base::Unretained(this)), |
+ chrome_build_proto, |
+ kUserSerialNumber, |
+ user_checkin_info_, |
+ getter_.get()); |
+ request.Start(); |
+ |
+ net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0); |
+ ASSERT_TRUE(fetcher); |
+ fetcher->set_response_code(net::HTTP_OK); |
+ fetcher->SetResponseString(std::string()); |
+ fetcher->delegate()->OnURLFetchComplete(fetcher); |
+ EXPECT_EQ(0u, android_id_); |
+ EXPECT_EQ(0u, security_token_); |
+} |
+ |
+TEST_F(GCMCheckinRequestTest, ResponseBodyCorrupted) { |
+ checkin_proto::ChromeBuildProto chrome_build_proto = CreateChromeBuildProto(); |
+ GCMCheckinRequest request(base::Bind(&GCMCheckinRequestTest::FetcherCallback, |
+ base::Unretained(this)), |
+ chrome_build_proto, |
+ kUserSerialNumber, |
+ user_checkin_info_, |
+ getter_.get()); |
+ request.Start(); |
+ |
+ net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0); |
+ ASSERT_TRUE(fetcher); |
+ fetcher->set_response_code(net::HTTP_OK); |
+ fetcher->SetResponseString("Corrupted response body"); |
+ fetcher->delegate()->OnURLFetchComplete(fetcher); |
+ EXPECT_EQ(0u, android_id_); |
+ EXPECT_EQ(0u, security_token_); |
+} |
+ |
+TEST_F(GCMCheckinRequestTest, ResponseHttpStatusNotOK) { |
+ checkin_proto::ChromeBuildProto chrome_build_proto = CreateChromeBuildProto(); |
+ GCMCheckinRequest request(base::Bind(&GCMCheckinRequestTest::FetcherCallback, |
+ base::Unretained(this)), |
+ chrome_build_proto, |
+ kUserSerialNumber, |
+ user_checkin_info_, |
+ getter_.get()); |
+ request.Start(); |
+ |
+ net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0); |
+ ASSERT_TRUE(fetcher); |
+ fetcher->set_response_code(net::HTTP_UNAUTHORIZED); |
+ fetcher->SetResponseString(std::string()); |
+ fetcher->delegate()->OnURLFetchComplete(fetcher); |
+ EXPECT_EQ(0u, android_id_); |
+ EXPECT_EQ(0u, security_token_); |
+} |
+ |
+TEST_F(GCMCheckinRequestTest, ResponseMissingAndroidId) { |
+ checkin_proto::ChromeBuildProto chrome_build_proto = CreateChromeBuildProto(); |
+ GCMCheckinRequest request(base::Bind(&GCMCheckinRequestTest::FetcherCallback, |
+ base::Unretained(this)), |
+ chrome_build_proto, |
+ kUserSerialNumber, |
+ user_checkin_info_, |
+ getter_.get()); |
+ request.Start(); |
+ |
+ net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0); |
+ ASSERT_TRUE(fetcher); |
+ fetcher->set_response_code(net::HTTP_UNAUTHORIZED); |
+ SetResponseMissingValue(fetcher, MISSING_ANDROID_ID); |
+ fetcher->delegate()->OnURLFetchComplete(fetcher); |
+ EXPECT_EQ(0u, android_id_); |
+ EXPECT_EQ(0u, security_token_); |
+} |
+ |
+TEST_F(GCMCheckinRequestTest, ResponseMissingSecurityToken) { |
+ checkin_proto::ChromeBuildProto chrome_build_proto = CreateChromeBuildProto(); |
+ GCMCheckinRequest request(base::Bind(&GCMCheckinRequestTest::FetcherCallback, |
+ base::Unretained(this)), |
+ chrome_build_proto, |
+ kUserSerialNumber, |
+ user_checkin_info_, |
+ getter_.get()); |
+ request.Start(); |
+ |
+ net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0); |
+ ASSERT_TRUE(fetcher); |
+ fetcher->set_response_code(net::HTTP_UNAUTHORIZED); |
+ SetResponseMissingValue(fetcher, MISSING_SECURITY_TOKEN); |
+ fetcher->delegate()->OnURLFetchComplete(fetcher); |
+ EXPECT_EQ(0u, android_id_); |
+ EXPECT_EQ(0u, security_token_); |
+} |
+ |
+TEST_F(GCMCheckinRequestTest, ResponseValueOfAndroidIdIsZero) { |
+ checkin_proto::ChromeBuildProto chrome_build_proto = CreateChromeBuildProto(); |
+ GCMCheckinRequest request(base::Bind(&GCMCheckinRequestTest::FetcherCallback, |
+ base::Unretained(this)), |
+ chrome_build_proto, |
+ kUserSerialNumber, |
+ user_checkin_info_, |
+ getter_.get()); |
+ request.Start(); |
+ |
+ net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0); |
+ ASSERT_TRUE(fetcher); |
+ fetcher->set_response_code(net::HTTP_UNAUTHORIZED); |
+ SetResponse(fetcher, 0, kSecurityToken); |
+ fetcher->delegate()->OnURLFetchComplete(fetcher); |
+ EXPECT_EQ(0u, android_id_); |
+ EXPECT_EQ(0u, security_token_); |
+} |
+ |
+TEST_F(GCMCheckinRequestTest, ResponseValueOfSecurityTokenIsZero) { |
+ checkin_proto::ChromeBuildProto chrome_build_proto = CreateChromeBuildProto(); |
+ GCMCheckinRequest request(base::Bind(&GCMCheckinRequestTest::FetcherCallback, |
+ base::Unretained(this)), |
+ chrome_build_proto, |
+ kUserSerialNumber, |
+ user_checkin_info_, |
+ getter_.get()); |
+ request.Start(); |
+ |
+ net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0); |
+ ASSERT_TRUE(fetcher); |
+ fetcher->set_response_code(net::HTTP_UNAUTHORIZED); |
+ SetResponse(fetcher, kAndroidId, 0); |
+ fetcher->delegate()->OnURLFetchComplete(fetcher); |
+ EXPECT_EQ(0u, android_id_); |
+ EXPECT_EQ(0u, security_token_); |
+} |
+ |
+TEST_F(GCMCheckinRequestTest, SuccessfulFirstTimeCheckin) { |
+ checkin_proto::ChromeBuildProto chrome_build_proto = CreateChromeBuildProto(); |
+ GCMCheckinRequest request(base::Bind(&GCMCheckinRequestTest::FetcherCallback, |
+ base::Unretained(this)), |
+ chrome_build_proto, |
+ kUserSerialNumber, |
+ user_checkin_info_, |
+ getter_.get()); |
+ request.Start(); |
+ |
+ net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0); |
+ ASSERT_TRUE(fetcher); |
+ fetcher->set_response_code(net::HTTP_OK); |
+ SetResponse(fetcher, kAndroidId, kSecurityToken); |
+ fetcher->delegate()->OnURLFetchComplete(fetcher); |
+ EXPECT_EQ(kAndroidId, android_id_); |
+ EXPECT_EQ(kSecurityToken, security_token_); |
+} |
+ |
+TEST_F(GCMCheckinRequestTest, SuccessfulSubsequentCheckin) { |
+ checkin_proto::ChromeBuildProto chrome_build_proto = CreateChromeBuildProto(); |
+ // Having non-zero android_id and secret indicate it is a subsequent check in. |
+ user_checkin_info_.android_id = kAndroidId; |
+ user_checkin_info_.secret = kSecurityToken; |
+ GCMCheckinRequest request(base::Bind(&GCMCheckinRequestTest::FetcherCallback, |
+ base::Unretained(this)), |
+ chrome_build_proto, |
+ kUserSerialNumber, |
+ user_checkin_info_, |
+ getter_.get()); |
+ request.Start(); |
+ |
+ EXPECT_EQ(kBlankAndroidId, android_id_); |
+ EXPECT_EQ(kBlankSecurityToken, security_token_); |
+ net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0); |
+ ASSERT_TRUE(fetcher); |
+ fetcher->set_response_code(net::HTTP_OK); |
+ SetResponse(fetcher, kAndroidId, kSecurityToken); |
+ fetcher->delegate()->OnURLFetchComplete(fetcher); |
+ EXPECT_EQ(kAndroidId, android_id_); |
+ EXPECT_EQ(kSecurityToken, security_token_); |
+} |
+ |
+} // namespace gcm |