| // Copyright (c) 2012 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 "chrome/browser/chromeos/drive/drive_integration_service.h" |
| |
| #include "base/prefs/pref_service.h" |
| #include "chrome/browser/profiles/profile.h" |
| #include "chrome/browser/ui/browser.h" |
| #include "chrome/test/base/in_process_browser_test.h" |
| #include "components/drive/drive_pref_names.h" |
| |
| namespace drive { |
| |
| class DriveIntegrationServiceBrowserTest : public InProcessBrowserTest { |
| }; |
| |
| // Verify DriveIntegrationService is created during login. |
| IN_PROC_BROWSER_TEST_F(DriveIntegrationServiceBrowserTest, |
| CreatedDuringLogin) { |
| EXPECT_TRUE(DriveIntegrationServiceFactory::FindForProfile( |
| browser()->profile())); |
| } |
| |
| |
| IN_PROC_BROWSER_TEST_F(DriveIntegrationServiceBrowserTest, |
| DisableDrivePolicyTest) { |
| // First make sure the pref is set to its default value which should permit |
| // drive. |
| browser()->profile()->GetPrefs()->SetBoolean(prefs::kDisableDrive, false); |
| |
| drive::DriveIntegrationService* integration_service = |
| drive::DriveIntegrationServiceFactory::FindForProfile( |
| browser()->profile()); |
| |
| EXPECT_TRUE(integration_service); |
| EXPECT_TRUE(integration_service->is_enabled()); |
| |
| // ...next try to disable drive. |
| browser()->profile()->GetPrefs()->SetBoolean(prefs::kDisableDrive, true); |
| |
| EXPECT_EQ(integration_service, |
| drive::DriveIntegrationServiceFactory::FindForProfile( |
| browser()->profile())); |
| EXPECT_FALSE(integration_service->is_enabled()); |
| } |
| |
| } // namespace drive |