diff options
author | Micah Lee <micah@micahflee.com> | 2021-11-18 19:16:00 -0800 |
---|---|---|
committer | Micah Lee <micah@micahflee.com> | 2021-11-18 19:16:00 -0800 |
commit | 627f44c423840dba7a158512702755bbaa917d89 (patch) | |
tree | b6d689f286149ffb7821c784acb3b727d1415832 /desktop/tests | |
parent | 5f5b761fcf59a903b8636843cbcba9bc907c2be3 (diff) | |
parent | 096178a9e6133fd6ca9d95a00a67bba75ccab377 (diff) | |
download | onionshare-627f44c423840dba7a158512702755bbaa917d89.tar.gz onionshare-627f44c423840dba7a158512702755bbaa917d89.zip |
Merge branch 'advisory-fix-1' of github.com:onionshare/onionshare-ghsa-jh82-c5jw-pxpc into ros-fixes
Diffstat (limited to 'desktop/tests')
-rw-r--r-- | desktop/tests/test_gui_receive.py | 32 |
1 files changed, 17 insertions, 15 deletions
diff --git a/desktop/tests/test_gui_receive.py b/desktop/tests/test_gui_receive.py index ca69c957..423b63a0 100644 --- a/desktop/tests/test_gui_receive.py +++ b/desktop/tests/test_gui_receive.py @@ -1,3 +1,4 @@ +import glob import pytest import os import requests @@ -35,17 +36,17 @@ class TestReceive(GuiBaseTest): now = datetime.now() for _ in range(10): date_dir = now.strftime("%Y-%m-%d") - if identical_files_at_once: - time_dir = now.strftime("%H%M%S-1") - else: - time_dir = now.strftime("%H%M%S") + time_dir = now.strftime("%H%M%S") receive_mode_dir = os.path.join( tab.settings.get("receive", "data_dir"), date_dir, time_dir ) - expected_filename = os.path.join(receive_mode_dir, expected_basename) - if os.path.exists(expected_filename): - exists = True - break + # The directories have microseconds in the name, so we need + # to use globbing against directory names containing the same + # second in order to try to find the file. + for path in glob.glob(receive_mode_dir + "*"): + if os.path.exists(os.path.join(path, expected_basename)): + exists = True + break now = now - timedelta(seconds=1) self.assertTrue(exists) @@ -83,17 +84,18 @@ class TestReceive(GuiBaseTest): for _ in range(10): date_dir = now.strftime("%Y-%m-%d") time_dir = now.strftime("%H%M%S") - expected_filename = os.path.join( + expected_estimated_filename = os.path.join( tab.settings.get("receive", "data_dir"), date_dir, - f"{time_dir}-message.txt", + f"{time_dir}*-message.txt", ) - if os.path.exists(expected_filename): - with open(expected_filename) as f: - assert f.read() == message + for path in glob.glob(expected_estimated_filename): + if os.path.exists(path): + with open(path) as f: + assert f.read() == message - exists = True - break + exists = True + break now = now - timedelta(seconds=1) self.assertTrue(exists) |