Encapsulate data-format constants in EnterpriseInstallAttributes.

chrome/browser/chromeos/policy/enterprise_install_attributes.h

 // 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.
 
 #ifndef CHROME_BROWSER_CHROMEOS_POLICY_ENTERPRISE_INSTALL_ATTRIBUTES_H_
 #define CHROME_BROWSER_CHROMEOS_POLICY_ENTERPRISE_INSTALL_ATTRIBUTES_H_
 
 #include <map>
 #include <string>
 
 #include "base/basictypes.h"
 #include "base/callback.h"
 #include "base/compiler_specific.h"
 #include "base/files/file_path.h"
+#include "base/gtest_prod_util.h"
 #include "base/memory/weak_ptr.h"
 #include "chromeos/dbus/cryptohome_client.h"
 #include "chromeos/dbus/dbus_method_call_status.h"
 #include "components/policy/core/common/cloud/cloud_policy_constants.h"
 
 namespace policy {
 
 // Brokers access to the enterprise-related installation-time attributes on
 // ChromeOS.
 // TODO(zelidrag, mnissler): Rename + move this class - http://crbug.com/249513.
 class EnterpriseInstallAttributes {
  public:
   // Return codes for LockDevice().
   enum LockResult {
     LOCK_SUCCESS,
     LOCK_NOT_READY,
     LOCK_BACKEND_ERROR,
     LOCK_WRONG_USER,
   };
 
   // A callback to handle responses of methods returning a LockResult value.
   typedef base::Callback<void(LockResult lock_result)> LockResultCallback;
 
-  // Constants for the possible device modes that can be stored in the lockbox.
-  static const char kConsumerDeviceMode[];
-  static const char kEnterpriseDeviceMode[];
-  static const char kRetailKioskDeviceMode[];
-  static const char kConsumerKioskDeviceMode[];
-  static const char kUnknownDeviceMode[];
-
-  // Field names in the lockbox.
-  static const char kAttrEnterpriseDeviceId[];
-  static const char kAttrEnterpriseDomain[];
-  static const char kAttrEnterpriseMode[];
-  static const char kAttrEnterpriseOwned[];
-  static const char kAttrEnterpriseUser[];
-  static const char kAttrConsumerKioskEnabled[];
+  // Return serialized InstallAttributes of an enterprise-owned configuration.
+  static std::string GetEnterpriseOwnedInstallAttributesBlobForTesting(
+      const std::string& user_name);
 
   explicit EnterpriseInstallAttributes(
       chromeos::CryptohomeClient* cryptohome_client);
   ~EnterpriseInstallAttributes();
 
   // Reads data from the cache file which is created early during the boot
   // process.  The cache file is used to work around slow cryptohome startup,
   // which takes a while to register its DBus interface.  See
   // http://crosbug.com/37367 for background on this.
   void ReadCacheFile(const base::FilePath& cache_file);
@@ skipping 37 matching lines @@
   DeviceMode GetMode();
 
  protected:
   bool device_locked_;
   std::string registration_user_;
   std::string registration_domain_;
   std::string registration_device_id_;
   DeviceMode registration_mode_;
 
  private:
+  FRIEND_TEST_ALL_PREFIXES(EnterpriseInstallAttributesTest,
+                           DeviceLockedFromOlderVersion);
+  FRIEND_TEST_ALL_PREFIXES(EnterpriseInstallAttributesTest,
+                           ReadCacheFile);
+  FRIEND_TEST_ALL_PREFIXES(EnterpriseInstallAttributesTest,
+                           ReadCacheFileForConsumerKiosk);
+  FRIEND_TEST_ALL_PREFIXES(EnterpriseInstallAttributesTest,
+                           VerifyFakeInstallAttributesCache);
+
+  // Constants for the possible device modes that can be stored in the lockbox.
+  static const char kConsumerDeviceMode[];
+  static const char kEnterpriseDeviceMode[];
+  static const char kRetailKioskDeviceMode[];
+  static const char kConsumerKioskDeviceMode[];
+  static const char kUnknownDeviceMode[];
+
+  // Field names in the lockbox.
+  static const char kAttrEnterpriseDeviceId[];
+  static const char kAttrEnterpriseDomain[];
+  static const char kAttrEnterpriseMode[];
+  static const char kAttrEnterpriseOwned[];
+  static const char kAttrEnterpriseUser[];
+  static const char kAttrConsumerKioskEnabled[];
+
+  // Translates DeviceMode constants to strings used in the lockbox.
+  std::string GetDeviceModeString(DeviceMode mode);
+
+  // Translates strings used in the lockbox to DeviceMode values.
+  DeviceMode GetDeviceModeFromString(const std::string& mode);
+
   // Decodes the install attributes provided in |attr_map|.
   void DecodeInstallAttributes(
       const std::map<std::string, std::string>& attr_map);
 
   // Helper for ReadImmutableAttributes.
   void ReadAttributesIfReady(
       const base::Closure& callback,
       chromeos::DBusMethodCallStatus call_status,
       bool result);
 
@@ skipping 14 matching lines @@
   chromeos::CryptohomeClient* cryptohome_client_;
 
   base::WeakPtrFactory<EnterpriseInstallAttributes> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(EnterpriseInstallAttributes);
 };
 
 }  // namespace policy
 
 #endif  // CHROME_BROWSER_CHROMEOS_POLICY_ENTERPRISE_INSTALL_ATTRIBUTES_H_