Fix replace keys for nested json (#16557)

Azure · Feb 5, 2021 · a6a5074 · a6a5074
1 parent 6f497d6
commit a6a5074
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/sdk/communication/azure-communication-identity/tests/_shared/testcase.py b/sdk/communication/azure-communication-identity/tests/_shared/testcase.py
@@ -51,13 +51,12 @@ def process_response(self, response):
 
     def _replace_keys(self, body):
         def _replace_recursively(dictionary):
-            if type(dictionary) != dict:
-                return
             for key in dictionary:
+                value = dictionary[key]
                 if key in self._keys:
                     dictionary[key] = self._replacement
-                else:
-                    _replace_recursively(dictionary[key])
+                elif isinstance(value, dict):
+                    _replace_recursively(value)
 
         import json
         try: