extensions/shell/browser/shell_oauth2_token_service_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 "extensions/shell/browser/shell_oauth2_token_service.h"
   6
   7 #include "content/public/test/test_browser_thread_bundle.h"
   8 #include "extensions/browser/extensions_test.h"
   9
  10 namespace extensions {
  11
  12 class ShellOAuth2TokenServiceTest : public ExtensionsTest {
  13  public:
  14   ShellOAuth2TokenServiceTest() {}
  15   ~ShellOAuth2TokenServiceTest() override {}
  16
  17  private:
  18   content::TestBrowserThreadBundle thread_bundle_;
  19 };
  20
  21 // Verifies setting the refresh token makes it available.
  22 TEST_F(ShellOAuth2TokenServiceTest, SetRefreshToken) {
  23   ShellOAuth2TokenService service(nullptr, "larry@google.com", "token123");
  24
  25   // Only has a token for the account in the constructor.
  26   EXPECT_EQ("larry@google.com", service.account_id());
  27   EXPECT_TRUE(service.RefreshTokenIsAvailable("larry@google.com"));
  28   EXPECT_FALSE(service.RefreshTokenIsAvailable("sergey@google.com"));
  29
  30   service.SetRefreshToken("sergey@google.com", "token456");
  31
  32   // The account and refresh token have updated.
  33   EXPECT_EQ("sergey@google.com", service.account_id());
  34   EXPECT_FALSE(service.RefreshTokenIsAvailable("larry@google.com"));
  35   EXPECT_TRUE(service.RefreshTokenIsAvailable("sergey@google.com"));
  36 }
  37
  38 }  // namespace extensions