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/chromeos/drive/drive_integration_service.h" 6 7#include "base/prefs/pref_service.h" 8#include "chrome/browser/profiles/profile.h" 9#include "chrome/browser/ui/browser.h" 10#include "chrome/common/pref_names.h" 11#include "chrome/test/base/in_process_browser_test.h" 12 13namespace drive { 14 15class DriveIntegrationServiceBrowserTest : public InProcessBrowserTest { 16}; 17 18// Verify DriveIntegrationService is created during login. 19IN_PROC_BROWSER_TEST_F(DriveIntegrationServiceBrowserTest, 20 CreatedDuringLogin) { 21 EXPECT_TRUE(DriveIntegrationServiceFactory::FindForProfile( 22 browser()->profile())); 23} 24 25 26IN_PROC_BROWSER_TEST_F(DriveIntegrationServiceBrowserTest, 27 DisableDrivePolicyTest) { 28 // First make sure the pref is set to its default value which should permit 29 // drive. 30 browser()->profile()->GetPrefs()->SetBoolean(prefs::kDisableDrive, false); 31 32 drive::DriveIntegrationService* integration_service = 33 drive::DriveIntegrationServiceFactory::FindForProfile( 34 browser()->profile()); 35 36 EXPECT_TRUE(integration_service); 37 EXPECT_TRUE(integration_service->is_enabled()); 38 39 // ...next try to disable drive. 40 browser()->profile()->GetPrefs()->SetBoolean(prefs::kDisableDrive, true); 41 42 EXPECT_EQ(integration_service, 43 drive::DriveIntegrationServiceFactory::FindForProfile( 44 browser()->profile())); 45 EXPECT_FALSE(integration_service->is_enabled()); 46} 47 48} // namespace drive 49