Skip to content

Commit

Permalink
updates tests
Browse files Browse the repository at this point in the history
  • Loading branch information
BWMac committed Jan 30, 2025
1 parent 5421cbd commit a98a688
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions tests/test_process.py
Original file line number Diff line number Diff line change
Expand Up @@ -471,6 +471,7 @@ def setup_method(self):
"gx_folder": GX_FOLDER,
"gx_table": "syn321",
"staging_path": STAGING_PATH,
"team_images_id": "syn987",
"datasets": [{"a": {"b": "c"}}, {"d": {"e": "f"}}, {"g": {"h": "i"}}],
},
).start()
Expand All @@ -496,6 +497,9 @@ def setup_method(self):
load, "df_to_csv", return_value="path/to/csv"
).start()
self.patch_load = patch.object(load, "load", return_value=("syn123", 1)).start()
self.patch_upload_dataversion_metadata = patch.object(
process, "upload_dataversion_metadata", return_value=None
).start()
self.patch_update_table = patch.object(
ADTGXReporter,
"update_table",
Expand Down Expand Up @@ -546,6 +550,7 @@ def test_process_all_files_upload_false(self, syn: Any):
staging_path=STAGING_PATH,
filename="data_manifest.csv",
)
self.patch_upload_dataversion_metadata.assert_not_called()
self.patch_load.assert_not_called()
self.patch_format_link.assert_not_called()
self.patch_update_table.assert_called_once()
Expand Down Expand Up @@ -592,6 +597,14 @@ def test_process_all_files_upload_true(self, syn: Any):
staging_path=STAGING_PATH,
filename="data_manifest.csv",
)
self.patch_upload_dataversion_metadata.assert_called_once_with(
syn=syn,
file_id="syn123",
file_version=1,
team_images_id="syn987",
staging_path=STAGING_PATH,
destination="destination",
)
self.patch_load.assert_called_once_with(
file_path="path/to/csv",
provenance=["a", "b", "c"],
Expand Down Expand Up @@ -644,6 +657,7 @@ def test_process_all_files_upload_false_gx_failure(self, syn: Any):
)
self.patch_create_data_manifest.assert_not_called()
self.patch_df_to_csv.assert_not_called()
self.patch_upload_dataversion_metadata.assert_not_called()
self.patch_load.assert_not_called()
self.patch_format_link.assert_not_called()
self.patch_update_table.assert_called_once()
Expand Down Expand Up @@ -688,6 +702,7 @@ def test_process_all_files_upload_false_process_dataset_fail(self, syn: Any):
)
self.patch_create_data_manifest.assert_not_called()
self.patch_df_to_csv.assert_not_called()
self.patch_upload_dataversion_metadata.assert_not_called()
self.patch_load.assert_not_called()
self.patch_format_link.assert_not_called()
self.patch_update_table.assert_called_once()

0 comments on commit a98a688

Please sign in to comment.