mirror of
https://github.com/esphome/esphome.git
synced 2026-08-22 22:26:21 +00:00
88 lines
2.9 KiB
C++
88 lines
2.9 KiB
C++
#pragma once
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
#include <algorithm>
|
|
#include <cstdint>
|
|
#include <initializer_list>
|
|
#include <vector>
|
|
#include "esphome/components/uart/uart_component.h"
|
|
#include "esphome/components/mitsubishi_cn105/mitsubishi_cn105.h"
|
|
#include "esphome/components/mitsubishi_cn105/mitsubishi_cn105_component.h"
|
|
#include "esphome/components/mitsubishi_cn105/mitsubishi_cn105_climate.h"
|
|
|
|
namespace esphome::mitsubishi_cn105::testing {
|
|
|
|
class MockUARTComponent : public uart::UARTComponent {
|
|
public:
|
|
std::vector<uint8_t> tx;
|
|
std::vector<uint8_t> rx;
|
|
|
|
void push_rx(std::initializer_list<uint8_t> data) { this->rx.insert(this->rx.end(), data.begin(), data.end()); }
|
|
|
|
// UARTComponent
|
|
void write_array(const uint8_t *data, size_t len) override { this->tx.insert(this->tx.end(), data, data + len); }
|
|
|
|
bool read_array(uint8_t *data, size_t len) override {
|
|
if (this->rx.size() < len) {
|
|
return false;
|
|
}
|
|
|
|
std::copy(this->rx.begin(), this->rx.begin() + len, data);
|
|
this->rx.erase(this->rx.begin(), this->rx.begin() + len);
|
|
return true;
|
|
}
|
|
|
|
size_t available() override { return this->rx.size(); }
|
|
|
|
MOCK_METHOD(bool, peek_byte, (uint8_t * data), (override));
|
|
MOCK_METHOD(uart::UARTFlushResult, flush, (), (override));
|
|
MOCK_METHOD(void, check_logger_conflict, (), (override));
|
|
#if defined(USE_ESP8266) || defined(USE_ESP32)
|
|
void load_settings(bool dump_config) override {}
|
|
#endif // defined(USE_ESP8266) || defined(USE_ESP32)
|
|
};
|
|
|
|
class TestableMitsubishiCN105 : public MitsubishiCN105 {
|
|
public:
|
|
using MitsubishiCN105::MitsubishiCN105;
|
|
using MitsubishiCN105::State;
|
|
using MitsubishiCN105::PropertyId;
|
|
using MitsubishiCN105::state_;
|
|
using MitsubishiCN105::status_;
|
|
using MitsubishiCN105::operation_start_ms_;
|
|
using MitsubishiCN105::property_context_;
|
|
using MitsubishiCN105::status_update_wait_credit_ms_;
|
|
using MitsubishiCN105::pending_updates_;
|
|
|
|
void set_state(State s) { this->set_state_(s); }
|
|
void apply_settings() { this->apply_settings_(); }
|
|
|
|
static inline uint32_t test_loop_time_ms = 0;
|
|
|
|
void set_current_time(uint32_t ms) { test_loop_time_ms = ms; }
|
|
};
|
|
|
|
class TestableMitsubishiCN105Climate : public MitsubishiCN105Climate {
|
|
public:
|
|
TestableMitsubishiCN105Climate() { this->set_parent(&this->component_); }
|
|
|
|
using MitsubishiCN105Climate::apply_values_;
|
|
using MitsubishiCN105Climate::last_non_swing_vane_mode_;
|
|
using MitsubishiCN105Climate::last_non_swing_wide_vane_mode_;
|
|
|
|
MitsubishiCN105::Status &status() { return const_cast<MitsubishiCN105::Status &>(this->component_.status()); }
|
|
|
|
protected:
|
|
MitsubishiCN105Component component_;
|
|
};
|
|
|
|
class TestableMitsubishiCN105Component : public MitsubishiCN105Component {
|
|
public:
|
|
MitsubishiCN105::Status &mutable_status() { return const_cast<MitsubishiCN105::Status &>(this->status()); }
|
|
|
|
void notify_status() { this->status_callback_.call(); }
|
|
};
|
|
|
|
} // namespace esphome::mitsubishi_cn105::testing
|