예제 #1
0
    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)
예제 #2
0
    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