def test_publish(MockMqttClient): client = Client('host') def on_connect(*args, **kwargs): client._connected = True MockMqttClient.return_value.connect.side_effect = on_connect client.connect() client.publish('msg', 'topic') MockMqttClient.return_value.publish.assert_called_once_with('msg', 'topic')