Add basic test for ElastiCache IAM provider

alisaifee · Dec 6, 2024 · 4054e6e · 4054e6e
1 parent 6a4e40b
commit 4054e6e
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 0 deletions.
diff --git a/requirements/dev.txt b/requirements/dev.txt
@@ -1,5 +1,6 @@
 -r test.txt
 -r publishing.txt
+-r recipes.txt
 black==24.10.0
 ruff
 isort==5.13.2

diff --git a/requirements/recipes.txt b/requirements/recipes.txt
@@ -0,0 +1,3 @@
+botocore
+cachetools
+
diff --git a/requirements/test.txt b/requirements/test.txt
@@ -12,3 +12,6 @@ pytest-lazy-fixtures
 pytest-mock
 pytest-reverse
 redis
+
+# Recipe test dependencies
+moto
diff --git a/tests/recipes/credentials/__init__.py b/tests/recipes/credentials/__init__.py
diff --git a/tests/recipes/credentials/test_elasticache_iam_provider.py b/tests/recipes/credentials/test_elasticache_iam_provider.py
@@ -0,0 +1,14 @@
+from __future__ import annotations
+
+from moto import mock_aws
+
+from coredis.recipes.credentials import ElastiCacheIAMProvider
+
+
+class TestElastiCacheIAMProvider:
+    async def test_get_credentials(self):
+        with mock_aws():
+            provider = ElastiCacheIAMProvider("test_user", "test_cluster")
+            user_pass = await provider.get_credentials()
+            assert user_pass.username == "test_user"
+            assert "test_cluster/?Action=connect&User=test_user" in user_pass.password