def send(self, msg, handler, priority=None, after=None):
self.sent.append(H.Data(msg=msg, handler=handler))
# This is basically what the db modify is doing when it get's a ACK
# of the db add message. So we're just short circuiting that and
# pretending the message came back.
if isinstance(handler, IM.handler.DeviceDbModify):
handler.db.add_entry(handler.entry)
handler.on_done(True, "update", handler.entry)
def test_get_battery_voltage(self, mocker):
mocker.patch('insteon_mqtt.cmd_line.util.send')
IM.cmd_line.util.send.return_value = {"status": 10}
args = helpers.Data(topic="cmd_topic",
force=False,
quiet=True,
address="aa.bb.cc")
config = helpers.Data(a=1, b=2)
r = IM.cmd_line.device.get_battery_voltage(args, config)
assert r == 10
topic = "%s/%s" % (args.topic, args.address)
payload = {
"cmd": "get_battery_voltage",
}
self.check_call(IM.cmd_line.util.send, args, config, topic, payload)
def setup(mock_paho_mqtt, tmpdir):
proto = H.main.MockProtocol()
modem = IM.Modem(proto, IM.network.Stack(), IM.network.TimedCall())
modem.name = "modem"
modem.addr = IM.Address(0x20, 0x30, 0x40)
link = IM.network.Mqtt()
mqtt = IM.mqtt.Mqtt(link, modem)
mdev = IM.mqtt.Modem(mqtt, modem)
return H.Data(addr=modem.addr, name=modem.name, link=link,
mdev=mdev, proto=proto)
def test_set_button_led(self, mocker):
mocker.patch('insteon_mqtt.cmd_line.util.send')
IM.cmd_line.util.send.return_value = {"status": 10}
args = helpers.Data(topic="cmd_topic",
force=False,
quiet=True,
group=5,
address="aa.bb.cc",
is_on=True)
config = helpers.Data(a=1, b=2)
r = IM.cmd_line.device.set_button_led(args, config)
assert r == 10
topic = "%s/%s" % (args.topic, args.address)
payload = {
"cmd": "set_button_led",
"group": args.group,
"is_on": args.is_on,
}
self.check_call(IM.cmd_line.util.send, args, config, topic, payload)
def setup(mock_paho_mqtt, tmpdir):
proto = H.main.MockProtocol()
modem = H.main.MockModem(tmpdir)
addr = IM.Address(1, 2, 3)
name = "device name"
dev = IM.device.Motion(proto, modem, addr, name)
link = IM.network.Mqtt()
mqttModem = H.mqtt.MockModem()
mqtt = IM.mqtt.Mqtt(link, mqttModem)
mdev = IM.mqtt.Motion(mqtt, dev)
return H.Data(addr=addr, name=name, dev=dev, mdev=mdev, link=link)
def send(self, msg, handler, priority=None, after=None):
self.sent.append(H.Data(msg=msg, handler=handler))
def setup(mock_paho_mqtt):
link = IM.network.Mqtt()
mqttModem = H.mqtt.MockModem()
mqtt = IM.mqtt.Mqtt(link, mqttModem)
return H.Data(mqtt=mqtt, mqttModem=mqttModem, link=link)
