Fixed most bugs with I/O (except dangling resources)

aiosched/context.py

@@ -22,8 +22,7 @@ from aiosched.internals.syscalls import (
     cancel,
     set_context,
     close_context,
-    join,
+    join
-    current_task,
 )
 from typing import Any, Coroutine, Callable
 
@@ -88,7 +87,7 @@ class TaskContext(Task):
         """
 
         if isinstance(other, TaskContext):
-            return super().__eq__(self, other)
+            return super().__eq__(other)
         elif isinstance(other, Task):
             return other == self.entry_point
         return False

aiosched/io.py

@@ -75,7 +75,7 @@ class AsyncStream:
 
     async def write(self, data):
         """
-        Writes data b to the file.
+        Writes data to the stream.
         Returns the number of bytes
         written
         """

aiosched/kernel.py

@@ -120,6 +120,7 @@ class FIFOKernel:
             # There's tasks sleeping and/or on the
             # ready queue!
             return False
+        if self.selector.get_map():
             for key in self.selector.get_map().values():
                 # We don't just do any([self.paused, self.run_ready, self.selector.get_map()])
                 # because we don't want to just know if there's any resources we're waiting on,
@@ -406,7 +407,11 @@ class FIFOKernel:
             lambda o: o.fileobj == stream,
             dict(self.selector.get_map()).values(),
         ):
+            if k.data != self.current_task:
+                # We don't want to raise an error inside
+                # the task that's trying to close the stream!
                 self.handle_task_run(partial(k.data.throw, exc), k.data)
+        self.reschedule_running()
 
     def cancel(self, task: Task):
         """

tests/network_channel.py

@@ -8,7 +8,7 @@ async def producer(c: aiosched.NetworkChannel, n: int):
         await c.write(str(i).encode())
         print(f"[producer] Sent {i}")
         await aiosched.sleep(0.5)   # This makes the receiver wait on us!
-    #await c.close()
+    await c.close()
     print("[producer] Done")