From d65214525252bd6f4f3b35a1a6e237ab49611c4e Mon Sep 17 00:00:00 2001
From: Dengke <dengket@amazon.com>
Date: Fri, 30 Aug 2024 13:53:32 -0700
Subject: [PATCH] add the local test branch

---
 local-test.py | 71 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 local-test.py

diff --git a/local-test.py b/local-test.py
new file mode 100644
index 000000000..b4f1f94e1
--- /dev/null
+++ b/local-test.py
@@ -0,0 +1,71 @@
+import logging
+from awscrt import io
+from botocore.auth import *
+from botocore.awsrequest import AWSRequest
+from botocore.credentials import Credentials
+from botocore.session import get_session
+from botocore.crt.auth import *
+
+# Set up logging to capture trace-level logs
+logging.basicConfig()
+logging.getLogger('botocore').setLevel(logging.DEBUG)
+
+io.init_logging(io.LogLevel.Trace, 'stderr')
+
+
+def presign_post_request(
+        service_name,
+        region_name,
+        endpoint,
+        request_parameters,
+        access_key,
+        secret_key,
+        session_token=None):
+    # Create a session and get credentials
+    session = get_session()
+
+    # Create credentials object
+    credentials = Credentials(access_key, secret_key, session_token)
+
+    # Create the AWSRequest object
+    request = AWSRequest(
+        method="POST",
+        url=endpoint,
+        data=request_parameters,
+        headers={
+            "Host": "example.amazonaws.com",
+            'Content-Type': 'application/json',
+            "x-amz-content-sha256": "UNSIGNED-PAYLOAD",
+            'Content-Length': '123',
+        }
+    )
+    # request.context["payload_signing_enabled"] = False
+
+    # Sign the request using SigV4QueryAuth
+    # SigV4QueryAuth(credentials, service_name, region_name).add_auth(request)
+    CrtSigV4AsymQueryAuth(credentials, service_name, region_name).add_auth(request)
+
+    # The presigned URL is now in request.url
+    signed_request = request
+    return signed_request
+
+
+# Example usage
+service_name = 'vpc-lattice-svcs'
+region_name = 'us-west-2'
+endpoint = 'https://example.amazonaws.com/'
+request_parameters = '{\
+    "Param1": "value1"\
+}'
+access_key = 'your-access-key'
+secret_key = 'your-secret-key'
+session_token = None  # If using temporary credentials
+
+signed_request = presign_post_request(
+    service_name, region_name, endpoint, request_parameters, access_key, secret_key, session_token)
+
+print("Presigned URL:", signed_request.url)
+print("Headers:", signed_request.headers)
+print("Params:", signed_request.params)
+
+signed_request.prepare()