Fix a bug where inactive loops would be indexed (read desc)

Cleaned up Aergia.walk_back_until_loop to only get running event loops. Previously, this manifested as a bug. The asyncio.Process class would fork off a thread, and the event loop would be found in that thread's local variables, even though that thread was not executing that event loop. This would result in get_event_loops containing duplicate values, and affected event loops would be assigned duplicate time

3 files changed, 23 insertions, 5 deletions

diff --git a/t/eager_and_scheduled.py b/t/eager_and_scheduled.py
index 178a9f4..030b539 100644
--- a/t/eager_and_scheduled.py
+++ b/t/eager_and_scheduled.py
@@ -8,7 +8,8 @@ async def foo():
 
 
 async def bar():
-    await asyncio.create_subprocess_shell('sleep 1.0')
+    proc = await asyncio.create_subprocess_shell('sleep 1.0')
+    await proc.communicate()
 
 
 async def baz():
diff --git a/t/test_functionality.py b/t/test_functionality.py
index 5253534..12bf775 100644
--- a/t/test_functionality.py
+++ b/t/test_functionality.py
@@ -52,6 +52,22 @@ class BasicUsage(utils.AergiaUnitTestCase):
 
         samples = self.Aergia.get_samples()
 
-        print(self.expected_samples(delay * 3))
         self.assertFuncContains('b', [self.expected_samples(delay * 3)],
                                 samples)
+
+        # TODO samples from gather all execution time, should we trace this??
+        self.assertFuncContains('a', [self.expected_samples(delay)], samples)
+
+    def test_eager_task(self):
+        delay = 0.2
+
+        async def a():
+            proc = await asyncio.create_subprocess_shell(f'sleep {delay}')
+            await proc.communicate()
+
+        self.Aergia.start()
+        asyncio.run(a())
+        self.Aergia.stop()
+
+        samples = self.Aergia.get_samples()
+        self.assertFuncContains('a', [self.expected_samples(delay)], samples)
diff --git a/t/utils.py b/t/utils.py
index fe9db26..5772fac 100644
--- a/t/utils.py
+++ b/t/utils.py
@@ -13,16 +13,17 @@ class AergiaUnitTestCase(unittest.TestCase):
     def assertFuncContains(self, func_name, samples_expected, samples):
         samples_actual = self.extract_values_by_func(samples,
                                                      func_name)
-        self.assertTrue(len(samples_expected) == len(samples_actual))
+        self.assertTrue(len(samples_expected) == len(samples_actual),
+                        f'{samples_expected} (expected) not length of '
+                        f'{samples_actual} (actual)')
         s1 = sorted(samples_expected)
         s2 = sorted(samples_actual)
         for v1, v2 in zip(s1, s2):
             self.assertRoughlyEqual(v1, v2)
 
     def assertRoughlyEqual(self, v1, v2):
-        print(f'{v1}, {v2}')
         a = abs(v1 - v2)
-        self.assertTrue(a <= 1)
+        self.assertTrue(a <= 1, f'{v1} (expected) not roughly {v2} (actual)')
 
     def expected_samples(self, total_seconds):
         return (total_seconds / self.interval)