add testing

Author: claytonparnell

src/sagemaker/processing.py

@@ -1730,7 +1730,7 @@ def _pack_and_upload_code(
                 "sagemaker_session unspecified when creating your Processor to have one set up "
                 "automatically."
             )
-        if ("sourcedir.tar.gz" in estimator.uploaded_code.s3_prefix):
+        if ("/sourcedir.tar.gz" in estimator.uploaded_code.s3_prefix):
             # Upload the bootstrapping code as s3://.../jobname/source/runproc.sh.
             entrypoint_s3_uri = estimator.uploaded_code.s3_prefix.replace(
                 "sourcedir.tar.gz",

tests/integ/test_xgboost.py

@@ -40,6 +40,26 @@ def xgboost_training_job(
     )
 
 
+def test_sourcedir_naming(
+    sagemaker_session,
+    xgboost_latest_version,
+    xgboost_latest_py_version,
+    cpu_instance_type,
+):
+    with pytest.raises(RuntimeError):
+        processor = XGBoostProcessor(
+            framework_version=xgboost_latest_version,
+            role=ROLE,
+            instance_count=1,
+            instance_type=cpu_instance_type,
+            sagemaker_session=sagemaker_session,
+        )
+        processor.run(
+            source_dir="s3://bucket/deps.tar.gz",
+            code="main_script.py",
+        )
+
+
 @pytest.mark.release
 def test_framework_processing_job_with_deps(
     sagemaker_session,