1 #pragma once
2 
3 #include "pmbus.hpp"
4 #include "util_base.hpp"
5 
6 #include <gpiod.hpp>
7 
8 #include <bitset>
9 
10 #include <gmock/gmock.h>
11 
12 namespace phosphor
13 {
14 namespace pmbus
15 {
16 class MockedPMBus : public PMBusBase
17 {
18 
19   public:
20     virtual ~MockedPMBus() = default;
21 
22     MOCK_METHOD(uint64_t, read, (const std::string& name, Type type),
23                 (override));
24     MOCK_METHOD(std::string, readString, (const std::string& name, Type type),
25                 (override));
26     MOCK_METHOD(void, writeBinary,
27                 (const std::string& name, std::vector<uint8_t> data, Type type),
28                 (override));
29     MOCK_METHOD(void, findHwmonDir, (), (override));
30     MOCK_METHOD(const fs::path&, path, (), (const, override));
31 };
32 } // namespace pmbus
33 
34 namespace power
35 {
36 namespace psu
37 {
38 
39 class MockedUtil : public UtilBase
40 {
41   public:
42     virtual ~MockedUtil() = default;
43 
44     MOCK_METHOD(bool, getPresence,
45                 (sdbusplus::bus::bus & bus, const std::string& invpath),
46                 (const, override));
47     MOCK_METHOD(void, setPresence,
48                 (sdbusplus::bus::bus & bus, const std::string& invpath,
49                  bool present, const std::string& name),
50                 (const, override));
51 };
52 
53 class MockedGPIOReader : public GPIOInterface
54 {
55   public:
56     MOCK_METHOD(int, read, (), (override));
57 };
58 
59 const UtilBase& getUtils();
60 
61 void freeUtils();
62 
63 } // namespace psu
64 } // namespace power
65 
66 } // namespace phosphor
67