From 313caae0918fdba8de707318b548bb838f256811 Mon Sep 17 00:00:00 2001
From: Derek Kulinski <d@kulinski.us>
Date: Thu, 5 Oct 2023 17:08:10 -0700
Subject: [PATCH] unit tests

---
 tests/test_stubber.py | 65 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 tests/test_stubber.py

diff --git a/tests/test_stubber.py b/tests/test_stubber.py
new file mode 100644
index 00000000..777176b9
--- /dev/null
+++ b/tests/test_stubber.py
@@ -0,0 +1,65 @@
+import pytest
+
+from aiobotocore.awsrequest import AioAWSResponse
+from aiobotocore.session import AioSession
+from aiobotocore.stub import AioStubber
+
+from .mock_server import AIOServer
+
+
+@pytest.mark.asyncio
+async def test_add_response():
+    session = AioSession()
+
+    async with AIOServer() as server, session.create_client(
+        's3',
+        endpoint_url=server.endpoint_url,
+        aws_secret_access_key='xxx',
+        aws_access_key_id='xxx',
+    ) as s3_client:
+        stubber = AioStubber(s3_client)
+        operation_name = 'put_object'
+        service_response = dict(
+            ETag="6805f2cfc46c0f04559748bb039d69ae",
+            VersionId="psM2sYY4.o1501dSx8wMvnkOzSBB.V4a",
+        )
+        expected_params = dict()
+        stubber.add_response(operation_name, service_response, expected_params)
+
+        assert len(stubber._queue) == 1
+        assert stubber._queue[0][
+            'operation_name'
+        ] == s3_client.meta.method_to_api_mapping.get(operation_name)
+        assert isinstance(stubber._queue[0]['response'][0], AioAWSResponse)
+        assert stubber._queue[0]['response'][1] == service_response
+        assert stubber._queue[0]['expected_params'] == expected_params
+
+
+@pytest.mark.asyncio
+async def test_add_client_error():
+    session = AioSession()
+
+    async with AIOServer() as server, session.create_client(
+        's3',
+        endpoint_url=server.endpoint_url,
+        aws_secret_access_key='xxx',
+        aws_access_key_id='xxx',
+    ) as s3_client:
+        stubber = AioStubber(s3_client)
+        operation_name = 'put_object'
+        service_error_code = 'NoSuchBucket'
+        service_message = 'The specified bucket does not exist.'
+        http_status_code = 404
+        stubber.add_client_error(
+            operation_name,
+            service_error_code,
+            service_message,
+            http_status_code,
+        )
+
+        assert len(stubber._queue) == 1
+        assert stubber._queue[0][
+            'operation_name'
+        ] == s3_client.meta.method_to_api_mapping.get(operation_name)
+        assert isinstance(stubber._queue[0]['response'][0], AioAWSResponse)
+        assert stubber._queue[0]['response'][1]