更新trace功能报错跳过逻辑,完全删掉trace部分报错链路 (#562)

* 更新trace功能报错跳过逻辑,完全删掉trace部分报错链路 * 更新单元测试 --------- Co-authored-by: yinjiaqi <[email protected]>
baidubce · Oct 24, 2024 · 3e748fc · 3e748fc
1 parent 8489bfb
commit 3e748fc
Show file tree

Hide file tree

Showing 2 changed files with 211 additions and 50 deletions.
diff --git a/appbuilder/tests/test_trace_skip_raise_error.py b/appbuilder/tests/test_trace_skip_raise_error.py
@@ -133,6 +133,10 @@ def test_session_post(self):
         with self.assertRaises(TestException):
             mock_post_02()
 
+        os.environ["APPBUILDER_SDK_TRACE_ENABLE"] = "false"
+        with self.assertRaises(Exception):
+            mock_post_02()
+
     def test_client_run_trace(self):
         # test_client_run_trace APPBUILDER_TRACE_DEBUG = true
         os.environ["APPBUILDER_TRACE_DEBUG"] = "true"
@@ -145,6 +149,10 @@ def test_client_run_trace(self):
         with self.assertRaises(TestException):
             mock_client_run_trace_02()
 
+        os.environ["APPBUILDER_SDK_TRACE_ENABLE"] = "false"
+        with self.assertRaises(Exception):
+            mock_client_run_trace_02()
+
     def test_client_tool_trace(self):
         # test_client_tool_trace APPBUILDER_TRACE_DEBUG = true
         os.environ["APPBUILDER_TRACE_DEBUG"] = "true"
@@ -157,6 +165,10 @@ def test_client_tool_trace(self):
         with self.assertRaises(TestException):
             mock_client_tool_trace_02()
 
+        os.environ["APPBUILDER_SDK_TRACE_ENABLE"] = "false"
+        with self.assertRaises(Exception):
+            mock_client_tool_trace_02()
+
     def test_assistent_tool_trace(self):
         # test_assistent_tool_trace APPBUILDER_TRACE_DEBUG = true
         os.environ["APPBUILDER_TRACE_DEBUG"] = "true"
@@ -169,6 +181,10 @@ def test_assistent_tool_trace(self):
         with self.assertRaises(TestException):
             mock_assistent_tool_trace_02()
 
+        os.environ["APPBUILDER_SDK_TRACE_ENABLE"] = "false"
+        with self.assertRaises(Exception):
+            mock_assistent_tool_trace_02()
+
     def test_assistant_run_trace(self):
         # test_assistant_run_trace APPBUILDER_TRACE_DEBUG = true
         os.environ["APPBUILDER_TRACE_DEBUG"] = "true"
@@ -180,6 +196,10 @@ def test_assistant_run_trace(self):
             mock_assistant_run_trace_01()
         with self.assertRaises(TestException):
             mock_assistant_run_trace_02()
+
+        os.environ["APPBUILDER_SDK_TRACE_ENABLE"] = "false"
+        with self.assertRaises(Exception):
+            mock_assistant_run_trace_02()
 
     def test_assistent_stream_run_trace(self):
         # test_assistent_stream_run_trace APPBUILDER_TRACE_DEBUG = true
@@ -192,6 +212,10 @@ def test_assistent_stream_run_trace(self):
             mock_assistent_stream_run_trace_01()
         with self.assertRaises(TestException):
             mock_assistent_stream_run_trace_02()
+
+        os.environ["APPBUILDER_SDK_TRACE_ENABLE"] = "false"
+        with self.assertRaises(Exception):
+            mock_assistent_stream_run_trace_02()
 
     def test_assistent_stream_run_with_handler_trace(self):
         # test_assistent_stream_run_with_handler_trace APPBUILDER_TRACE_DEBUG = true
@@ -205,6 +229,10 @@ def test_assistent_stream_run_with_handler_trace(self):
         with self.assertRaises(TestException):
             mock_assistent_stream_run_with_handler_trace_02()
 
+        os.environ["APPBUILDER_SDK_TRACE_ENABLE"] = "false"
+        with self.assertRaises(Exception):
+            mock_assistent_stream_run_with_handler_trace_02()
+
     def test_components_run_trace(self):
         # test_components_run_trace APPBUILDER_TRACE_DEBUG = true
         os.environ["APPBUILDER_TRACE_DEBUG"] = "true"
@@ -216,6 +244,10 @@ def test_components_run_trace(self):
             mock_components_run_trace_01()
         with self.assertRaises(TestException):
             mock_components_run_trace_02()
+
+        os.environ["APPBUILDER_SDK_TRACE_ENABLE"] = "false"
+        with self.assertRaises(Exception):
+            mock_components_run_trace_02()
 
     def test_components_run_stream_trace(self):
         # test_components_run_stream_trace APPBUILDER_TRACE_DEBUG = true
@@ -229,6 +261,10 @@ def test_components_run_stream_trace(self):
         with self.assertRaises(TestException):
             mock_components_run_stream_trace_02()
 
+        os.environ["APPBUILDER_SDK_TRACE_ENABLE"] = "false"
+        with self.assertRaises(Exception):
+            mock_components_run_stream_trace_02()
+
     def test_list_trace(self):
         # test_list_trace APPBUILDER_TRACE_DEBUG = true
         os.environ["APPBUILDER_TRACE_DEBUG"] = "true"
@@ -241,6 +277,10 @@ def test_list_trace(self):
         with self.assertRaises(TestException):
             mock_list_trace_02()
 
+        os.environ["APPBUILDER_SDK_TRACE_ENABLE"] = "false"
+        with self.assertRaises(Exception):
+            mock_list_trace_02()
+
 
 if __name__ == '__main__':
     unittest.main()