laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

get even more ones

Browse Source
This commit is contained in:
Khaleel Al-Adhami 2024-11-04 10:46:25 -08:00
parent 8977ab7918
commit 641f07861d
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/integration/test_upload.py
View File

@ -207,11 +207,11 @@ async def test_upload_file(
file_data = await AppHarness._poll_for_async(get_file_data) file_data = await AppHarness._poll_for_async(get_file_data)
assert isinstance(file_data, dict) assert isinstance(file_data, dict)
normalized_file_data = {Path(k).name: v for k, v in file_data.items()} normalized_file_data = {Path(k).name: v for k, v in file_data.items()}
assert normalized_file_data[exp_name] == exp_contents assert normalized_file_data[Path(exp_name).name] == exp_contents
# check that the selected files are displayed # check that the selected files are displayed
selected_files = driver.find_element(By.ID, f"selected_files{suffix}") selected_files = driver.find_element(By.ID, f"selected_files{suffix}")
assert selected_files.text == exp_name assert Path(selected_files.text).name == Path(exp_name).name
state = await upload_file.get_state(substate_token) state = await upload_file.get_state(substate_token)
if secondary: if secondary:
@ -275,8 +275,9 @@ async def test_upload_file_multiple(tmp_path, upload_file: AppHarness, driver):
file_data = await AppHarness._poll_for_async(get_file_data) file_data = await AppHarness._poll_for_async(get_file_data)
assert isinstance(file_data, dict) assert isinstance(file_data, dict)
normalized_file_data = {Path(k).name: v for k, v in file_data.items()}
for exp_name, exp_contents in exp_files.items(): for exp_name, exp_contents in exp_files.items():
assert file_data[exp_name] == exp_contents assert normalized_file_data[Path(exp_name).name] == exp_contents
@pytest.mark.parametrize("secondary", [False, True]) @pytest.mark.parametrize("secondary", [False, True])
@ -375,6 +376,9 @@ async def test_cancel_upload(tmp_path, upload_file: AppHarness, driver: WebDrive
# look up the backend state and assert on progress # look up the backend state and assert on progress
state = await upload_file.get_state(substate_token) state = await upload_file.get_state(substate_token)
assert state.substates[state_name].progress_dicts assert state.substates[state_name].progress_dicts
assert exp_name not in state.substates[state_name]._file_data file_data = state.substates[state_name]._file_data
assert isinstance(file_data, dict)
normalized_file_data = {Path(k).name: v for k, v in file_data.items()}
assert Path(exp_name).name not in normalized_file_data
target_file.unlink() target_file.unlink()