pyright p2

microsoft · Sep 12, 2023 · cafe6b5 · cafe6b5
1 parent 10a82a6
commit cafe6b5
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/pythonFiles/unittestadapter/discovery.py b/pythonFiles/unittestadapter/discovery.py
@@ -141,8 +141,13 @@ def post_response(payload: PayloadDict | EOTPayloadDict, port: int, uuid: str) -
     # Perform test discovery.
     port, uuid = parse_discovery_cli_args(argv[:index])
     # Post this discovery payload.
-    payload = discover_tests(start_dir, pattern, top_level_dir, uuid)
-    post_response(payload, port, uuid)
-    # Post EOT token.
-    eot_payload: EOTPayloadDict = {"command_type": "discovery", "eot": True}
-    post_response(eot_payload, port, uuid)
+    if uuid is not None:
+        payload = discover_tests(start_dir, pattern, top_level_dir, uuid)
+        post_response(payload, port, uuid)
+        # Post EOT token.
+        eot_payload: EOTPayloadDict = {"command_type": "discovery", "eot": True}
+        post_response(eot_payload, port, uuid)
+    else:
+        print("Error: no uuid provided or parsed.")
+        eot_payload: EOTPayloadDict = {"command_type": "discovery", "eot": True}
+        post_response(eot_payload, port, "")