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)