35 lines
951 B
Python
35 lines
951 B
Python
import time
|
|
from unittest.mock import patch
|
|
|
|
import paho.mqtt.client as mqtt
|
|
|
|
import power_mqtt1
|
|
|
|
|
|
class SMBus:
|
|
def __init__(self, _bus):
|
|
self.storage = {}
|
|
|
|
def read_byte(self, _i2c_addr):
|
|
time.sleep(5)
|
|
# return random.randbytes(1)[0]
|
|
return self.storage.get(_i2c_addr, -1)
|
|
|
|
def write_byte(self, i2c_addr, data):
|
|
self.storage[i2c_addr] = data
|
|
|
|
|
|
class MqttClient(mqtt.Client):
|
|
def connect(self, host, *args, **kwargs):
|
|
return super().connect("127.0.0.1", *args, **kwargs)
|
|
# return super().connect(host, *args, **kwargs)
|
|
|
|
def publish(self, topic, payload=None, qos=0, retain=False, properties=None):
|
|
print(f"publish topic={topic}, data={payload}")
|
|
super().publish(topic, payload, qos, retain, properties)
|
|
|
|
|
|
with patch('smbus.SMBus', new=SMBus) as SMBus_mock:
|
|
with patch("paho.mqtt.client.Client", new=MqttClient) as connect_mock:
|
|
power_mqtt1.main()
|