Skip to content

Commit 7076799

Browse files
committed
Fix integration tests
1 parent 4beba37 commit 7076799

File tree

5 files changed

+10
-5
lines changed

5 files changed

+10
-5
lines changed

tests/test_integration/test_run_transfer/test_ftp.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -452,7 +452,8 @@ async def test_run_transfer_postgres_to_ftp_with_incremental_strategy(
452452
await run_transfer_and_verify(client, group_owner, postgres_to_ftp.id)
453453

454454
downloader.run()
455-
verify_file_name_template(list(Path.iterdir(tmp_path)), expected_extension)
455+
file_names = [f.name for f in Path.iterdir(tmp_path) if f.is_file()]
456+
verify_file_name_template(file_names, expected_extension)
456457

457458
df_with_increment = reader.run()
458459
df_with_increment, second_transfer_df = cast_dataframe_types(df_with_increment, second_transfer_df)

tests/test_integration/test_run_transfer/test_ftps.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,8 @@ async def test_run_transfer_postgres_to_ftps_with_incremental_strategy(
317317
await run_transfer_and_verify(client, group_owner, postgres_to_ftps.id)
318318

319319
downloader.run()
320-
verify_file_name_template(list(Path.iterdir(tmp_path)), expected_extension)
320+
file_names = [f.name for f in Path.iterdir(tmp_path) if f.is_file()]
321+
verify_file_name_template(file_names, expected_extension)
321322

322323
df_with_increment = reader.run()
323324
df, second_transfer_df = cast_dataframe_types(df, second_transfer_df)

tests/test_integration/test_run_transfer/test_samba.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,8 @@ async def test_run_transfer_postgres_to_samba_with_incremental_strategy(
317317
await run_transfer_and_verify(client, group_owner, postgres_to_samba.id)
318318

319319
downloader.run()
320-
verify_file_name_template(list(Path.iterdir(tmp_path)), expected_extension)
320+
file_names = [f.name for f in Path.iterdir(tmp_path) if f.is_file()]
321+
verify_file_name_template(file_names, expected_extension)
321322

322323
df_with_increment = reader.run()
323324
df_with_increment, second_transfer_df = cast_dataframe_types(df_with_increment, second_transfer_df)

tests/test_integration/test_run_transfer/test_sftp.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -323,7 +323,8 @@ async def test_run_transfer_postgres_to_sftp_with_incremental_strategy(
323323
await run_transfer_and_verify(client, group_owner, postgres_to_sftp.id)
324324

325325
downloader.run()
326-
verify_file_name_template(list(Path.iterdir(tmp_path)), expected_extension)
326+
file_names = [f.name for f in Path.iterdir(tmp_path) if f.is_file()]
327+
verify_file_name_template(file_names, expected_extension)
327328

328329
df_with_increment = reader.run()
329330
df_with_increment, second_transfer_df = cast_dataframe_types(df_with_increment, init_df)

tests/test_integration/test_run_transfer/test_webdav.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -316,7 +316,8 @@ async def test_run_transfer_postgres_to_webdav_with_incremental_strategy(
316316
await run_transfer_and_verify(client, group_owner, postgres_to_webdav.id)
317317

318318
downloader.run()
319-
verify_file_name_template(list(Path.iterdir(tmp_path)), expected_extension)
319+
file_names = [f.name for f in Path.iterdir(tmp_path) if f.is_file()]
320+
verify_file_name_template(file_names, expected_extension)
320321

321322
df_with_increment = reader.run()
322323
df_with_increment, second_transfer_df = cast_dataframe_types(

0 commit comments

Comments
 (0)