def test_get_callback_func(self): dbroker = DataBroker() test_topic = "/uri/to/topic" client_id = device_id_gen() message = WIFIPayloadGenerator().get() callback = dbroker.get_callback_func(test_topic) res = callback(client_id, None, message) self.assertEqual(res, None)
def test_run_loop(self): dbroker = DataBroker() # Get callback to stop the loop disconnect_callback = dbroker.get_callback_func("disconnect") # On system messages, stop listening sub_topic = "$SYS/#" dbroker.set_callback_func(sub_topic, disconnect_callback) # Start looping. Must exit at any message with True dbroker.initialize() res = dbroker.run_loop() self.assertTrue(res)