OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2012 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 "chrome/browser/signin/signin_manager_base.h" |
| 6 |
| 7 #include "base/compiler_specific.h" |
| 8 #include "base/prefs/pref_service.h" |
| 9 #include "base/prefs/testing_pref_service.h" |
| 10 #include "base/stringprintf.h" |
| 11 #include "chrome/browser/browser_process.h" |
| 12 #include "chrome/browser/prefs/browser_prefs.h" |
| 13 #include "chrome/browser/signin/token_service.h" |
| 14 #include "chrome/browser/signin/token_service_unittest.h" |
| 15 #include "chrome/common/pref_names.h" |
| 16 #include "chrome/common/url_constants.h" |
| 17 #include "chrome/test/base/testing_browser_process.h" |
| 18 #include "chrome/test/base/testing_profile.h" |
| 19 |
| 20 #include "testing/gmock/include/gmock/gmock.h" |
| 21 #include "testing/gtest/include/gtest/gtest.h" |
| 22 |
| 23 class SigninManagerBaseTest : public TokenServiceTestHarness { |
| 24 public: |
| 25 virtual void SetUp() OVERRIDE { |
| 26 prefs_.reset(new TestingPrefServiceSimple); |
| 27 chrome::RegisterLocalState(prefs_->registry()); |
| 28 TestingBrowserProcess::GetGlobal()->SetLocalState( |
| 29 prefs_.get()); |
| 30 TokenServiceTestHarness::SetUp(); |
| 31 manager_.reset(new SigninManagerBase()); |
| 32 } |
| 33 |
| 34 virtual void TearDown() OVERRIDE { |
| 35 // Destroy the SigninManager here, because it relies on profile_ which is |
| 36 // freed in the base class. |
| 37 manager_->Shutdown(); |
| 38 manager_.reset(NULL); |
| 39 TestingBrowserProcess::GetGlobal()->SetLocalState(NULL); |
| 40 prefs_.reset(NULL); |
| 41 TokenServiceTestHarness::TearDown(); |
| 42 } |
| 43 |
| 44 scoped_ptr<SigninManagerBase> manager_; |
| 45 scoped_ptr<TestingPrefServiceSimple> prefs_; |
| 46 }; |
| 47 |
| 48 TEST_F(SigninManagerBaseTest, Prohibited) { |
| 49 g_browser_process->local_state()->SetString( |
| 50 prefs::kGoogleServicesUsernamePattern, ".*@google.com"); |
| 51 manager_->Initialize(profile_.get()); |
| 52 EXPECT_TRUE(manager_->IsAllowedUsername("test@google.com")); |
| 53 EXPECT_TRUE(manager_->IsAllowedUsername("happy@google.com")); |
| 54 EXPECT_FALSE(manager_->IsAllowedUsername("test@invalid.com")); |
| 55 EXPECT_FALSE(manager_->IsAllowedUsername("test@notgoogle.com")); |
| 56 EXPECT_FALSE(manager_->IsAllowedUsername(std::string())); |
| 57 } |
| 58 |
| 59 TEST_F(SigninManagerBaseTest, TestAlternateWildcard) { |
| 60 // Test to make sure we accept "*@google.com" as a pattern (treat it as if |
| 61 // the admin entered ".*@google.com"). |
| 62 g_browser_process->local_state()->SetString( |
| 63 prefs::kGoogleServicesUsernamePattern, "*@google.com"); |
| 64 manager_->Initialize(profile_.get()); |
| 65 EXPECT_TRUE(manager_->IsAllowedUsername("test@google.com")); |
| 66 EXPECT_TRUE(manager_->IsAllowedUsername("happy@google.com")); |
| 67 EXPECT_FALSE(manager_->IsAllowedUsername("test@invalid.com")); |
| 68 EXPECT_FALSE(manager_->IsAllowedUsername("test@notgoogle.com")); |
| 69 EXPECT_FALSE(manager_->IsAllowedUsername(std::string())); |
| 70 } |
| 71 |
| 72 TEST_F(SigninManagerBaseTest, ProhibitedAtStartup) { |
| 73 profile_->GetPrefs()->SetString(prefs::kGoogleServicesUsername, |
| 74 "monkey@invalid.com"); |
| 75 g_browser_process->local_state()->SetString( |
| 76 prefs::kGoogleServicesUsernamePattern, ".*@google.com"); |
| 77 manager_->Initialize(profile_.get()); |
| 78 // Currently signed in user is prohibited by policy, so should be signed out. |
| 79 EXPECT_EQ("", manager_->GetAuthenticatedUsername()); |
| 80 } |
| 81 |
| 82 TEST_F(SigninManagerBaseTest, ProhibitedAfterStartup) { |
| 83 std::string user("monkey@invalid.com"); |
| 84 profile_->GetPrefs()->SetString(prefs::kGoogleServicesUsername, user); |
| 85 manager_->Initialize(profile_.get()); |
| 86 EXPECT_EQ(user, manager_->GetAuthenticatedUsername()); |
| 87 // Update the profile - user should be signed out. |
| 88 g_browser_process->local_state()->SetString( |
| 89 prefs::kGoogleServicesUsernamePattern, ".*@google.com"); |
| 90 EXPECT_EQ("", manager_->GetAuthenticatedUsername()); |
| 91 } |
| 92 |
OLD | NEW |