Fix issue after rebase.

neuroinformatics-unit · Apr 10, 2024 · bafb5d0 · bafb5d0
1 parent 14911d3
commit bafb5d0
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/tests/tests_integration/test_ssh_file_transfer.py b/tests/tests_integration/test_ssh_file_transfer.py
@@ -83,7 +83,7 @@ def ssh_setup(self, pathtable_and_project):
         ssh_test_utils.setup_project_and_container_for_ssh(project)
         ssh_test_utils.setup_ssh_connection(project)
 
-        project.upload_all()
+        project.upload_rawdata()
 
         return [pathtable, project]
 
@@ -122,7 +122,9 @@ def test_combinations_filesystem_transfer(
             swap_last_folder_only=False,
         )[0]
 
-        transfer_function("rawdata", sub_names, ses_names, datatype, init_log=False)
+        transfer_function(
+            "rawdata", sub_names, ses_names, datatype, init_log=False
+        )
 
         if upload_or_download == "download":
             test_utils.swap_local_and_central_paths(
@@ -196,7 +198,9 @@ def test_combinations_ssh_transfer(
         )
         project.update_config("central_path", tmp_central_path)
 
-        project.upload(sub_names, ses_names, datatype, init_log=False)
+        project.upload_custom(
+            "rawdata", sub_names, ses_names, datatype, init_log=False
+        )
 
         expected_transferred_paths = self.get_expected_transferred_paths(
             pathtable, sub_names, ses_names, datatype
@@ -225,7 +229,9 @@ def test_combinations_ssh_transfer(
         project.update_config("local_path", tmp_local_path)
         project.update_config("central_path", true_central_path)
 
-        project.download(sub_names, ses_names, datatype, init_log=False)
+        project.download_custom(
+            "rawdata", sub_names, ses_names, datatype, init_log=False
+        )
 
         # Find the transferred paths, tidy them up
         # and check expected paths were transferred.
@@ -281,7 +287,7 @@ def get_expected_transferred_paths(
         expected_paths = pd.concat([datatype_folders, extra_folders])
         expected_paths = expected_paths.drop_duplicates(subset="path")
 
-        expected_paths = self.remove_path_before_rawdata(expected_paths)
+        expected_paths = self.remove_path_before_rawdata(expected_paths.path)
 
         return expected_paths
 
@@ -309,7 +315,7 @@ def make_pathtable_search_filter(self, sub_names, ses_names, datatype):
                         ]
                     else:
                         for dtype in datatype:
-                            if dtype == "all_ses_level_non_datatype":
+                            if dtype == "all_non_datatype":
                                 extra_arguments += [
                                     f"(parent_sub == '{sub}' & parent_ses == '{ses}' "
                                     f"& is_ses_level_non_datatype == True)"