Configure test coverage files and add tests

Signed-off-by: Zack Koppert <[email protected]>

Author: zkoppert

.coveragerc

@@ -0,0 +1,4 @@
+[run]
+omit =
+    # omit test files
+    test_*.py

evergreen.py

@@ -9,7 +9,7 @@
 from dependabot_file import build_dependabot_file
 
 
-def main():
+def main():  # pragma: no cover
     """Run the main program"""
 
     # Get the environment variables
@@ -174,4 +174,4 @@ def commit_changes(title, body, repo, dependabot_file):
 
 
 if __name__ == "__main__":
-    main()
+    main()  # pragma: no cover

test_evergreen.py

@@ -1,9 +1,13 @@
 """ Test the evergreen.py module. """
 import unittest
-from unittest.mock import patch
+from unittest.mock import MagicMock, patch
+import uuid
 
 from evergreen import (
+    check_pending_pulls_for_duplicates,
+    commit_changes,
     enable_dependabot_security_updates,
+    get_repos_iterator,
     is_dependabot_security_updates_enabled,
 )
 
@@ -176,5 +180,126 @@ def test_enable_dependabot_security_updates_failed(self):
                 )
 
 
+class TestCommitChanges(unittest.TestCase):
+    """Test the commit_changes function in evergreen.py"""
+
+    @patch("uuid.uuid4")
+    def test_commit_changes(self, mock_uuid):
+        """Test the commit_changes function."""
+        mock_uuid.return_value = uuid.UUID(
+            "12345678123456781234567812345678"
+        )  # Mock UUID generation
+        mock_repo = MagicMock()  # Mock repo object
+        mock_repo.default_branch = "main"
+        mock_repo.ref.return_value.object.sha = "abc123"  # Mock SHA for latest commit
+        mock_repo.create_ref.return_value = True
+        mock_repo.create_file.return_value = True
+        mock_repo.create_pull.return_value = "MockPullRequest"
+
+        title = "Test Title"
+        body = "Test Body"
+        dependabot_file = 'dependencies:\n  - package_manager: "python"\n    directory: "/"\n    update_schedule: "live"'
+        branch_name = "dependabot-12345678-1234-5678-1234-567812345678"
+        result = commit_changes(title, body, mock_repo, dependabot_file)
+
+        # Assert that the methods were called with the correct arguments
+        mock_repo.create_ref.assert_called_once_with(
+            f"refs/heads/{branch_name}", "abc123"
+        )
+        mock_repo.create_file.assert_called_once_with(
+            path=".github/dependabot.yaml",
+            message="Create dependabot.yaml",
+            content=dependabot_file.encode(),
+            branch=branch_name,
+        )
+        mock_repo.create_pull.assert_called_once_with(
+            title=title,
+            body=body,
+            head=branch_name,
+            base="main",
+        )
+
+        # Assert that the function returned the expected result
+        self.assertEqual(result, "MockPullRequest")
+
+
+class TestCheckPendingPullsForDuplicates(unittest.TestCase):
+    """Test the check_pending_pulls_for_duplicates function."""
+
+    def test_check_pending_pulls_for_duplicates_no_duplicates(self):
+        """Test the check_pending_pulls_for_duplicates function where there are no duplicates to be found."""
+        mock_repo = MagicMock()  # Mock repo object
+        mock_pull_request = MagicMock()
+        mock_pull_request.head.ref = "not-dependabot-branch"
+        mock_repo.pull_requests.return_value = [mock_pull_request]
+
+        result = check_pending_pulls_for_duplicates(mock_repo)
+
+        # Assert that the function returned the expected result
+        self.assertEqual(result, False)
+
+    def test_check_pending_pulls_for_duplicates_with_duplicates(self):
+        """Test the check_pending_pulls_for_duplicates function where there are duplicates to be found."""
+        mock_repo = MagicMock()  # Mock repo object
+        mock_pull_request = MagicMock()
+        mock_pull_request.head.ref = "dependabot-branch"
+        mock_repo.pull_requests.return_value = [mock_pull_request]
+
+        result = check_pending_pulls_for_duplicates(mock_repo)
+
+        # Assert that the function returned the expected result
+        self.assertEqual(result, True)
+
+
+class TestGetReposIterator(unittest.TestCase):
+    """Test the get_repos_iterator function in evergreen.py"""
+
+    @patch("github3.login")
+    def test_get_repos_iterator_with_organization(self, mock_github):
+        """Test the get_repos_iterator function with an organization"""
+        organization = "my_organization"
+        repository_list = []
+        github_connection = mock_github.return_value
+
+        mock_organization = MagicMock()
+        mock_repositories = MagicMock()
+        mock_organization.repositories.return_value = mock_repositories
+        github_connection.organization.return_value = mock_organization
+
+        result = get_repos_iterator(organization, repository_list, github_connection)
+
+        # Assert that the organization method was called with the correct argument
+        github_connection.organization.assert_called_once_with(organization)
+
+        # Assert that the repositories method was called on the organization object
+        mock_organization.repositories.assert_called_once()
+
+        # Assert that the function returned the expected result
+        self.assertEqual(result, mock_repositories)
+
+    @patch("github3.login")
+    def test_get_repos_iterator_with_repository_list(self, mock_github):
+        """Test the get_repos_iterator function with a repository list"""
+        organization = None
+        repository_list = ["org/repo1", "org/repo2"]
+        github_connection = mock_github.return_value
+
+        mock_repository = MagicMock()
+        mock_repository_list = [mock_repository, mock_repository]
+        github_connection.repository.side_effect = mock_repository_list
+
+        result = get_repos_iterator(organization, repository_list, github_connection)
+
+        # Assert that the repository method was called with the correct arguments for each repository in the list
+        expected_calls = [
+            unittest.mock.call("org", "repo1"),
+            unittest.mock.call("org", "repo2"),
+        ]
+        github_connection.repository.assert_has_calls(expected_calls)
+
+        # Assert that the function returned the expected result
+        self.assertEqual(result, mock_repository_list)
+
+
 if __name__ == "__main__":
     unittest.main()