Support getting and deleting token for Instance ID.

components/gcm_driver/instance_id/instance_id_driver_unittest.cc

 // Copyright 2015 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 "components/gcm_driver/instance_id/instance_id_driver.h"
 
 #include <cmath>
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/strings/string_util.h"
 #include "components/gcm_driver/instance_id/fake_gcm_driver_for_instance_id.h"
 #include "components/gcm_driver/instance_id/instance_id.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace instance_id {
 
 namespace {
 
 const char kTestAppID1[] = "TestApp1";
 const char kTestAppID2[] = "TestApp2";
+const char kAuthorizedEntity1[] = "Sender 1";
+const char kAuthorizedEntity2[] = "Sender 2";
+const char kScope1[] = "GCM1";
+const char kScope2[] = "FooBar";
 
 bool VerifyInstanceID(const std::string& str) {
   // Checks the length.
   if (str.length() != static_cast<size_t>(
           std::ceil(InstanceID::kInstanceIDByteLength * 8 / 6.0)))
     return false;
 
   // Checks if it is URL-safe base64 encoded.
   for (auto ch : str) {
     if (!IsAsciiAlpha(ch) && !IsAsciiDigit(ch) && ch != '_' && ch != '-')
@@ skipping 14 matching lines @@
 
   void WaitForAsyncOperation();
 
   // Recreates InstanceIDDriver to simulate restart.
   void RecreateInstanceIDDriver();
 
   // Sync wrappers for async version.
   std::string GetID(InstanceID* instance_id);
   base::Time GetCreationTime(InstanceID* instance_id);
   InstanceID::Result DeleteID(InstanceID* instance_id);
+  std::string GetToken(
+      InstanceID* instance_id,
+      const std::string& authorized_entity,
+      const std::string& scope,
+      const std::map<std::string, std::string>& options);
+  InstanceID::Result DeleteToken(
+      InstanceID* instance_id,
+      const std::string& authorized_entity,
+      const std::string& scope);
 
   InstanceIDDriver* driver() const { return driver_.get(); }
 
  private:
   void GetIDCompleted(const std::string& id);
   void GetCreationTimeCompleted(const base::Time& creation_time);
   void DeleteIDCompleted(InstanceID::Result result);
+  void GetTokenCompleted(const std::string& token, InstanceID::Result result);
+  void DeleteTokenCompleted(InstanceID::Result result);
 
   base::MessageLoopForUI message_loop_;
   scoped_ptr<FakeGCMDriverForInstanceID> gcm_driver_;
   scoped_ptr<InstanceIDDriver> driver_;
 
   std::string id_;
   base::Time creation_time_;
+  std::string token_;
   InstanceID::Result result_;
 
   bool async_operation_completed_;
   base::Closure async_operation_completed_callback_;
 
   DISALLOW_COPY_AND_ASSIGN(InstanceIDDriverTest);
 };
 
 InstanceIDDriverTest::InstanceIDDriverTest()
     : result_(InstanceID::UNKNOWN_ERROR),
@@ skipping 42 matching lines @@
 
 InstanceID::Result InstanceIDDriverTest::DeleteID(InstanceID* instance_id) {
   async_operation_completed_ = false;
   result_ = InstanceID::UNKNOWN_ERROR;;
   instance_id->DeleteID(base::Bind(&InstanceIDDriverTest::DeleteIDCompleted,
                         base::Unretained(this)));
   WaitForAsyncOperation();
   return result_;
 }
 
+std::string InstanceIDDriverTest::GetToken(
+    InstanceID* instance_id,
+    const std::string& authorized_entity,
+    const std::string& scope,
+    const std::map<std::string, std::string>& options) {
+  async_operation_completed_ = false;
+  token_.clear();
+  result_ = InstanceID::UNKNOWN_ERROR;;
+  instance_id->GetToken(
+      authorized_entity,
+      scope,
+      options,
+      base::Bind(&InstanceIDDriverTest::GetTokenCompleted,
+                 base::Unretained(this)));
+  WaitForAsyncOperation();
+  return token_;
+}
+
+InstanceID::Result InstanceIDDriverTest::DeleteToken(
+    InstanceID* instance_id,
+    const std::string& authorized_entity,
+    const std::string& scope) {
+  async_operation_completed_ = false;
+  result_ = InstanceID::UNKNOWN_ERROR;;
+  instance_id->DeleteToken(
+      authorized_entity,
+      scope,
+      base::Bind(&InstanceIDDriverTest::DeleteTokenCompleted,
+                 base::Unretained(this)));
+  WaitForAsyncOperation();
+  return result_;
+}
+
 void InstanceIDDriverTest::GetIDCompleted(const std::string& id) {
+  DCHECK(!async_operation_completed_);
   async_operation_completed_ = true;
   id_ = id;
   if (!async_operation_completed_callback_.is_null())
     async_operation_completed_callback_.Run();
 }
 
 void InstanceIDDriverTest::GetCreationTimeCompleted(
     const base::Time& creation_time) {
+  DCHECK(!async_operation_completed_);
   async_operation_completed_ = true;
   creation_time_ = creation_time;
   if (!async_operation_completed_callback_.is_null())
     async_operation_completed_callback_.Run();
 }
 
 void InstanceIDDriverTest::DeleteIDCompleted(InstanceID::Result result) {
+  DCHECK(!async_operation_completed_);
   async_operation_completed_ = true;
   result_ = result;
   if (!async_operation_completed_callback_.is_null())
+    async_operation_completed_callback_.Run();
+}
+
+void InstanceIDDriverTest::GetTokenCompleted(
+    const std::string& token, InstanceID::Result result){
+  DCHECK(!async_operation_completed_);
+  async_operation_completed_ = true;
+  token_ = token;
+  result_ = result;
+  if (!async_operation_completed_callback_.is_null())
+    async_operation_completed_callback_.Run();
+}
+
+void InstanceIDDriverTest::DeleteTokenCompleted(InstanceID::Result result) {
+  DCHECK(!async_operation_completed_);
+  async_operation_completed_ = true;
+  result_ = result;
+  if (!async_operation_completed_callback_.is_null())
     async_operation_completed_callback_.Run();
 }
 
 TEST_F(InstanceIDDriverTest, NewID) {
   // Creation time should not be set when the ID is not created.
   InstanceID* instance_id1 = driver()->GetInstanceID(kTestAppID1);
   EXPECT_TRUE(GetCreationTime(instance_id1).is_null());
 
   // New ID is generated for the first time.
   std::string id1 = GetID(instance_id1);
@@ skipping 52 matching lines @@
   // New ID will be generated from GetID after calling DeleteID.
   EXPECT_EQ(InstanceID::SUCCESS, DeleteID(instance_id));
   EXPECT_TRUE(GetCreationTime(instance_id).is_null());
 
   std::string id2 = GetID(instance_id);
   EXPECT_FALSE(id2.empty());
   EXPECT_NE(id1, id2);
   EXPECT_FALSE(GetCreationTime(instance_id).is_null());
 }
 
+TEST_F(InstanceIDDriverTest, GetToken) {
+  InstanceID* instance_id = driver()->GetInstanceID(kTestAppID1);
+  std::map<std::string, std::string> options;
+  std::string token1 =
+      GetToken(instance_id, kAuthorizedEntity1, kScope1, options);
+  EXPECT_FALSE(token1.empty());
+
+  // Same token is returned for same authorized entity and scope.
+  EXPECT_EQ(token1,
+            GetToken(instance_id, kAuthorizedEntity1, kScope1, options));
+
+  // Different token is returned for different authorized entity or scope.
+  std::string token2 =
+      GetToken(instance_id, kAuthorizedEntity1, kScope2, options);
+  EXPECT_FALSE(token2.empty());
+  EXPECT_NE(token1, token2);
+
+  std::string token3 =
+      GetToken(instance_id, kAuthorizedEntity2, kScope1, options);
+  EXPECT_FALSE(token3.empty());
+  EXPECT_NE(token1, token3);
+  EXPECT_NE(token2, token3);
+}
+
+TEST_F(InstanceIDDriverTest, DeleteToken) {
+  InstanceID* instance_id = driver()->GetInstanceID(kTestAppID1);
+  std::map<std::string, std::string> options;
+
+  // Gets 2 tokens.
+  std::string token1 =
+      GetToken(instance_id, kAuthorizedEntity1, kScope1, options);
+  EXPECT_FALSE(token1.empty());
+  std::string token2 =
+      GetToken(instance_id, kAuthorizedEntity2, kScope1, options);
+  EXPECT_FALSE(token1.empty());
+  EXPECT_NE(token1, token2);
+
+  // Different token is returned for same authorized entity and scope after
+  // deletion.
+  EXPECT_EQ(InstanceID::SUCCESS,
+            DeleteToken(instance_id, kAuthorizedEntity1, kScope1));
+  std::string new_token1 =
+      GetToken(instance_id, kAuthorizedEntity1, kScope2, options);
+  EXPECT_FALSE(new_token1.empty());
+  EXPECT_NE(token1, new_token1);
+
+  // The other token is not affected by the deletion.
+  EXPECT_EQ(token2,
+            GetToken(instance_id, kAuthorizedEntity2, kScope1, options));
+}
+
 }  // instance_id