zircon/system/dev/light/lp50xx-light/lp50xx-light-test.cc

// Copyright 2019 The Fuchsia 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 "lp50xx-light.h"

#include <lib/fake_ddk/fake_ddk.h>
#include <lib/mock-i2c/mock-i2c.h>

#include <ddk/platform-defs.h>
#include <zxtest/zxtest.h>

namespace lp50xx_light {

class Lp50xxLightTest : public Lp50xxLight {
 public:
  Lp50xxLightTest() : Lp50xxLight(fake_ddk::kFakeParent) {}

  virtual zx_status_t InitHelper() {
    auto proto = ddk::I2cProtocolClient(mock_i2c.GetProto());
    i2c_ = std::move(proto);

    mock_i2c.ExpectWriteStop({0x00, 0x40}).ExpectWriteStop({0x01, 0x3C});

    pid_ = PDEV_PID_TI_LP5018;
    return ZX_OK;
  }

  zx_status_t GetRgb(uint32_t index, llcpp::fuchsia::hardware::light::Rgb* rgb) {
    return Lp50xxLight::GetRgbValue(index, rgb);
  }

  zx_status_t SetRgb(uint32_t index, llcpp::fuchsia::hardware::light::Rgb rgb) {
    return Lp50xxLight::SetRgbValue(index, rgb);
  }

  void Verify() { mock_i2c.VerifyAndClear(); }

  mock_i2c::MockI2c mock_i2c;

 private:
  fake_ddk::Bind ddk;
};

TEST(Lp50xxLightTest, InitTest) {
  Lp50xxLightTest dut;
  EXPECT_OK(dut.Init());
  dut.Verify();
}

TEST(Lp50xxLightTest, GetRgbTest) {
  Lp50xxLightTest dut;
  EXPECT_OK(dut.Init());

  dut.mock_i2c.ExpectWrite({0x10})
      .ExpectReadStop({0xAA})
      .ExpectWrite({0x11})
      .ExpectReadStop({0xBB})
      .ExpectWrite({0x0f})
      .ExpectReadStop({0xCC});

  llcpp::fuchsia::hardware::light::Rgb rgb = {};
  EXPECT_OK(dut.GetRgb(0, &rgb));
  EXPECT_EQ(rgb.red, 0xAA);
  EXPECT_EQ(rgb.green, 0xBB);
  EXPECT_EQ(rgb.blue, 0xCC);

  dut.Verify();
}

TEST(Lp50xxLightTest, SetRgbTest) {
  Lp50xxLightTest dut;
  EXPECT_OK(dut.Init());

  llcpp::fuchsia::hardware::light::Rgb rgb = {};
  rgb.red = 0xAA;
  rgb.green = 0xBB;
  rgb.blue = 0xCC;

  dut.mock_i2c.ExpectWriteStop({0x10, 0xAA})
      .ExpectWriteStop({0x11, 0xBB})
      .ExpectWriteStop({0x0f, 0xCC});

  EXPECT_OK(dut.SetRgb(0, rgb));

  dut.Verify();
}

}  // namespace lp50xx_light