def test_on_connect_failed(self):
userdata = Mock()
userdata.product_type = 'ptype'
userdata.serial = 'serial'
DysonPureCoolLink.on_connect(None, userdata, None, 1)
userdata.connection_callback.assert_called_with(False)
self.assertEqual(userdata.connection_callback.call_count, 1)
def test_on_connect(self):
client = Mock()
client.subscribe = Mock()
userdata = Mock()
userdata.status_topic = "ptype/serial/status/current"
DysonPureCoolLink.on_connect(client, userdata, None, 0)
userdata.connection_callback.assert_called_with(True)
self.assertEqual(userdata.connection_callback.call_count, 1)
client.subscribe.assert_called_with("ptype/serial/status/current")