power_manager/powerd/system/ambient_light_sensor_watcher_mojo_test.cc

// Copyright 2022 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "power_manager/powerd/system/ambient_light_sensor_watcher_mojo.h"

#include <map>
#include <memory>
#include <optional>
#include <string>
#include <utility>

#include <base/run_loop.h>
#include <gtest/gtest.h>

#include "power_manager/powerd/system/ambient_light_sensor_watcher_observer_stub.h"
#include "power_manager/powerd/system/fake_light.h"
#include "power_manager/powerd/system/fake_sensor_service.h"
#include "power_manager/powerd/system/sensor_service_handler.h"
#include "power_manager/powerd/testing/test_environment.h"

namespace power_manager::system {

namespace {

constexpr char kBadSyspath[] =
    "/sys/my/mock/device/HID-SENSOR-200040/more/mock/path";

constexpr char kGoodSyspath[] =
    "/sys/my/mock/device/HID-SENSOR-200041/more/mock/path";

constexpr char kGoodSyspath2[] =
    "/sys/my/mock/device/HID-SENSOR-200041/more/mock/path";

}  // namespace

class AmbientLightSensorWatcherMojoTest : public MojoTestEnvironment {
 public:
  AmbientLightSensorWatcherMojoTest() = default;
  ~AmbientLightSensorWatcherMojoTest() override = default;

 protected:
  void SetUp() override {
    watcher_ = std::make_unique<AmbientLightSensorWatcherMojo>(
        &sensor_service_handler_);

    auto sensor_device = std::make_unique<FakeLight>(false /*is_color_sensor*/,
                                                     std::nullopt /*name*/,
                                                     cros::mojom::kLocationLid);

    SetSensor(kBadSyspath);  // id = 0

    mojo::PendingRemote<cros::mojom::SensorService> pending_remote;
    sensor_service_.AddReceiver(
        pending_remote.InitWithNewPipeAndPassReceiver());
    sensor_service_handler_.SetUpChannel(std::move(pending_remote));
  }

  void SetSensor(std::string sys_path) {
    auto sensor_device = std::make_unique<FakeLight>(false /*is_color_sensor*/,
                                                     std::nullopt /*name*/,
                                                     cros::mojom::kLocationLid);
    sensor_device->SetAttribute(cros::mojom::kSysPath, sys_path);

    auto iio_device_id = sensor_num_++;
    fake_lights_[iio_device_id] = sensor_device.get();
    sensor_service_.SetSensorDevice(iio_device_id, std::move(sensor_device));
  }

  FakeSensorService sensor_service_;
  std::map<int32_t, FakeLight*> fake_lights_;

  SensorServiceHandler sensor_service_handler_;

  std::unique_ptr<AmbientLightSensorWatcherMojo> watcher_;

  int32_t sensor_num_ = 0;
};

TEST_F(AmbientLightSensorWatcherMojoTest, Basic) {
  AmbientLightSensorWatcherObserverStub observer(watcher_.get());

  // Wait until all initialization tasks are done.
  base::RunLoop().RunUntilIdle();

  EXPECT_EQ(0, observer.num_als_changes());

  SetSensor(kGoodSyspath);  // id = 1

  // Wait until all tasks are done.
  base::RunLoop().RunUntilIdle();

  EXPECT_EQ(1, observer.num_als_changes());
  EXPECT_EQ(1, observer.num_als());

  sensor_service_.ClearReceivers();

  // Wait until all tasks are done.
  base::RunLoop().RunUntilIdle();

  EXPECT_EQ(2, observer.num_als_changes());
  EXPECT_EQ(0, observer.num_als());
}

TEST_F(AmbientLightSensorWatcherMojoTest, SensorDeviceDisconnect) {
  AmbientLightSensorWatcherObserverStub observer(watcher_.get());

  SetSensor(kGoodSyspath);   // id = 1
  SetSensor(kGoodSyspath2);  // id = 2

  // Wait until all tasks are done.
  base::RunLoop().RunUntilIdle();

  EXPECT_EQ(2, observer.num_als_changes());
  EXPECT_EQ(2, observer.num_als());

  // AmbientLightSensorWatcherMojo doesn't keep SensorDevice remotes after
  // getting device attributes.
  EXPECT_FALSE(fake_lights_[1]->HasReceivers());
  EXPECT_FALSE(fake_lights_[2]->HasReceivers());

  sensor_service_.RemoveSensorDevice(2);

  // Wait until all tasks are done.
  base::RunLoop().RunUntilIdle();

  EXPECT_EQ(3, observer.num_als_changes());
  EXPECT_EQ(1, observer.num_als());

  sensor_service_.ClearReceivers();

  // Wait until all tasks are done.
  base::RunLoop().RunUntilIdle();

  EXPECT_EQ(4, observer.num_als_changes());
  EXPECT_EQ(0, observer.num_als());
}

}  // namespace power_manager::system