chromeos: Remove PowerStateOverride.

chromeos/dbus/power_policy_controller_unittest.cc

+// 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.
+
+#include "chromeos/dbus/power_policy_controller.h"
+
+#include "chromeos/dbus/dbus_thread_manager.h"
+#include "chromeos/dbus/mock_dbus_thread_manager.h"
+#include "chromeos/dbus/mock_power_manager_client.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using ::testing::_;
+using ::testing::SaveArg;
+
+namespace chromeos {
+
+class PowerPolicyControllerTest : public testing::Test {
+ public:
+  PowerPolicyControllerTest() {}
+  virtual ~PowerPolicyControllerTest() {}
+
+  virtual void SetUp() OVERRIDE {
+    dbus_manager_ = new MockDBusThreadManager;
+    DBusThreadManager::InitializeForTesting(dbus_manager_);  // Takes ownership.
+    power_client_ = dbus_manager_->mock_power_manager_client();
+    EXPECT_CALL(*power_client_, SetPolicy(_))
+        .WillRepeatedly(SaveArg<0>(&last_policy_));
+
+    policy_controller_ = dbus_manager_->GetPowerPolicyController();
+
+    // TODO(derat): Write what looks like it will be a ridiculously large
+    // amount of code to register prefs so that UpdatePolicyFromPrefs() can
+    // be tested.
+  }
+
+  virtual void TearDown() OVERRIDE {
+    DBusThreadManager::Shutdown();
+  }
+
+ protected:
+  MockDBusThreadManager* dbus_manager_;  // Not owned.
+  MockPowerManagerClient* power_client_;  // Not owned.
+  PowerPolicyController* policy_controller_;  // Not owned.
+
+  power_manager::PowerManagementPolicy last_policy_;
+};
+
+TEST_F(PowerPolicyControllerTest, Blocks) {
+  const char kSuspendBlockReason[] = "suspend";
+  const int suspend_id =
+      policy_controller_->AddSuspendBlock(kSuspendBlockReason);
+  power_manager::PowerManagementPolicy expected_policy;
+  expected_policy.set_idle_action(
+      power_manager::PowerManagementPolicy_Action_DO_NOTHING);
+  expected_policy.set_reason(kSuspendBlockReason);
+  EXPECT_EQ(expected_policy.SerializeAsString(),
+            last_policy_.SerializeAsString());
+
+  const char kScreenBlockReason[] = "screen";
+  const int screen_id = policy_controller_->AddScreenBlock(kScreenBlockReason);
+  expected_policy.mutable_ac_delays()->set_screen_dim_ms(0);
+  expected_policy.mutable_ac_delays()->set_screen_off_ms(0);
+  expected_policy.mutable_battery_delays()->set_screen_dim_ms(0);
+  expected_policy.mutable_battery_delays()->set_screen_off_ms(0);
+  expected_policy.set_reason(
+      std::string(kScreenBlockReason) + ", " + kSuspendBlockReason);
+  EXPECT_EQ(expected_policy.SerializeAsString(),
+            last_policy_.SerializeAsString());
+
+  policy_controller_->RemoveBlock(suspend_id);
+  expected_policy.set_reason(kScreenBlockReason);
+  EXPECT_EQ(expected_policy.SerializeAsString(),
+            last_policy_.SerializeAsString());
+
+  policy_controller_->RemoveBlock(screen_id);
+  expected_policy.Clear();
+  EXPECT_EQ(expected_policy.SerializeAsString(),
+            last_policy_.SerializeAsString());
+}
+
+}  // namespace chromeos