Support VariationsRestrictParameter in VariationsService for Chrome OS

chrome/browser/chromeos/policy/device_policy_cros_browsertest.h

+// Copyright (c) 2013 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_DEVICE_POLICY_CROS_BROWSERTEST_H_

[Alexei Svitkine (slow), 2013/04/19 18:11:55]

Rename this to device_policy_cros_browser_test.h/cc, similar to
cros_in_process_browser_test.h. (underscore between 'browser' and 'test')

(Since this isn't actually a browsertest test file, but a base class for other
tests.)

[Mathieu, 2013/04/19 18:37:31]

Done.

+#define CHROME_BROWSER_CHROMEOS_POLICY_DEVICE_POLICY_CROS_BROWSERTEST_H_
+
+#include "base/files/scoped_temp_dir.h"
+#include "chrome/browser/chromeos/cros/cros_in_process_browser_test.h"
+#include "chrome/browser/chromeos/policy/device_policy_builder.h"
+#include "chrome/browser/chromeos/policy/proto/chrome_device_policy.pb.h"
+#include "chrome/browser/metrics/variations/variations_service.h"
+#include "chrome/common/chrome_paths.h"
+#include "chrome/test/base/testing_browser_process.h"
+#include "chromeos/dbus/fake_session_manager_client.h"
+#include "chromeos/dbus/mock_dbus_thread_manager.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace policy {
+
+// Used to test Device policy changes in Chrome OS.
+class DevicePolicyCrosBrowserTest :
+    public chromeos::CrosInProcessBrowserTest {
+ protected:
+  DevicePolicyCrosBrowserTest();
+  virtual ~DevicePolicyCrosBrowserTest() {}

[Alexei Svitkine (slow), 2013/04/19 18:11:55]

Move the implementation to the .cc file.

[Mathieu, 2013/04/19 18:37:31]

Done.

+
+  virtual void SetUpInProcessBrowserTestFixture() OVERRIDE;
+
+  virtual void SetMockDBusThreadManagerExpectations();
+
+  virtual void SetUpSessionManager();
+
+  virtual void SetUpAdditionalCrosMocks() {}
+
+  virtual void SetSpecificDevicePolicies() {}
+
+  virtual void SetDeviceLocalAccountPolicy() {}
+
+  void RefreshDevicePolicy();
+
+  virtual void TearDownInProcessBrowserTestFixture() OVERRIDE;
+
+  BrowserProcess* browser_process() { return g_browser_process; }
+
+  chromeos::MockDBusThreadManager* mock_dbus_thread_manager() {
+    return mock_dbus_thread_manager_;
+  }
+
+  chromeos::FakeSessionManagerClient* session_manager_client() {
+    return &session_manager_client_;
+  }
+
+  DevicePolicyBuilder* device_policy() { return &device_policy_; }
+
+ private:
+  // Install the owner key.
+  void InstallOwnerKey();
+
+  chromeos::MockDBusThreadManager* mock_dbus_thread_manager_;
+
+  // Mock out policy loads/stores from/to the device.
+  chromeos::FakeSessionManagerClient session_manager_client_;
+
+  // Carries Chrome OS device policies for tests.
+  DevicePolicyBuilder device_policy_;
+
+  // Stores the device owner key.
+  base::ScopedTempDir temp_dir_;
+
+  DISALLOW_COPY_AND_ASSIGN(DevicePolicyCrosBrowserTest);
+};
+
+}  // namespace policy
+
+#endif  // CHROME_BROWSER_CHROMEOS_POLICY_DEVICE_POLICY_CROS_BROWSERTEST_H_