add tests for four more tools

Author: julian-risch

integrations/github/tests/test_issue_commenter_tool.py

@@ -0,0 +1,54 @@
+# SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+from haystack.utils import Secret
+
+from haystack_integrations.prompts.github.issue_commenter_prompt import ISSUE_COMMENTER_PROMPT, ISSUE_COMMENTER_SCHEMA
+from haystack_integrations.tools.github.issue_commenter_tool import GitHubIssueCommenterTool
+
+
+class TestGitHubIssueCommenterTool:
+    def test_init(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool = GitHubIssueCommenterTool()
+        assert tool.name == "issue_commenter"
+        assert tool.description == ISSUE_COMMENTER_PROMPT
+        assert tool.parameters == ISSUE_COMMENTER_SCHEMA
+        assert tool.retry_attempts == 2
+
+    def test_from_dict(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool_dict = {
+            "type": "haystack_integrations.tools.github.issue_commenter_tool.GitHubIssueCommenterTool",
+            "init_parameters": {
+                "name": "issue_commenter",
+                "description": ISSUE_COMMENTER_PROMPT,
+                "parameters": ISSUE_COMMENTER_SCHEMA,
+                "github_token": {"env_vars": ["GITHUB_TOKEN"], "strict": True, "type": "env_var"},
+                "raise_on_failure": True,
+                "retry_attempts": 2,
+            },
+        }
+        tool = GitHubIssueCommenterTool.from_dict(tool_dict)
+        assert tool.name == "issue_commenter"
+        assert tool.description == ISSUE_COMMENTER_PROMPT
+        assert tool.parameters == ISSUE_COMMENTER_SCHEMA
+        assert tool.github_token == Secret.from_env_var("GITHUB_TOKEN")
+        assert tool.raise_on_failure
+        assert tool.retry_attempts == 2
+
+    def test_to_dict(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool = GitHubIssueCommenterTool()
+        tool_dict = tool.to_dict()
+        assert tool_dict["type"] == "haystack_integrations.tools.github.issue_commenter_tool.GitHubIssueCommenterTool"
+        assert tool_dict["init_parameters"]["name"] == "issue_commenter"
+        assert tool_dict["init_parameters"]["description"] == ISSUE_COMMENTER_PROMPT
+        assert tool_dict["init_parameters"]["parameters"] == ISSUE_COMMENTER_SCHEMA
+        assert tool_dict["init_parameters"]["github_token"] == {
+            "env_vars": ["GITHUB_TOKEN"],
+            "strict": True,
+            "type": "env_var",
+        }
+        assert tool_dict["init_parameters"]["raise_on_failure"]
+        assert tool_dict["init_parameters"]["retry_attempts"] == 2

integrations/github/tests/test_issue_viewer_tool.py

@@ -0,0 +1,48 @@
+# SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+from haystack_integrations.prompts.github.issue_viewer_prompt import ISSUE_VIEWER_PROMPT, ISSUE_VIEWER_SCHEMA
+from haystack_integrations.tools.github.issue_viewer_tool import GitHubIssueViewerTool
+
+
+class TestGitHubIssueViewerTool:
+    def test_init(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool = GitHubIssueViewerTool()
+        assert tool.name == "issue_viewer"
+        assert tool.description == ISSUE_VIEWER_PROMPT
+        assert tool.parameters == ISSUE_VIEWER_SCHEMA
+        assert tool.retry_attempts == 2
+
+    def test_from_dict(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool_dict = {
+            "type": "haystack_integrations.tools.github.issue_viewer_tool.GitHubIssueViewerTool",
+            "init_parameters": {
+                "name": "issue_viewer",
+                "description": ISSUE_VIEWER_PROMPT,
+                "parameters": ISSUE_VIEWER_SCHEMA,
+                "github_token": None,
+                "raise_on_failure": True,
+                "retry_attempts": 2,
+            },
+        }
+        tool = GitHubIssueViewerTool.from_dict(tool_dict)
+        assert tool.name == "issue_viewer"
+        assert tool.description == ISSUE_VIEWER_PROMPT
+        assert tool.parameters == ISSUE_VIEWER_SCHEMA
+        assert tool.github_token is None
+        assert tool.raise_on_failure
+        assert tool.retry_attempts == 2
+
+    def test_to_dict(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool = GitHubIssueViewerTool()
+        tool_dict = tool.to_dict()
+        assert tool_dict["type"] == "haystack_integrations.tools.github.issue_viewer_tool.GitHubIssueViewerTool"
+        assert tool_dict["init_parameters"]["name"] == "issue_viewer"
+        assert tool_dict["init_parameters"]["description"] == ISSUE_VIEWER_PROMPT
+        assert tool_dict["init_parameters"]["parameters"] == ISSUE_VIEWER_SCHEMA
+        assert tool_dict["init_parameters"]["github_token"] is None
+        assert tool_dict["init_parameters"]["raise_on_failure"]
+        assert tool_dict["init_parameters"]["retry_attempts"] == 2

integrations/github/tests/test_pr_creator_tool.py

@@ -0,0 +1,50 @@
+# SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+from haystack.utils import Secret
+
+from haystack_integrations.prompts.github.pr_creator_prompt import PR_CREATOR_PROMPT, PR_CREATOR_SCHEMA
+from haystack_integrations.tools.github.pr_creator_tool import GitHubPRCreatorTool
+
+
+class TestGitHubPRCreatorTool:
+    def test_init(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool = GitHubPRCreatorTool()
+        assert tool.name == "pr_creator"
+        assert tool.description == PR_CREATOR_PROMPT
+        assert tool.parameters == PR_CREATOR_SCHEMA
+
+    def test_from_dict(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool_dict = {
+            "type": "haystack_integrations.tools.github.pr_creator_tool.GitHubPRCreatorTool",
+            "init_parameters": {
+                "name": "pr_creator",
+                "description": PR_CREATOR_PROMPT,
+                "parameters": PR_CREATOR_SCHEMA,
+                "github_token": {"env_vars": ["GITHUB_TOKEN"], "strict": True, "type": "env_var"},
+                "raise_on_failure": True,
+            },
+        }
+        tool = GitHubPRCreatorTool.from_dict(tool_dict)
+        assert tool.name == "pr_creator"
+        assert tool.description == PR_CREATOR_PROMPT
+        assert tool.parameters == PR_CREATOR_SCHEMA
+        assert tool.github_token == Secret.from_env_var("GITHUB_TOKEN")
+        assert tool.raise_on_failure
+
+    def test_to_dict(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool = GitHubPRCreatorTool()
+        tool_dict = tool.to_dict()
+        assert tool_dict["type"] == "haystack_integrations.tools.github.pr_creator_tool.GitHubPRCreatorTool"
+        assert tool_dict["init_parameters"]["name"] == "pr_creator"
+        assert tool_dict["init_parameters"]["description"] == PR_CREATOR_PROMPT
+        assert tool_dict["init_parameters"]["parameters"] == PR_CREATOR_SCHEMA
+        assert tool_dict["init_parameters"]["github_token"] == {
+            "env_vars": ["GITHUB_TOKEN"],
+            "strict": True,
+            "type": "env_var",
+        }
+        assert tool_dict["init_parameters"]["raise_on_failure"]

integrations/github/tests/test_repo_viewer_tool.py

@@ -0,0 +1,54 @@
+# SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+from haystack_integrations.prompts.github.repo_viewer_prompt import REPO_VIEWER_PROMPT, REPO_VIEWER_SCHEMA
+from haystack_integrations.tools.github.repo_viewer_tool import GitHubRepoViewerTool
+
+
+class TestGitHubRepoViewerTool:
+    def test_init(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool = GitHubRepoViewerTool()
+        assert tool.name == "repo_viewer"
+        assert tool.description == REPO_VIEWER_PROMPT
+        assert tool.parameters == REPO_VIEWER_SCHEMA
+        assert tool.max_file_size == 1_000_000
+
+    def test_from_dict(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool_dict = {
+            "type": "haystack_integrations.tools.github.repo_viewer_tool.GitHubRepoViewerTool",
+            "init_parameters": {
+                "name": "repo_viewer",
+                "description": REPO_VIEWER_PROMPT,
+                "parameters": REPO_VIEWER_SCHEMA,
+                "github_token": None,
+                "repo": None,
+                "branch": "main",
+                "raise_on_failure": True,
+                "max_file_size": 1_000_000,
+            },
+        }
+        tool = GitHubRepoViewerTool.from_dict(tool_dict)
+        assert tool.name == "repo_viewer"
+        assert tool.description == REPO_VIEWER_PROMPT
+        assert tool.parameters == REPO_VIEWER_SCHEMA
+        assert tool.github_token is None
+        assert tool.repo is None
+        assert tool.branch == "main"
+        assert tool.raise_on_failure
+        assert tool.max_file_size == 1_000_000
+
+    def test_to_dict(self, monkeypatch):
+        monkeypatch.setenv("GITHUB_TOKEN", "test-token")
+        tool = GitHubRepoViewerTool()
+        tool_dict = tool.to_dict()
+        assert tool_dict["type"] == "haystack_integrations.tools.github.repo_viewer_tool.GitHubRepoViewerTool"
+        assert tool_dict["init_parameters"]["name"] == "repo_viewer"
+        assert tool_dict["init_parameters"]["description"] == REPO_VIEWER_PROMPT
+        assert tool_dict["init_parameters"]["parameters"] == REPO_VIEWER_SCHEMA
+        assert tool_dict["init_parameters"]["github_token"] is None
+        assert tool_dict["init_parameters"]["repo"] is None
+        assert tool_dict["init_parameters"]["branch"] == "main"
+        assert tool_dict["init_parameters"]["raise_on_failure"]
+        assert tool_dict["init_parameters"]["max_file_size"] == 1_000_000