Add support for the reset state operation (#4)

Author: giograno

localstack-sdk-python/localstack/sdk/state/__init__.py

@@ -0,0 +1,3 @@
+from localstack.sdk.state.client import StateClient
+
+__all__ = ["StateClient"]

localstack-sdk-python/localstack/sdk/state/client.py

@@ -0,0 +1,11 @@
+from localstack.clients import BaseClient
+from localstack.sdk.api import StateApi
+
+
+class StateClient(BaseClient):
+    def __init__(self, **args) -> None:
+        super().__init__(**args)
+        self._client = StateApi(self._api_client)
+
+    def reset_state(self) -> None:
+        self._client.localstack_state_reset_post_0()

pyproject.toml

@@ -26,7 +26,8 @@ readme = { file = ["README.md"], content-type = "text/markdown"}
 [tool.uv]
 dev-dependencies=[
     "pytest",
-    "ruff"
+    "ruff",
+    "boto3"
 ]
 
 [tool.uv.sources]

tests/integration/test_state.py

@@ -0,0 +1,22 @@
+import boto3
+import pytest
+
+from localstack.sdk.state import StateClient
+
+
+class TestStateClient:
+    client = StateClient()
+
+    def test_reset_state(self):
+        sqs_client = boto3.client(
+            "sqs", endpoint_url=self.client.configuration.host, region_name="us-east-1"
+        )
+        sqs_client.create_queue(QueueName="test-queue")
+        url = sqs_client.get_queue_url(QueueName="test-queue")["QueueUrl"]
+        assert url
+
+        self.client.reset_state()
+
+        with pytest.raises(Exception) as exc:
+            sqs_client.get_queue_url(QueueName="test-queue")
+        assert "AWS.SimpleQueueService.NonExistentQueue" == exc.value.response["Error"]["Code"]  # noqa