components/proximity_auth/cryptauth/cryptauth_account_token_fetcher_unittest.cc

   1 // Copyright 2014 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 "components/proximity_auth/cryptauth/cryptauth_account_token_fetcher.h"
   6
   7 #include <string>
   8
   9 #include "base/bind.h"
  10 #include "google_apis/gaia/fake_oauth2_token_service.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 namespace proximity_auth {
  14
  15 namespace {
  16
  17 const char kAccountId[] = "account_id";
  18 const char kAccessToken[] = "access_token";
  19 const char kInvalidResult[] = "invalid_result";
  20
  21 // Callback that saves the fetched access token to the first argument.
  22 void SaveAccessToken(std::string* out_token, const std::string& in_token) {
  23   *out_token = in_token;
  24 }
  25
  26 }  // namespace
  27
  28 class ProximityAuthCryptAuthAccountTokenFetcherTest : public testing::Test {
  29  protected:
  30   ProximityAuthCryptAuthAccountTokenFetcherTest()
  31       : fetcher_(&token_service_, kAccountId) {
  32     token_service_.AddAccount(kAccountId);
  33   }
  34
  35   FakeOAuth2TokenService token_service_;
  36   CryptAuthAccountTokenFetcher fetcher_;
  37
  38   DISALLOW_COPY_AND_ASSIGN(ProximityAuthCryptAuthAccountTokenFetcherTest);
  39 };
  40
  41 TEST_F(ProximityAuthCryptAuthAccountTokenFetcherTest, FetchSuccess) {
  42   std::string result;
  43   fetcher_.FetchAccessToken(base::Bind(SaveAccessToken, &result));
  44   token_service_.IssueAllTokensForAccount(kAccountId, kAccessToken,
  45                                           base::Time::Max());
  46
  47   EXPECT_EQ(kAccessToken, result);
  48 }
  49
  50 TEST_F(ProximityAuthCryptAuthAccountTokenFetcherTest, FetchFailure) {
  51   std::string result(kInvalidResult);
  52   fetcher_.FetchAccessToken(base::Bind(SaveAccessToken, &result));
  53   token_service_.IssueErrorForAllPendingRequestsForAccount(
  54       kAccountId,
  55       GoogleServiceAuthError(GoogleServiceAuthError::SERVICE_ERROR));
  56
  57   EXPECT_EQ(std::string(), result);
  58 }
  59
  60 TEST_F(ProximityAuthCryptAuthAccountTokenFetcherTest, FetcherReuse) {
  61   std::string result1;
  62   fetcher_.FetchAccessToken(base::Bind(SaveAccessToken, &result1));
  63
  64   {
  65     std::string result2(kInvalidResult);
  66     fetcher_.FetchAccessToken(base::Bind(SaveAccessToken, &result2));
  67     EXPECT_EQ(std::string(), result2);
  68   }
  69
  70   token_service_.IssueAllTokensForAccount(kAccountId, kAccessToken,
  71                                           base::Time::Max());
  72   EXPECT_EQ(kAccessToken, result1);
  73 }
  74
  75 }  // namespace proximity_auth