1 files changed, 17 insertions, 1 deletions
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)