chore: import upstream snapshot with attribution

python/ray/tests/aws/conftest.py

@@ -0,0 +1,55 @@
+import pytest
+from botocore.stub import Stubber
+
+from ray.autoscaler._private.aws.utils import client_cache, resource_cache
+from ray.autoscaler._private.constants import BOTO_MAX_RETRIES
+
+
+@pytest.fixture()
+def iam_client_stub(request):
+    region = getattr(request, "param", "us-west-2")
+    resource = resource_cache("iam", region)
+    with Stubber(resource.meta.client) as stubber:
+        yield stubber
+        stubber.assert_no_pending_responses()
+
+
+@pytest.fixture()
+def ec2_client_stub(request):
+    region = getattr(request, "param", "us-west-2")
+    resource = resource_cache("ec2", region)
+    with Stubber(resource.meta.client) as stubber:
+        yield stubber
+        stubber.assert_no_pending_responses()
+
+
+@pytest.fixture()
+def ec2_client_stub_fail_fast():
+    resource = resource_cache("ec2", "us-west-2", 0)
+    with Stubber(resource.meta.client) as stubber:
+        yield stubber
+        stubber.assert_no_pending_responses()
+
+
+@pytest.fixture()
+def ec2_client_stub_max_retries():
+    resource = resource_cache("ec2", "us-west-2", BOTO_MAX_RETRIES)
+    with Stubber(resource.meta.client) as stubber:
+        yield stubber
+        stubber.assert_no_pending_responses()
+
+
+@pytest.fixture()
+def cloudwatch_client_stub():
+    resource = resource_cache("cloudwatch", "us-west-2")
+    with Stubber(resource.meta.client) as stubber:
+        yield stubber
+        stubber.assert_no_pending_responses()
+
+
+@pytest.fixture()
+def ssm_client_stub():
+    client = client_cache("ssm", "us-west-2")
+    with Stubber(client) as stubber:
+        yield stubber
+        stubber.assert_no_pending_responses()