chore: Add unit test coverage

Author: anubhav756

packages/toolbox-core/tests/test_sync_tool.py

@@ -237,6 +237,39 @@ def test_toolbox_sync_tool_add_auth_token_getters(
     )
 
 
+def test_toolbox_sync_tool_add_auth_token_getter(
+    toolbox_sync_tool: ToolboxSyncTool,
+    mock_async_tool: MagicMock,
+    event_loop: asyncio.AbstractEventLoop,
+    mock_thread: MagicMock,
+):
+    """Tests the add_auth_token_getter method."""
+    auth_service = "service1"
+    auth_token_getter = lambda: "token1"
+
+    new_mock_async_tool = mock_async_tool.add_auth_token_getters.return_value
+    new_mock_async_tool.__name__ = "new_async_tool_with_auth"
+
+    new_sync_tool = toolbox_sync_tool.add_auth_token_getter(
+        auth_service, auth_token_getter
+    )
+
+    mock_async_tool.add_auth_token_getters.assert_called_once_with(
+        {auth_service: auth_token_getter}
+    )
+
+    assert isinstance(new_sync_tool, ToolboxSyncTool)
+    assert new_sync_tool is not toolbox_sync_tool
+    assert new_sync_tool._ToolboxSyncTool__async_tool is new_mock_async_tool
+    assert new_sync_tool._ToolboxSyncTool__loop is event_loop  # Should be the same loop
+    assert (
+        new_sync_tool._ToolboxSyncTool__thread is mock_thread
+    )  # Should be the same thread
+    assert (
+        new_sync_tool.__qualname__ == f"ToolboxSyncTool.{new_mock_async_tool.__name__}"
+    )
+
+
 def test_toolbox_sync_tool_bind_params(
     toolbox_sync_tool: ToolboxSyncTool,
     mock_async_tool: MagicMock,

packages/toolbox-core/tests/test_tool.py

@@ -502,6 +502,35 @@ def test_add_auth_token_getters_unused_token(
         tool_instance.add_auth_token_getters(unused_auth_getters)
 
 
+def test_add_auth_token_getter_unused_token(
+    http_session: ClientSession,
+    sample_tool_params: list[ParameterSchema],
+    sample_tool_description: str,
+    unused_auth_getters: Mapping[str, Callable[[], str]],
+):
+    """
+    Tests ValueError when add_auth_token_getters is called with a getter for
+    an unused authentication service.
+    """
+    tool_instance = ToolboxTool(
+        session=http_session,
+        base_url=HTTPS_BASE_URL,
+        name=TEST_TOOL_NAME,
+        description=sample_tool_description,
+        params=sample_tool_params,
+        required_authn_params={},
+        required_authz_tokens=[],
+        auth_service_token_getters={},
+        bound_params={},
+        client_headers={},
+    )
+
+    expected_error_message = "Authentication source\(s\) \`unused-auth-service\` unused by tool \`sample_tool\`."
+
+    with pytest.raises(ValueError, match=expected_error_message):
+        tool_instance.add_auth_token_getter(next(iter(unused_auth_getters)), unused_auth_getters[next(iter(unused_auth_getters))])
+
+
 def test_toolbox_tool_underscore_name_property(toolbox_tool: ToolboxTool):
     """Tests the _name property."""
     assert toolbox_tool._name == TEST_TOOL_NAME