Deduplicate DURATION constant

Ketok4321 · Jan 25, 2024 · 3cb933b · 3cb933b
1 parent 845992d
commit 3cb933b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/src/backends/librespeed.py b/src/backends/librespeed.py
@@ -8,7 +8,6 @@
 
 DOWNLOAD_SIZE = 100
 
-DURATION = 15
 DL_STREAMS = 6
 UP_STREAMS = 3
 
@@ -52,11 +51,11 @@ async def check_server(server, results):
 
                 return results
 
-    async def start(self, server, res, notify):
+    async def start(self, server, res, notify, duration):
         async def perform_test(test, streams, res):
             tasks = []
 
-            timeout = asyncio.create_task(asyncio.sleep(DURATION))
+            timeout = asyncio.create_task(asyncio.sleep(duration))
 
             for _ in range(streams):
                 tasks.append(asyncio.create_task(test(server, res)))

diff --git a/src/test_worker.py b/src/test_worker.py
@@ -6,7 +6,7 @@
 
 from gi.repository import GLib
 
-DURATION = 15 #TODO: This constant is in two places now
+DURATION = 15
 OVERHEAD_COMPENSATION = 1.06
 
 @dataclass
@@ -88,7 +88,7 @@ def on_event(type):
 
             GLib.idle_add(self.app.win.test_view.progress.set_visible, True)
             self.results = TestResults()
-            await self.app.backend.start(self.server, self.results, lambda type: GLib.idle_add(on_event, type))
+            await self.app.backend.start(self.server, self.results, lambda type: GLib.idle_add(on_event, type), DURATION)
             GLib.idle_add(self.app.win.test_view.progress.set_visible, False)
         except Exception as e:
             print(e)