Render Ash material design battery image icon without PNGs

ash/common/system/chromeos/power/power_status_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 "ash/common/system/chromeos/power/power_status.h"
 
 #include <memory>
 
+#include "ash/common/material_design/material_design_controller.h"
+#include "ash/test/material_design_controller_test_api.h"
 #include "base/message_loop/message_loop.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/cros_system_api/dbus/service_constants.h"
 
 using power_manager::PowerSupplyProperties;
 
 namespace ash {
 namespace {
 
 class TestObserver : public PowerStatus::Observer {
  public:
   TestObserver() : power_changed_count_(0) {}
   ~TestObserver() override {}
 
   int power_changed_count() const { return power_changed_count_; }
 
   // PowerStatus::Observer overrides:
   void OnPowerStatusChanged() override { ++power_changed_count_; }
 
  private:
   int power_changed_count_;
 
   DISALLOW_COPY_AND_ASSIGN(TestObserver);
 };
 
 }  // namespace
 
-class PowerStatusTest : public testing::Test {
+class PowerStatusTest
+    : public testing::Test,
+      public testing::WithParamInterface<MaterialDesignController::Mode> {
  public:
   PowerStatusTest() : power_status_(NULL) {}
   ~PowerStatusTest() override {}
 
   void SetUp() override {
     chromeos::DBusThreadManager::Initialize();
     PowerStatus::Initialize();
     power_status_ = PowerStatus::Get();
     test_observer_.reset(new TestObserver);
     power_status_->AddObserver(test_observer_.get());
+    material_design_state_.reset(
+        new test::MaterialDesignControllerTestAPI(GetParam()));
   }
 
   void TearDown() override {
+    material_design_state_.reset();
     power_status_->RemoveObserver(test_observer_.get());
     test_observer_.reset();
     PowerStatus::Shutdown();
     chromeos::DBusThreadManager::Shutdown();
   }
 
  protected:
   base::MessageLoopForUI message_loop_;
   PowerStatus* power_status_;  // Not owned.
   std::unique_ptr<TestObserver> test_observer_;
 
  private:
+  std::unique_ptr<test::MaterialDesignControllerTestAPI> material_design_state_;
+
   DISALLOW_COPY_AND_ASSIGN(PowerStatusTest);
 };
 
-TEST_F(PowerStatusTest, InitializeAndUpdate) {
+// The prefix has intentionally been left blank since there is only one
+// parameterization of this test fixture.
+INSTANTIATE_TEST_CASE_P(
+    /* prefix intentionally left blank */,
+    PowerStatusTest,
+    testing::Values(MaterialDesignController::NON_MATERIAL,
+                    MaterialDesignController::MATERIAL_NORMAL,
+                    MaterialDesignController::MATERIAL_EXPERIMENTAL));
+
+TEST_P(PowerStatusTest, InitializeAndUpdate) {
   // Test that the initial power supply state should be acquired after
   // PowerStatus is instantiated. This depends on
   // PowerManagerClientStubImpl, which responds to power status update
   // requests, pretends there is a battery present, and generates some valid
   // power supply status data.
   message_loop_.RunUntilIdle();
   EXPECT_EQ(1, test_observer_->power_changed_count());
 
   // Test RequestUpdate, test_obsever_ should be notified for power suuply
   // status change.
   power_status_->RequestStatusUpdate();
   message_loop_.RunUntilIdle();
   EXPECT_EQ(2, test_observer_->power_changed_count());
 }
 
-TEST_F(PowerStatusTest, ShouldDisplayBatteryTime) {
+TEST_P(PowerStatusTest, ShouldDisplayBatteryTime) {
   EXPECT_FALSE(PowerStatus::ShouldDisplayBatteryTime(
       base::TimeDelta::FromSeconds(-1)));
   EXPECT_FALSE(PowerStatus::ShouldDisplayBatteryTime(
       base::TimeDelta::FromSeconds(0)));
   EXPECT_FALSE(PowerStatus::ShouldDisplayBatteryTime(
       base::TimeDelta::FromSeconds(59)));
   EXPECT_TRUE(PowerStatus::ShouldDisplayBatteryTime(
       base::TimeDelta::FromSeconds(60)));
   EXPECT_TRUE(PowerStatus::ShouldDisplayBatteryTime(
       base::TimeDelta::FromSeconds(600)));
   EXPECT_TRUE(PowerStatus::ShouldDisplayBatteryTime(
       base::TimeDelta::FromSeconds(3600)));
   EXPECT_TRUE(PowerStatus::ShouldDisplayBatteryTime(
       base::TimeDelta::FromSeconds(
           PowerStatus::kMaxBatteryTimeToDisplaySec)));
   EXPECT_FALSE(PowerStatus::ShouldDisplayBatteryTime(
       base::TimeDelta::FromSeconds(
           PowerStatus::kMaxBatteryTimeToDisplaySec + 1)));
 }
 
-TEST_F(PowerStatusTest, SplitTimeIntoHoursAndMinutes) {
+TEST_P(PowerStatusTest, SplitTimeIntoHoursAndMinutes) {
   int hours = 0, minutes = 0;
   PowerStatus::SplitTimeIntoHoursAndMinutes(
       base::TimeDelta::FromSeconds(0), &hours, &minutes);
   EXPECT_EQ(0, hours);
   EXPECT_EQ(0, minutes);
 
   PowerStatus::SplitTimeIntoHoursAndMinutes(
       base::TimeDelta::FromSeconds(60), &hours, &minutes);
   EXPECT_EQ(0, hours);
   EXPECT_EQ(1, minutes);
@@ skipping 30 matching lines @@
       base::TimeDelta::FromSecondsD(3599.9), &hours, &minutes);
   EXPECT_EQ(1, hours);
   EXPECT_EQ(0, minutes);
 
   PowerStatus::SplitTimeIntoHoursAndMinutes(
       base::TimeDelta::FromSecondsD(3600.1), &hours, &minutes);
   EXPECT_EQ(1, hours);
   EXPECT_EQ(0, minutes);
 }
 
-TEST_F(PowerStatusTest, GetBatteryImageInfo) {
+TEST_P(PowerStatusTest, GetBatteryImageInfo) {
+  const bool use_md_icon =
+      ash::MaterialDesignController::UseMaterialDesignSystemIcons();
+
   PowerSupplyProperties prop;
   prop.set_external_power(PowerSupplyProperties::AC);
   prop.set_battery_state(PowerSupplyProperties::CHARGING);
   prop.set_battery_percent(98.0);
   power_status_->SetProtoForTesting(prop);
   const PowerStatus::BatteryImageInfo info_charging_98 =
       power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT);
 
   // 99% should use the same icon as 98%.
   prop.set_battery_percent(99.0);
   power_status_->SetProtoForTesting(prop);
   EXPECT_EQ(info_charging_98,
             power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT));
 
-  // The dark icon set should use a different image.
+  // The dark icon set should use a different image for non-MD, but the
+  // same image for MD.
   prop.set_battery_percent(98.0);
-  EXPECT_NE(info_charging_98,
-            power_status_->GetBatteryImageInfo(PowerStatus::ICON_DARK));
+  if (use_md_icon) {
+    EXPECT_EQ(info_charging_98,
+              power_status_->GetBatteryImageInfo(PowerStatus::ICON_DARK));
+  } else {
+    EXPECT_NE(info_charging_98,
+              power_status_->GetBatteryImageInfo(PowerStatus::ICON_DARK));
+  }
 
   // A different icon should be used when the battery is full, too.
   prop.set_battery_state(PowerSupplyProperties::FULL);
   prop.set_battery_percent(100.0);
   power_status_->SetProtoForTesting(prop);
   EXPECT_NE(info_charging_98,
             power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT));
 
   // A much-lower battery level should use a different icon.
   prop.set_battery_state(PowerSupplyProperties::CHARGING);
   prop.set_battery_percent(20.0);
   power_status_->SetProtoForTesting(prop);
   EXPECT_NE(info_charging_98,
             power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT));
 
   // Ditto for 98%, but on USB instead of AC.
   prop.set_external_power(PowerSupplyProperties::USB);
   prop.set_battery_percent(98.0);
   power_status_->SetProtoForTesting(prop);
   EXPECT_NE(info_charging_98,
             power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT));
 }
 
+// Tests that the |icon_badge| member of BatteryImageInfo is set correctly
+// with various power supply property values.
+TEST_P(PowerStatusTest, BatteryImageInfoIconBadge) {
+  // The |icon_badge| member is only populated for the material design
+  // battery icon.
+  if (!ash::MaterialDesignController::UseMaterialDesignSystemIcons())
+    return;
+
+  PowerSupplyProperties prop;
+
+  // A charging battery connected to AC power should have an ICON_BADGE_BOLT.
+  prop.set_external_power(PowerSupplyProperties::AC);
+  prop.set_battery_state(PowerSupplyProperties::CHARGING);
+  prop.set_battery_percent(98.0);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      PowerStatus::ICON_BADGE_BOLT,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).icon_badge);
+
+  // A discharging battery connected to AC should also have an ICON_BADGE_BOLT.
+  prop.set_battery_state(PowerSupplyProperties::DISCHARGING);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      PowerStatus::ICON_BADGE_BOLT,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).icon_badge);
+
+  // A charging battery connected to USB power should have an
+  // ICON_BADGE_UNRELIABLE.
+  prop.set_external_power(PowerSupplyProperties::USB);
+  prop.set_battery_state(PowerSupplyProperties::CHARGING);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      PowerStatus::ICON_BADGE_UNRELIABLE,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).icon_badge);
+
+  // A discharging battery connected to USB power should also have an
+  // ICON_BADGE_UNRELIABLE.
+  prop.set_battery_state(PowerSupplyProperties::DISCHARGING);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      PowerStatus::ICON_BADGE_UNRELIABLE,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).icon_badge);
+
+  // Show an ICON_BADGE_X when no battery is present.
+  prop.set_external_power(PowerSupplyProperties::DISCONNECTED);
+  prop.set_battery_state(PowerSupplyProperties::NOT_PRESENT);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      PowerStatus::ICON_BADGE_X,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).icon_badge);
+
+  // Do not show a badge when the battery is discharging.
+  prop.set_battery_state(PowerSupplyProperties::DISCHARGING);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      PowerStatus::ICON_BADGE_NONE,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).icon_badge);
+
+  // Show ICON_BADGE_ALERT for a discharging battery when it falls below
+  // a charge level of PowerStatus::kCriticalBatteryChargePercentageMd.
+  prop.set_battery_percent(PowerStatus::kCriticalBatteryChargePercentageMd);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      PowerStatus::ICON_BADGE_NONE,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).icon_badge);
+  prop.set_battery_percent(PowerStatus::kCriticalBatteryChargePercentageMd - 1);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      PowerStatus::ICON_BADGE_ALERT,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).icon_badge);
+}
+
+// Tests that the |charge_level| member of BatteryImageInfo is set correctly
+// with various power supply property values.
+TEST_P(PowerStatusTest, BatteryImageInfoChargeLevel) {
+  // The |charge_level| member is only populated for the material design
+  // battery icon.
+  if (!ash::MaterialDesignController::UseMaterialDesignSystemIcons())
+    return;
+
+  PowerSupplyProperties prop;
+
+  // No charge level is drawn when the battery is not present.
+  prop.set_external_power(PowerSupplyProperties::DISCONNECTED);
+  prop.set_battery_state(PowerSupplyProperties::NOT_PRESENT);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      0,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).charge_level);
+
+  // A charge level of 0 when the battery is 0% full.
+  prop.set_external_power(PowerSupplyProperties::AC);
+  prop.set_battery_state(PowerSupplyProperties::CHARGING);
+  prop.set_battery_percent(0.0);
+  EXPECT_EQ(
+      0,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).charge_level);
+
+  // A charge level of 1 when the battery is up to 16% full, and a level of 2
+  // for 17% full.
+  prop.set_battery_percent(16.0);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      1,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).charge_level);
+  prop.set_battery_percent(17.0);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      2,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).charge_level);
+
+  // A charge level of 6 when the battery is 50% full.
+  prop.set_battery_percent(50.0);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      6,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).charge_level);
+
+  // A charge level of 11 when the battery is 99% full, and a level of 12 when
+  // the battery is 100% full.
+  prop.set_battery_percent(99.0);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      11,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).charge_level);
+  prop.set_battery_percent(100.0);
+  power_status_->SetProtoForTesting(prop);
+  EXPECT_EQ(
+      12,
+      power_status_->GetBatteryImageInfo(PowerStatus::ICON_LIGHT).charge_level);
+}
+
 }  // namespace ash