fixing the test for empty secret with new format

zillow · May 7, 2024 · d4e7d9f · d4e7d9f
1 parent 567c657
commit d4e7d9f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/zdatasets/tests/utils/test_secret_fetcher.py b/zdatasets/tests/utils/test_secret_fetcher.py
@@ -109,8 +109,11 @@ def test_fetch_aws_secret():
         SecretFetcher(aws_secret_arn="not-json-decodable", key="key").value
 
     # Empty string
-    with pytest.raises(ValueError):
-        SecretFetcher(aws_secret_arn="empty").value
+    with pytest.raises(ValueError) as ve:
+        SecretFetcher(aws_secret_arn="empty")
+    assert (
+            "Secrets Manager can't find the specified secret value for staging label: AWSCURRENT"
+            == ve.value.response["Error"]["Message"])
 
 
 @mock.patch("zdatasets.utils.secret_fetcher.get_current_namespace")