| 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 |