bpo-32418: Postfix, raise NotImplementdError and close resources in t… · python/cpython@ffcb4c0 (original) (raw)

Original file line number Diff line number Diff line change
@@ -2826,19 +2826,36 @@ class TestCGetEventLoop(GetEventLoopTestsMixin, unittest.TestCase):
2826 2826 get_running_loop_impl = events._c_get_running_loop
2827 2827 get_event_loop_impl = events._c_get_event_loop
2828 2828
2829 +
2829 2830 class TestServer(unittest.TestCase):
2830 2831
2831 2832 def test_get_loop(self):
2832 2833 loop = asyncio.new_event_loop()
2834 +self.addCleanup(loop.close)
2833 2835 proto = MyProto(loop)
2834 2836 server = loop.run_until_complete(loop.create_server(lambda: proto, '0.0.0.0', 0))
2835 2837 self.assertEqual(server.get_loop(), loop)
2836 -loop.close()
2838 +server.close()
2839 +loop.run_until_complete(server.wait_closed())
2840 +
2837 2841
2838 2842 class TestAbstractServer(unittest.TestCase):
2839 2843
2844 +def test_close(self):
2845 +with self.assertRaises(NotImplementedError):
2846 +events.AbstractServer().close()
2847 +
2848 +def test_wait_closed(self):
2849 +loop = asyncio.new_event_loop()
2850 +self.addCleanup(loop.close)
2851 +
2852 +with self.assertRaises(NotImplementedError):
2853 +loop.run_until_complete(events.AbstractServer().wait_closed())
2854 +
2840 2855 def test_get_loop(self):
2841 -self.assertEqual(events.AbstractServer().get_loop(), NotImplemented)
2856 +with self.assertRaises(NotImplementedError):
2857 +events.AbstractServer().get_loop()
2858 +
2842 2859
2843 2860 if __name__ == '__main__':
2844 2861 unittest.main()