drive_integration_service_browsertest.cc revision 90dce4d38c5ff5333bea97d859d4e484e27edf0c 
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::GetForProfile(
34          browser()->profile());
35
36  EXPECT_TRUE(integration_service);
37
38  // ...next try to disable drive.
39  browser()->profile()->GetPrefs()->SetBoolean(prefs::kDisableDrive, true);
40
41  integration_service =
42      drive::DriveIntegrationServiceFactory::GetForProfile(
43          browser()->profile());
44
45  EXPECT_FALSE(integration_service);
46}
47
48}  // namespace drive
49