Abstract away service name (#268)

* Abstract away service name Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Compuute x-amz-content-256 header Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Fix async signing Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Adds types-six to dependencies Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Optionally remove Content-Length Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Fix dict typo Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Remove requirement for x-amz-content-sha256 header Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Remove deletion of content-length Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Fix capitalization Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Adding unit tests Signed-off-by: Harsha Vamsi Kalluri <[email protected]> Signed-off-by: Harsha Vamsi Kalluri <[email protected]>
2023-01-18 18:06:59 -05:00
parent c73e46381e
commit 5c1c890f69
11 changed files with 132 additions and 36 deletions
@@ -63,3 +63,18 @@ class TestAsyncSigner(TestCase):

        with pytest.raises(ValueError) as e:
            assert str(e.value) == "Credentials cannot be empty"
+
+    @pytest.mark.skipif(
+        sys.version_info < (3, 6), reason="AWSV4SignerAsyncAuth requires python3.6+"
+    )
+    async def test_aws_signer_async_when_service_is_specified(self):
+        region = "us-west-2"
+        service = "aoss"
+
+        from opensearchpy.helpers.asyncsigner import AWSV4SignerAsyncAuth
+
+        auth = AWSV4SignerAsyncAuth(self.mock_session(), region, service)
+        headers = auth("GET", "http://localhost")
+        self.assertIn("Authorization", headers)
+        self.assertIn("X-Amz-Date", headers)
+        self.assertIn("X-Amz-Security-Token", headers)