test wip

attdona · Oct 1, 2017 · 197276b · 197276b
1 parent fd7e566
commit 197276b
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/burba/tests/tc.py b/burba/tests/tc.py
@@ -1,4 +1,7 @@
+"""global test configuration
+"""
 import os
+import asyncio
 
 #broker_url
 broker='mqtt://localhost:1883'
@@ -16,5 +19,18 @@
 }
 
 def check_test_requirements():
+    """required test values
+    """
     assert 'TEST_UID' in os.environ, "unset env variable TEST_UID"
     assert 'TEST_PWD' in os.environ, "unset env variable TEST_PWD"
+
+def cancel_tasks():
+    """Cancel background tasks to reset environment between test cases
+    """
+    pendings = asyncio.Task.all_tasks()
+
+    current_task = asyncio.Task.current_task()
+    for task in pendings:
+
+        if (task != current_task):
+            task.cancel()
diff --git a/burba/tests/test_cc3200_mqtt_leds.py b/burba/tests/test_cc3200_mqtt_leds.py
@@ -110,3 +110,4 @@ async def test_switch_led():
         await asyncio.sleep(1)
         await cli.disconnect()
         await broker.shutdown()
+        tc.cancel_tasks()
diff --git a/burba/tests/test_cc3200_mqtt_profile.py b/burba/tests/test_cc3200_mqtt_profile.py
@@ -104,5 +104,6 @@ async def test_add_profile():
     finally:
         await cli.disconnect()
         await broker.shutdown()
+        tc.cancel_tasks()