rmad/utils/sys_utils_test.cc

// Copyright 2021 The Chromium OS 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 "rmad/utils/sys_utils_impl.h"
#include "rmad/utils/fake_sys_utils.h"

#include <memory>

#include <base/files/file_path.h>
#include <base/files/file_util.h>
#include <base/files/scoped_temp_dir.h>
#include <brillo/file_utils.h>
#include <gtest/gtest.h>

#include "rmad/constants.h"

namespace rmad {

class SysUtilsTest : public testing::Test {
 public:
  SysUtilsTest() = default;
  ~SysUtilsTest() override = default;

 protected:
  void SetUp() override {
    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
    sys_utils_ = std::make_unique<SysUtilsImpl>(temp_dir_.GetPath());
  }

  base::ScopedTempDir temp_dir_;
  std::unique_ptr<SysUtils> sys_utils_;
};

TEST_F(SysUtilsTest, IsPowerSourcePresent_Present) {
  // Create power supply files.
  base::FilePath supply_path =
      temp_dir_.GetPath().AppendASCII("class/power_supply/1");
  EXPECT_TRUE(base::CreateDirectory(supply_path));
  EXPECT_TRUE(base::WriteFile(supply_path.AppendASCII("type"), "USB_PD"));
  EXPECT_TRUE(base::WriteFile(supply_path.AppendASCII("online"), "1"));

  EXPECT_TRUE(sys_utils_->IsPowerSourcePresent());
}

TEST_F(SysUtilsTest, IsPowerSourcePresent_NotPresent) {
  EXPECT_FALSE(sys_utils_->IsPowerSourcePresent());
}

TEST_F(SysUtilsTest, IsPowerSourcePresent_OnBattery) {
  // Create power supply files.
  base::FilePath supply_path =
      temp_dir_.GetPath().AppendASCII("class/power_supply/1");
  EXPECT_TRUE(base::CreateDirectory(supply_path));
  EXPECT_TRUE(base::WriteFile(supply_path.AppendASCII("type"), "Battery"));
  EXPECT_TRUE(base::WriteFile(supply_path.AppendASCII("online"), "1"));

  EXPECT_FALSE(sys_utils_->IsPowerSourcePresent());
}

TEST_F(SysUtilsTest, IsPowerSourcePresent_NotOnline) {
  // Create power supply files.
  base::FilePath supply_path =
      temp_dir_.GetPath().AppendASCII("class/power_supply/1");
  EXPECT_TRUE(base::CreateDirectory(supply_path));
  EXPECT_TRUE(base::WriteFile(supply_path.AppendASCII("type"), "USB_PD"));
  EXPECT_TRUE(base::WriteFile(supply_path.AppendASCII("online"), "0"));

  EXPECT_FALSE(sys_utils_->IsPowerSourcePresent());
}

namespace fake {

class FakeSysUtilsTest : public testing::Test {
 public:
  FakeSysUtilsTest() = default;
  ~FakeSysUtilsTest() override = default;

  base::FilePath GetPowerSourcePresentFilePath() const {
    return temp_dir_.GetPath().AppendASCII(kPowerSourcePresentFilePath);
  }

 protected:
  void SetUp() override {
    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
    fake_sys_utils_ = std::make_unique<FakeSysUtils>(temp_dir_.GetPath());
  }

  base::ScopedTempDir temp_dir_;
  std::unique_ptr<FakeSysUtils> fake_sys_utils_;
};

TEST_F(FakeSysUtilsTest, IsPowerSourcePresent_Present) {
  brillo::TouchFile(GetPowerSourcePresentFilePath());
  ASSERT_TRUE(fake_sys_utils_->IsPowerSourcePresent());
}

TEST_F(FakeSysUtilsTest, IsPowerSourcePresent_NotPresent) {
  ASSERT_FALSE(fake_sys_utils_->IsPowerSourcePresent());
}

}  // namespace fake

}  // namespace rmad