update mockapi tests

Author: iscai-msft

packages/typespec-python/test/generic_mock_api_tests/asynctests/test_generation_subdir_async.py

@@ -4,10 +4,15 @@
 # license information.
 # --------------------------------------------------------------------------
 import pytest
-from authentication.api.key.subdir.aio import CustomizedApiKeyClient
+from generation.subdir import CustomizedClient, Extension
 
 
 @pytest.mark.asyncio
-async def test_custom_method(key_credential):
-    client = CustomizedApiKeyClient(key_credential("valid-key"))
-    assert await client.custom_method()
+async def test_custom_method():
+    client = CustomizedClient()
+    assert (await client.custom_method()) == Extension(
+        {
+            "level": 0,
+            "extension": [{"level": 1, "extension": [{"level": 2}]}, {"level": 1}],
+        }
+    )

packages/typespec-python/test/generic_mock_api_tests/test_generation_subdir.py

@@ -3,19 +3,14 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-import pytest
-from generation.subdir import CustomizedApiKeyClient
+from generation.subdir import CustomizedClient, Extension
 
 
-def test_custom_method(key_credential):
-    client = CustomizedApiKeyClient(key_credential("valid-key"))
-    assert client.custom_method()
-
-
-def test_custom_model():
-    try:
-        from authentication.api.key.subdir import InvalidAuth
-
-        assert InvalidAuth is not None
-    except ImportError:
-        pytest.fail("InvalidAuth could not be imported")
+def test_custom_method():
+    client = CustomizedClient()
+    assert client.custom_method() == Extension(
+        {
+            "level": 0,
+            "extension": [{"level": 1, "extension": [{"level": 2}]}, {"level": 1}],
+        }
+    )