1 files changed, 38 insertions, 0 deletions
diff --git a/chromium/extensions/shell/browser/shell_oauth2_token_service_unittest.cc b/chromium/extensions/shell/browser/shell_oauth2_token_service_unittest.cc
new file mode 100644
index 00000000000..649fd3abd54
--- /dev/null
+++ b/chromium/extensions/shell/browser/shell_oauth2_token_service_unittest.cc
@@ -0,0 +1,38 @@
+// Copyright 2014 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 "extensions/shell/browser/shell_oauth2_token_service.h"
+
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "extensions/browser/extensions_test.h"
+
+namespace extensions {
+
+class ShellOAuth2TokenServiceTest : public ExtensionsTest {
+ public:
+  ShellOAuth2TokenServiceTest() {}
+  ~ShellOAuth2TokenServiceTest() override {}
+
+ private:
+  content::TestBrowserThreadBundle thread_bundle_;
+};
+
+// Verifies setting the refresh token makes it available.
+TEST_F(ShellOAuth2TokenServiceTest, SetRefreshToken) {
+  ShellOAuth2TokenService service(nullptr, "larry@google.com", "token123");
+
+  // Only has a token for the account in the constructor.
+  EXPECT_EQ("larry@google.com", service.AccountId());
+  EXPECT_TRUE(service.RefreshTokenIsAvailable("larry@google.com"));
+  EXPECT_FALSE(service.RefreshTokenIsAvailable("sergey@google.com"));
+
+  service.SetRefreshToken("sergey@google.com", "token456");
+
+  // The account and refresh token have updated.
+  EXPECT_EQ("sergey@google.com", service.AccountId());
+  EXPECT_FALSE(service.RefreshTokenIsAvailable("larry@google.com"));
+  EXPECT_TRUE(service.RefreshTokenIsAvailable("sergey@google.com"));
+}
+
+}  // namespace extensions