fix tests

Author: kumaraditya303

Lib/test/test_asyncio/test_events.py

@@ -2930,11 +2930,16 @@ class GetEventLoopTestsMixin:
     get_running_loop_impl = None
     get_event_loop_impl = None
 
+    Task = None
+    Future = None
+
     def setUp(self):
         self._get_running_loop_saved = events._get_running_loop
         self._set_running_loop_saved = events._set_running_loop
         self.get_running_loop_saved = events.get_running_loop
         self.get_event_loop_saved = events.get_event_loop
+        self._Task_saved = asyncio.Task
+        self._Future_saved = asyncio.Future
 
         events._get_running_loop = type(self)._get_running_loop_impl
         events._set_running_loop = type(self)._set_running_loop_impl
@@ -2946,6 +2951,8 @@ def setUp(self):
         asyncio.get_running_loop = type(self).get_running_loop_impl
         asyncio.get_event_loop = type(self).get_event_loop_impl
 
+        asyncio.Task = asyncio.tasks.Task = type(self).Task
+        asyncio.Future = asyncio.futures.Future = type(self).Future
         super().setUp()
 
         self.loop = asyncio.new_event_loop()
@@ -2968,8 +2975,10 @@ def tearDown(self):
             asyncio.get_running_loop = self.get_running_loop_saved
             asyncio.get_event_loop = self.get_event_loop_saved
 
+            asyncio.Task = asyncio.tasks.Task = self._Task_saved
+            asyncio.Future = asyncio.futures.Future = self._Future_saved
+
     if sys.platform != 'win32':
-        @unittest.skip("skip")
         def test_get_event_loop_new_process(self):
             # bpo-32126: The multiprocessing module used by
             # ProcessPoolExecutor is not functional when the
@@ -2995,7 +3004,6 @@ async def main():
                 self.loop.run_until_complete(main()),
                 'hello')
 
-    @unittest.skip("skip")
     def test_get_event_loop_returns_running_loop(self):
         class TestError(Exception):
             pass
@@ -3043,7 +3051,6 @@ async def func():
 
         self.assertIs(asyncio._get_running_loop(), None)
 
-    @unittest.skip("skip")
     def test_get_event_loop_returns_running_loop2(self):
         old_policy = asyncio._get_event_loop_policy()
         try:
@@ -3090,6 +3097,8 @@ class TestPyGetEventLoop(GetEventLoopTestsMixin, unittest.TestCase):
     get_running_loop_impl = events._py_get_running_loop
     get_event_loop_impl = events._py_get_event_loop
 
+    Task = asyncio.tasks._PyTask
+    Future = asyncio.futures._PyFuture
 
 try:
     import _asyncio  # NoQA
@@ -3104,6 +3113,8 @@ class TestCGetEventLoop(GetEventLoopTestsMixin, unittest.TestCase):
         get_running_loop_impl = events._c_get_running_loop
         get_event_loop_impl = events._c_get_event_loop
 
+        Task = asyncio.tasks._CTask
+        Future = asyncio.futures._CFuture
 
 class TestServer(unittest.TestCase):
 

Lib/test/test_asyncio/test_tasks.py

@@ -2252,7 +2252,7 @@ def test_wait_invalid_args(self):
         # wait() expects at least a future
         self.assertRaises(ValueError, self.loop.run_until_complete,
             asyncio.wait([]))
-    @unittest.skip("This test is flaky")
+
     def test_log_destroyed_pending_task(self):
 
         async def kill_me(loop):
@@ -3012,6 +3012,9 @@ def done(self):
     def test__enter_task(self):
         task = mock.Mock()
         loop = mock.Mock()
+        # _enter_task is called by Task.__step while the loop
+        # is running, so set the loop as the running loop
+        # for a more realistic test.
         asyncio._set_running_loop(loop)
         self.assertIsNone(self.current_task(loop))
         self._enter_task(loop, task)
@@ -3044,6 +3047,9 @@ def test__leave_task_failure1(self):
         task1 = mock.Mock()
         task2 = mock.Mock()
         loop = mock.Mock()
+        # _leave_task is called by Task.__step while the loop
+        # is running, so set the loop as the running loop
+        # for a more realistic test.
         asyncio._set_running_loop(loop)
         self._enter_task(loop, task1)
         with self.assertRaises(RuntimeError):