Try separate file_manager.stop() and start() calls to better

moodyjon · eukreign · commit 4ef03bb1f4ae · 2023-04-03T13:34:36.000-04:00
control order of events in test.
While file_manager is stopped, we get no response to file_list().
diff --git a/tests/integration/datanetwork/test_streaming.py b/tests/integration/datanetwork/test_streaming.py
@@ -4,6 +4,7 @@
 import aiohttp.web
 import asyncio
 
+from lbry.file.source import ManagedDownloadSource
 from lbry.utils import aiohttp_request
 from lbry.blob.blob_file import MAX_BLOB_SIZE
 from lbry.testcase import CommandTestCase
@@ -354,10 +355,15 @@ async def test_file_save_stop_before_finished_streaming_only(self, wait_for_star
         if wait_for_start_writing:
             await stream.started_writing.wait()
             self.assertTrue(os.path.isfile(path))
-        await self._restart_stream_manager()
+        await self.daemon.file_manager.stop()
+        # while stopped, we get no response to query and no file is present
+        self.assertEqual((await self.daemon.jsonrpc_file_list())['items'], [])
+        self.assertEqual(os.path.isfile(path), stream.status == ManagedDownloadSource.STATUS_FINISHED)
+        await self.daemon.file_manager.start()
+        # after restart, we get a response to query and same file path
         stream = (await self.daemon.jsonrpc_file_list())['items'][0]
         self.assertIsNotNone(stream.full_path)
-        self.assertFalse(os.path.isfile(path))
+        self.assertEqual(stream.full_path, path)
         if wait_for_start_writing:
             await stream.started_writing.wait()
             self.assertTrue(os.path.isfile(path))
