def test_on_message_hot(self): def on_message(msg): assert isinstance(msg, DysonPureHotCoolState) device = DysonPureCoolLink({ "Active": True, "Serial": "device-id-1", "Name": "device-1", "ScaleUnit": "SU01", "Version": "21.03.08", "LocalCredentials": "1/aJ5t52WvAfn+z+fjDuef86kQDQPefbQ6/" "70ZGysII1Ke1i0ZHakFH84DZuxsSQ4KTT2v" "bCm7uYeTORULKLKQ==", "AutoUpdate": True, "NewVersionAvailable": False, "ProductType": "455" }) device.add_message_listener(on_message) msg = Mock() payload = open("tests/data/state_hot.json", "r").read() msg.payload = Mock() msg.payload.decode.return_value = payload DysonPureCoolLink.on_message(None, device, msg)
def test_add_message_listener(self): def on_message(): pass def on_message_2(): pass device = DysonPureCoolLink({ "Active": True, "Serial": "device-id-1", "Name": "device-1", "ScaleUnit": "SU01", "Version": "21.03.08", "LocalCredentials": "1/aJ5t52WvAfn+z+fjDuef86kQDQPefbQ6/70ZGysII1K" "e1i0ZHakFH84DZuxsSQ4KTT2vbCm7uYeTORULKLKQ==", "AutoUpdate": True, "NewVersionAvailable": False, "ProductType": "475" }) device.add_message_listener(on_message) assert len(device.callback_message) == 1 device.remove_message_listener(on_message) assert len(device.callback_message) == 0 device.add_message_listener(on_message_2) device.add_message_listener(on_message) assert len(device.callback_message) == 2 device.clear_message_listener() assert len(device.callback_message) == 0