@@ -88,8 +88,8 @@ def test_dataset_url_import_job(url, svc_client_with_repo):
8888@pytest .mark .service
8989@retry_failed
9090@pytest .mark .vcr
91- def test_dataset_import_job (doi , svc_client_with_repo ):
92- """Test dataset import via doi."""
91+ def test_dataset_import_job_with_slug (doi , svc_client_with_repo ):
92+ """Test dataset import via doi and give it a slug ."""
9393 svc_client , headers , project_id , url_components = svc_client_with_repo
9494
9595 user_id = encode_b64 (secure_filename ("9ab2fc80-3a5c-426d-ae78-56de01d214df" ))
@@ -118,7 +118,14 @@ def test_dataset_import_job(doi, svc_client_with_repo):
118118 job_id = response .json ["result" ]["job_id" ]
119119
120120 commit_message = "service: import remote dataset"
121- dataset_import (user , job_id , project_id , doi , commit_message = commit_message )
121+ dataset_import (
122+ user = user ,
123+ user_job_id = job_id ,
124+ project_id = project_id ,
125+ dataset_uri = doi ,
126+ slug = "dataset-slug" ,
127+ commit_message = commit_message ,
128+ )
122129
123130 new_commit = Repository (dest ).head .commit
124131 assert old_commit .hexsha != new_commit .hexsha
@@ -279,7 +286,16 @@ def test_dataset_add_remote_file(url, svc_client_with_repo):
279286 job_id = response .json ["result" ]["files" ][0 ]["job_id" ]
280287 commit_message = "service: dataset add remote file"
281288
282- dataset_add_remote_file (user , job_id , project_id , True , commit_message , payload ["slug" ], url )
289+ # user, user_job_id, project_id, create_dataset, commit_message, slug, url
290+ dataset_add_remote_file (
291+ user = user ,
292+ user_job_id = job_id ,
293+ project_id = project_id ,
294+ create_dataset = True ,
295+ commit_message = commit_message ,
296+ slug = payload ["slug" ],
297+ url = url ,
298+ )
283299
284300 new_commit = Repository (dest ).head .commit
285301
0 commit comments