wip

attdona · Oct 1, 2017 · 6da27fc · 6da27fc
1 parent 197276b
commit 6da27fc
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/burba/tests/test_cc3200_mqtt_leds.py b/burba/tests/test_cc3200_mqtt_leds.py
@@ -2,6 +2,8 @@
 """
 import os
 import asyncio
+import logging
+import sys
 
 # the Leds message
 import commands as cmd
@@ -20,6 +22,13 @@
 # import firmware configuration
 import tc
 
+logging.basicConfig(
+    level=logging.DEBUG,
+    format='%(levelname)s:%(name)s:%(lineno)s: %(message)s',
+    stream=sys.stderr
+)
+LOG = logging.getLogger()
+
 EVENTS = (
     ('board init', 'BOARD_INIT'),
     ('board ready', 'BOARD_READY')
@@ -98,11 +107,12 @@ async def test_switch_led():
     await cli.publish('{}/{}'.format(tc.network, tc.board), msg_off, qos=QOS_0)
 
     try:
-        # wait for the ack
-        message = await cli.deliver_message(timeout=3)
-        packet = message.publish_packet
-        print("%s => %s" %
-              (packet.variable_header.topic_name, str(packet.payload.data)))
+        for _ in range(2):
+            # wait for the ack
+            message = await cli.deliver_message(timeout=3)
+            packet = message.publish_packet
+            LOG.debug("%s => %s",
+                packet.variable_header.topic_name, str(packet.payload.data))
     except asyncio.TimeoutError:
         assert 0
     finally: