[PATCH 1/2] auto-t: add API wait_for_object_change

* [PATCH 1/2] auto-t: add API wait_for_object_change
@ 2021-02-05 19:06 James Prestwood
  2021-02-05 19:06 ` [PATCH 2/2] auto-t: update roaming tests with wait_for_object_change James Prestwood
  2021-02-05 20:44 ` [PATCH 1/2] auto-t: add API wait_for_object_change Denis Kenzior
  0 siblings, 2 replies; 7+ messages in thread
From: James Prestwood @ 2021-02-05 19:06 UTC (permalink / raw)
  To: iwd

[-- Attachment #1: Type: text/plain, Size: 2634 bytes --]

This is similar to wait_for_object_condition, but will not allow
any intermediate state changes between the initial and expected
conditions. This is useful for roaming tests when the expected
state change is 'connected' --> 'roaming' with no changes in
between.
---
 autotests/util/iwd.py | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/autotests/util/iwd.py b/autotests/util/iwd.py
index 6b153ea7..8ea1dfb9 100755
--- a/autotests/util/iwd.py
+++ b/autotests/util/iwd.py
@@ -1034,6 +1034,48 @@ class IWD(AsyncOpAbstract):
     def wait_for_object_condition(self, *args, **kwargs):
         self._wait_for_object_condition(*args, **kwargs)
 
+    def wait_for_object_change(self, obj, from_str, to_str, max_wait = 50):
+        '''
+            Expects condition 'from_str' to evaluate true while waiting for 'to_str'. If
+            at any point during the wait 'from_str' evaluates false, an exception is
+            raised.
+
+            This allows an object to be checked for a state transition without any
+            intermediate state changes.
+        '''
+        self._wait_timed_out = False
+
+        def wait_timeout_cb():
+            self._wait_timed_out = True
+            return False
+
+        # Does initial condition pass?
+        if not eval(from_str):
+            raise Exception("initial condition [%s] not met" % from_str)
+
+        try:
+            timeout = GLib.timeout_add_seconds(max_wait, wait_timeout_cb)
+            context = ctx.mainloop.get_context()
+            while True:
+                context.iteration(may_block=True)
+
+                # If neither the initial or expected condition evaluate the
+                # object must be in another unexpected state.
+                if not eval(from_str) and not eval(to_str):
+                    raise Exception('unexpected condition between [%s] and [%s]' % from_str, to_str)
+
+                # Initial condition does not evaluate but expected does, pass
+                if not eval(from_str) and eval(to_str):
+                    break
+
+                if self._wait_timed_out and ctx.args.gdb == None:
+                    raise TimeoutError('[' + to_str + ']'\
+                                       ' condition was not met in '\
+                                       + str(max_wait) + ' sec')
+        finally:
+            if not self._wait_timed_out:
+                GLib.source_remove(timeout)
+
     def wait(self, time):
         self._wait_timed_out = False
         def wait_timeout_cb():
-- 
2.26.2

^ permalink raw reply related	[flat|nested] 7+ messages in thread