update inference integ test to add dependency for testing wait until inservice (#121)

Author: mollyheamazon

setup.py

@@ -81,6 +81,7 @@
         "pytest==8.3.2",
         "pytest-cov==5.0.0",
         "pytest-order==1.3.0",
+        "pytest-dependency==0.6.0",
         "tox==4.18.0",
         "ruff==0.6.2",
         "hera-workflows==5.16.3",

test/integration_tests/inference/cli/test_cli_custom_fsx_inference.py

@@ -43,7 +43,7 @@ def sagemaker_client():
     return boto3.client("sagemaker", region_name=REGION)
 
 # --------- Custom Endpoint Tests ---------
-
+@pytest.mark.dependency(name="create")
 def test_custom_create(runner, custom_endpoint_name):
     result = runner.invoke(custom_create, [
         "--namespace", NAMESPACE,
@@ -72,6 +72,7 @@ def test_custom_list(runner, custom_endpoint_name):
     assert custom_endpoint_name in result.output
 
 
+@pytest.mark.dependency(name="describe")
 def test_custom_describe(runner, custom_endpoint_name):
     result = runner.invoke(custom_describe, [
         "--name", custom_endpoint_name,
@@ -82,6 +83,7 @@ def test_custom_describe(runner, custom_endpoint_name):
     assert custom_endpoint_name in result.output
 
 
+@pytest.mark.dependency(depends=["create", "describe"])
 def test_wait_until_inservice(custom_endpoint_name):
     """Poll SDK until specific JumpStart endpoint reaches DeploymentComplete"""
     print(f"[INFO] Waiting for JumpStart endpoint '{custom_endpoint_name}' to be DeploymentComplete...")

test/integration_tests/inference/cli/test_cli_custom_inference.py

@@ -43,7 +43,7 @@ def sagemaker_client():
     return boto3.client("sagemaker", region_name=REGION)
 
 # --------- Custom Endpoint Tests ---------
-
+@pytest.mark.dependency(name="create")
 def test_custom_create(runner, custom_endpoint_name):
     result = runner.invoke(custom_create, [
         "--namespace", NAMESPACE,
@@ -72,6 +72,7 @@ def test_custom_list(runner, custom_endpoint_name):
     assert custom_endpoint_name in result.output
 
 
+@pytest.mark.dependency(name="describe")
 def test_custom_describe(runner, custom_endpoint_name):
     result = runner.invoke(custom_describe, [
         "--name", custom_endpoint_name,
@@ -82,6 +83,7 @@ def test_custom_describe(runner, custom_endpoint_name):
     assert custom_endpoint_name in result.output
 
 
+@pytest.mark.dependency(depends=["create", "describe"])
 def test_wait_until_inservice(custom_endpoint_name):
     """Poll SDK until specific JumpStart endpoint reaches DeploymentComplete"""
     print(f"[INFO] Waiting for JumpStart endpoint '{custom_endpoint_name}' to be DeploymentComplete...")

test/integration_tests/inference/cli/test_cli_custom_s3_inference.py

@@ -28,7 +28,7 @@ def sagemaker_client():
     return boto3.client("sagemaker", region_name=REGION)
 
 # --------- JumpStart Endpoint Tests ---------
-
+@pytest.mark.dependency(name="create")
 def test_js_create(runner, js_endpoint_name):
     result = runner.invoke(js_create, [
         "--namespace", NAMESPACE,
@@ -46,6 +46,7 @@ def test_js_list(runner, js_endpoint_name):
     assert js_endpoint_name in result.output
 
 
+@pytest.mark.dependency(name="describe")
 def test_js_describe(runner, js_endpoint_name):
     result = runner.invoke(js_describe, [
         "--name", js_endpoint_name,
@@ -56,6 +57,7 @@ def test_js_describe(runner, js_endpoint_name):
     assert js_endpoint_name in result.output
 
 
+@pytest.mark.dependency(depends=["create", "describe"])
 def test_wait_until_inservice(js_endpoint_name):
     """Poll SDK until specific JumpStart endpoint reaches DeploymentComplete"""
     print(f"[INFO] Waiting for JumpStart endpoint '{js_endpoint_name}' to be DeploymentComplete...")

test/integration_tests/inference/cli/test_cli_jumpstart_inference.py

@@ -84,6 +84,7 @@ def custom_endpoint():
         worker=worker,
     )
 
+@pytest.mark.dependency(name="create")
 def test_create_endpoint(custom_endpoint):
     custom_endpoint.create(namespace=NAMESPACE)
     assert custom_endpoint.metadata.name == ENDPOINT_NAME
@@ -93,10 +94,12 @@ def test_list_endpoint():
     names = [ep.metadata.name for ep in endpoints]
     assert ENDPOINT_NAME in names
 
+@pytest.mark.dependency(name="describe")
 def test_get_endpoint():
     ep = HPEndpoint.get(name=ENDPOINT_NAME, namespace=NAMESPACE)
     assert ep.modelName == MODEL_NAME
 
+@pytest.mark.dependency(depends=["create", "describe"])
 def test_wait_until_inservice():
     """Poll SDK until specific JumpStart endpoint reaches DeploymentComplete"""
     print(f"[INFO] Waiting for JumpStart endpoint '{ENDPOINT_NAME}' to be DeploymentComplete...")