From 79a399568d3fef95909732adefdd96c2c40ef004 Mon Sep 17 00:00:00 2001
From: Zack Koppert <zkoppert@github.com>
Date: Mon, 29 Apr 2024 13:41:27 -0700
Subject: [PATCH] test: Add test to cover ignored users in mentor_count

Signed-off-by: Zack Koppert <zkoppert@github.com>
---
 test_most_active_mentors.py | 41 ++++++++++++++++++++++++++++++++++++-
 1 file changed, 40 insertions(+), 1 deletion(-)

diff --git a/test_most_active_mentors.py b/test_most_active_mentors.py
index 9fc5db5..44664bd 100755
--- a/test_most_active_mentors.py
+++ b/test_most_active_mentors.py
@@ -22,7 +22,7 @@
 class TestCountCommentsPerUser(unittest.TestCase):
     """Test the count_comments_per_user function."""
 
-    def test_count_comments_per_user(self):
+    def test_count_comments_per_user_limit(self):
         """Test that count_comments_per_user correctly counts user comments.
 
         This test mocks the GitHub connection and issue comments, and checks
@@ -54,6 +54,45 @@ def test_count_comments_per_user(self):
         # Check the results
         self.assertEqual(result, expected_result)
 
+    def test_count_comments_per_user_with_ignores(self):
+        """Test that count_comments_per_user correctly counts user comments with some users ignored."""
+        # Set up the mock GitHub issues
+        mock_issue1 = MagicMock()
+        mock_issue1.comments = 2
+        mock_issue1.issue.user.login = "issue_owner"
+        mock_issue1.created_at = "2023-01-01T00:00:00Z"
+
+        # Set up mock GitHub issue comments by several users
+        mock_issue1.issue.comments.return_value = []
+        for i in range(5):
+            mock_comment1 = MagicMock()
+            mock_comment1.user.login = "very_active_user"
+            mock_comment1.created_at = datetime.fromisoformat(
+                f"2023-01-02T{i:02d}:00:00Z"
+            )
+            # pylint: disable=maybe-no-member
+            mock_issue1.issue.comments.return_value.append(mock_comment1)
+        for i in range(5):
+            mock_comment1 = MagicMock()
+            mock_comment1.user.login = "very_active_user_ignored"
+            mock_comment1.created_at = datetime.fromisoformat(
+                f"2023-01-02T{i:02d}:00:00Z"
+            )
+            # pylint: disable=maybe-no-member
+            mock_issue1.issue.comments.return_value.append(mock_comment1)
+
+        # Call the function
+        result = count_comments_per_user(
+            mock_issue1, ignore_users=["very_active_user_ignored"]
+        )
+        # Only the comments by "very_active_user" should be counted,
+        # so the count should be 3 since that is the threshold for heavily involved
+        expected_result = {"very_active_user": 3}
+
+        # Check the results
+        self.assertEqual(result, expected_result)
+        self.assertNotIn("very_active_user_ignored", result)
+
     def test_get_mentor_count(self):
         """Test that get_mentor_count correctly counts comments per user."""
         mentor_activity = {"sue": 15, "bob": 10}