Add support for kiosk mode on the client. Make sure the settings are written in the lockbox.

chrome/browser/chromeos/login/login_utils_browsertest.cc

 // 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/login/login_utils.h"
 
 #include "base/basictypes.h"
 #include "base/command_line.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/scoped_temp_dir.h"
 #include "base/string_util.h"
 #include "chrome/browser/chromeos/cros/cros_library.h"
 #include "chrome/browser/chromeos/cros/mock_cryptohome_library.h"
 #include "chrome/browser/chromeos/cros/mock_library_loader.h"
 #include "chrome/browser/chromeos/dbus/mock_dbus_thread_manager.h"
 #include "chrome/browser/chromeos/dbus/mock_session_manager_client.h"
 #include "chrome/browser/chromeos/login/authenticator.h"
 #include "chrome/browser/chromeos/login/login_status_consumer.h"
 #include "chrome/browser/chromeos/login/user_manager.h"
 #include "chrome/browser/io_thread.h"
 #include "chrome/browser/net/predictor.h"
 #include "chrome/browser/policy/browser_policy_connector.h"
+#include "chrome/browser/policy/cloud_policy_data_store.h"
 #include "chrome/browser/policy/proto/device_management_backend.pb.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/net/gaia/gaia_urls.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/net/gaia/gaia_auth_consumer.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_pref_service.h"
 #include "content/public/browser/browser_thread.h"
@@ skipping 13 matching lines @@
 
 using ::testing::DoAll;
 using ::testing::Return;
 using ::testing::SetArgPointee;
 using ::testing::_;
 using content::BrowserThread;
 
 const char kTrue[] = "true";
 const char kDomain[] = "domain.com";
 const char kUsername[] = "user@domain.com";
+const char kMode[] = "enterprise";
+const char kDeviceId[] = "100200300";
 const char kUsernameOtherDomain[] = "user@other.com";
 const char kAttributeOwned[] = "enterprise.owned";
 const char kAttributeOwner[] = "enterprise.user";
+const char kAttrEnterpriseDomain[] = "enterprise.domain";
+const char kAttrEnterpriseMode[] = "enterprise.mode";
+const char kAttrEnterpriseDeviceId[] = "enterprise.device_id";
 
 const char kOAuthTokenCookie[] = "oauth_token=1234";
 const char kOAuthGetAccessTokenData[] =
     "oauth_token=1234&oauth_token_secret=1234";
 const char kOAuthServiceTokenData[] =
     "wrap_access_token=1234&wrap_access_token_expires_in=123456789";
 
 const char kDMServer[] = "http://server/device_management";
 const char kDMRegisterRequest[] =
     "http://server/device_management?request=register";
@@ skipping 73 matching lines @@
         .WillRepeatedly(Return(true));
     EXPECT_CALL(*cryptohome_, TpmIsEnabled())
         .WillRepeatedly(Return(false));
     EXPECT_CALL(*cryptohome_, IsMounted())
         .WillRepeatedly(Return(true));
     EXPECT_CALL(*cryptohome_, InstallAttributesSet(kAttributeOwned, kTrue))
         .WillRepeatedly(Return(true));
     EXPECT_CALL(*cryptohome_, InstallAttributesSet(kAttributeOwner,
                                                    kUsername))
         .WillRepeatedly(Return(true));
+    EXPECT_CALL(*cryptohome_, InstallAttributesSet(kAttrEnterpriseDomain,
+                                                   kDomain))
+        .WillRepeatedly(Return(true));
+    EXPECT_CALL(*cryptohome_, InstallAttributesSet(kAttrEnterpriseMode,
+                                                   kMode))
+        .WillRepeatedly(Return(true));
+    EXPECT_CALL(*cryptohome_, InstallAttributesSet(kAttrEnterpriseDeviceId,
+                                                   kDeviceId))
+        .WillRepeatedly(Return(true));
     EXPECT_CALL(*cryptohome_, InstallAttributesFinalize())
         .WillRepeatedly(Return(true));
     EXPECT_CALL(*cryptohome_, InstallAttributesGet(kAttributeOwned, _))
         .WillRepeatedly(DoAll(SetArgPointee<1>(kTrue),
                               Return(true)));
     EXPECT_CALL(*cryptohome_, InstallAttributesGet(kAttributeOwner, _))
         .WillRepeatedly(DoAll(SetArgPointee<1>(kUsername),
                               Return(true)));
+    EXPECT_CALL(*cryptohome_, InstallAttributesGet(kAttrEnterpriseDomain, _))
+        .WillRepeatedly(DoAll(SetArgPointee<1>(kDomain),
+                              Return(true)));
+    EXPECT_CALL(*cryptohome_, InstallAttributesGet(kAttrEnterpriseMode, _))
+        .WillRepeatedly(DoAll(SetArgPointee<1>(kMode),
+                              Return(true)));
+    EXPECT_CALL(*cryptohome_, InstallAttributesGet(kAttrEnterpriseDeviceId, _))
+        .WillRepeatedly(DoAll(SetArgPointee<1>(kDeviceId),
+                              Return(true)));
     test_api->SetCryptohomeLibrary(cryptohome_, true);
 
     browser_process_->SetProfileManager(
         new ProfileManagerWithoutInit(scoped_temp_dir_.path()));
     connector_ = browser_process_->browser_policy_connector();
     connector_->Init();
 
     loop_.RunAllPending();
   }
 
@@ skipping 52 matching lines @@
                               const GaiaAuthConsumer::ClientLoginResult& creds,
                               bool pending_requests,
                               bool using_oauth) OVERRIDE {
     FAIL() << "OnLoginSuccess not expected";
   }
 
   void LockDevice(const std::string& username) {
     EXPECT_CALL(*cryptohome_, InstallAttributesIsFirstInstall())
         .WillOnce(Return(true))
         .WillRepeatedly(Return(false));
+    policy::CloudPolicyDataStore* device_data_store =
+        const_cast<policy::CloudPolicyDataStore*>(
+            connector_->GetDeviceCloudPolicyDataStore());

[Mattias Nissler (ping if slow), 2012/02/20 12:22:54]

can't you just call connector_->LockDevice()?

[pastarmovj, 2012/02/20 14:26:30]

Done. As discussed offline made the accessors non-const.

+    device_data_store->set_device_mode(
+        policy::CloudPolicyDataStore::DEVICE_MODE_ENTERPRISE);
+    device_data_store->set_device_id(kDeviceId);
     EXPECT_EQ(policy::EnterpriseInstallAttributes::LOCK_SUCCESS,
               connector_->LockDevice(username));
     loop_.RunAllPending();
   }
 
   void PrepareProfile(const std::string& username) {
     MockSessionManagerClient* session_managed_client =
         mock_dbus_thread_manager_.mock_session_manager_client();
     EXPECT_CALL(*session_managed_client, StartSession(_));
     EXPECT_CALL(*cryptohome_, GetSystemSalt())
@@ skipping 42 matching lines @@
     EXPECT_TRUE(response.SerializeToString(&data));
     fetcher->SetResponseString(data);
     return fetcher;
   }
 
   TestURLFetcher* PrepareDMRegisterFetcher() {
     em::DeviceManagementResponse response;
     em::DeviceRegisterResponse* register_response =
         response.mutable_register_response();
     register_response->set_device_management_token(kDMToken);
+    register_response->set_enrollment_type(
+        em::DeviceRegisterResponse::ENTERPRISE);
     return PrepareDMServiceFetcher(kDMRegisterRequest, response);
   }
 
   TestURLFetcher* PrepareDMPolicyFetcher() {
     em::DeviceManagementResponse response;
     response.mutable_policy_response()->add_response();
     return PrepareDMServiceFetcher(kDMPolicyRequest, response);
   }
 
  protected:
@@ skipping 150 matching lines @@
 }
 
 INSTANTIATE_TEST_CASE_P(
     LoginUtilsBlockingLoginTestInstance,
     LoginUtilsBlockingLoginTest,
     testing::Values(0, 1, 2, 3, 4, 5));
 
 }  // namespace
 
 }