Verify JSON attributes in Key Vault test playback (#9918)

sdk/keyvault/azure-keyvault-certificates/tests/_shared/json_attribute_matcher.py

@@ -0,0 +1,18 @@
+# ------------------------------------
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+# ------------------------------------
+import json
+
+import six
+
+_has_json_body = lambda req: req.body and "json" in req.headers.get("Content-Type", "")
+
+
+def json_attribute_matcher(r1, r2):
+    """Tests whether two vcr.py requests have JSON content with identical attributes (ignoring values)."""
+
+    if _has_json_body(r1) and _has_json_body(r2):
+        c1 = json.loads(six.ensure_str(r1.body))
+        c2 = json.loads(six.ensure_str(r2.body))
+        assert sorted(c1.keys()) == sorted(c2.keys())

sdk/keyvault/azure-keyvault-certificates/tests/test_merge_certificate.py

@@ -9,13 +9,15 @@
 from devtools_testutils import ResourceGroupPreparer, KeyVaultPreparer
 from OpenSSL import crypto
 
+from _shared.json_attribute_matcher import json_attribute_matcher
 from _shared.preparer import KeyVaultClientPreparer
 from _shared.test_case import KeyVaultTestCase
 
 
 class MergeCertificateTest(KeyVaultTestCase):
     def __init__(self, *args, **kwargs):
         kwargs["match_body"] = False
+        kwargs["custom_request_matchers"] = [json_attribute_matcher]
         super(MergeCertificateTest, self).__init__(*args, **kwargs)
 
     @ResourceGroupPreparer(random_name_enabled=True)

sdk/keyvault/azure-keyvault-certificates/tests/test_merge_certificate_async.py

@@ -10,13 +10,14 @@
 from devtools_testutils import ResourceGroupPreparer, KeyVaultPreparer
 from OpenSSL import crypto
 
+from _shared.json_attribute_matcher import json_attribute_matcher
 from _shared.preparer_async import KeyVaultClientPreparer
 from _shared.test_case_async import KeyVaultTestCase
 
 
 class MergeCertificateTest(KeyVaultTestCase):
     def __init__(self, *args, **kwargs):
-        super().__init__(*args, match_body=False, **kwargs)
+        super().__init__(*args, match_body=False, custom_request_matchers=[json_attribute_matcher], **kwargs)
 
     @ResourceGroupPreparer(random_name_enabled=True)
     @KeyVaultPreparer()

sdk/keyvault/azure-keyvault-keys/tests/_shared/json_attribute_matcher.py

@@ -0,0 +1,18 @@
+# ------------------------------------
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+# ------------------------------------
+import json
+
+import six
+
+_has_json_body = lambda req: req.body and "json" in req.headers.get("Content-Type", "")
+
+
+def json_attribute_matcher(r1, r2):
+    """Tests whether two vcr.py requests have JSON content with identical attributes (ignoring values)."""
+
+    if _has_json_body(r1) and _has_json_body(r2):
+        c1 = json.loads(six.ensure_str(r1.body))
+        c2 = json.loads(six.ensure_str(r2.body))
+        assert sorted(c1.keys()) == sorted(c2.keys())