Add test_activity_as_tool_extracts_activity_name_from_trigger test

Author: AnatoliB

tests/openai_agents/test_context.py

@@ -234,6 +234,48 @@ def test_activity_as_tool_run_activity_without_retry(self, mock_function_tool, m
         )
         assert result == "activity_result"
 
+    @patch('azure.durable_functions.openai_agents.context.function_schema')
+    @patch('azure.durable_functions.openai_agents.context.FunctionTool')
+    def test_activity_as_tool_extracts_activity_name_from_trigger(self, mock_function_tool, mock_function_schema):
+        """Test that the run_activity function calls task tracker with the activity name specified in the trigger."""
+        orchestration_context = self._create_mock_orchestration_context()
+        task_tracker = self._create_mock_task_tracker()
+
+        mock_activity_func = Mock()
+        mock_activity_func._function._name = "test_activity"
+        mock_activity_func._function._trigger.activity = "activity_name_from_trigger"
+        mock_activity_func._function._func = lambda x: x
+
+        mock_schema = Mock()
+        mock_schema.name = "test_activity"
+        mock_schema.description = ""
+        mock_schema.params_json_schema = {"type": "object"}
+        mock_function_schema.return_value = mock_schema
+
+        mock_tool = Mock(spec=FunctionTool)
+        mock_function_tool.return_value = mock_tool
+
+        ai_context = DurableAIAgentContext(orchestration_context, task_tracker, None)
+
+        ai_context.create_activity_tool(mock_activity_func, retry_options=None)
+
+        # Get the run_activity function that was passed to FunctionTool
+        call_args = mock_function_tool.call_args
+        run_activity = call_args[1]['on_invoke_tool']
+
+        # Create a mock context wrapper
+        mock_ctx = Mock()
+
+        # Call the run_activity function
+        import asyncio
+        result = asyncio.run(run_activity(mock_ctx, "test_input"))
+
+        # Verify the task tracker was called without retry options
+        task_tracker.get_activity_call_result.assert_called_once_with(
+            "activity_name_from_trigger", "test_input"
+        )
+        assert result == "activity_result"
+
     def test_context_delegation_methods_work(self):
         """Test that common context methods work through delegation."""
         orchestration_context = self._create_mock_orchestration_context()