From 3545ef8bb49d9ad02e85bd13f478072f8059c582 Mon Sep 17 00:00:00 2001
From: Damien George
Date: Thu, 21 Apr 2016 12:38:22 +0000
Subject: tests/thread: Remove need to sleep to wait for completion in some
 tests.

Use a lock and a counter instead, and busy wait for all threads to
complete.  This makes test run faster and they no longer rely on the time
module.
---
 tests/thread/thread_stacksize1.py | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

(limited to 'tests/thread/thread_stacksize1.py')

diff --git a/tests/thread/thread_stacksize1.py b/tests/thread/thread_stacksize1.py
index b0118843b..e3e77fada 100644
--- a/tests/thread/thread_stacksize1.py
+++ b/tests/thread/thread_stacksize1.py
@@ -3,10 +3,6 @@
 # MIT license; Copyright (c) 2016 Damien P. George on behalf of Pycom Ltd
 
 import sys
-try:
-    import utime as time
-except ImportError:
-    import time
 import _thread
 
 # different implementations have different minimum sizes
@@ -20,6 +16,9 @@ def foo():
 
 def thread_entry():
     foo()
+    with lock:
+        global n_finished
+        n_finished += 1
 
 # test set/get of stack size
 print(_thread.stack_size())
@@ -27,10 +26,16 @@ print(_thread.stack_size(sz))
 print(_thread.stack_size() == sz)
 print(_thread.stack_size())
 
+lock = _thread.allocate_lock()
+n_thread = 2
+n_finished = 0
+
 # set stack size and spawn a few threads
 _thread.stack_size(sz)
-for i in range(2):
+for i in range(n_thread):
     _thread.start_new_thread(thread_entry, ())
 
-time.sleep(0.2)
+# busy wait for threads to finish
+while n_finished < n_thread:
+    pass
 print('done')
-- 
cgit v1.2.3