From 3ca5ffe9c750ec1acaad6904046ab7558a2e2a71 Mon Sep 17 00:00:00 2001
From: Kandukuri saikrishna
 <50510272+KANDUKURIsaikrishna@users.noreply.github.com>
Date: Tue, 29 Oct 2024 16:52:18 +0100
Subject: [PATCH] added test cases

---
 tests/tests_main.py | 50 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 tests/tests_main.py

diff --git a/tests/tests_main.py b/tests/tests_main.py
new file mode 100644
index 0000000..0449701
--- /dev/null
+++ b/tests/tests_main.py
@@ -0,0 +1,50 @@
+# tests/test_main.py
+
+import unittest
+from unittest.mock import patch, MagicMock
+from main import get_images
+from config import MIN_SCORE
+
+class TestImageRetrieval(unittest.TestCase):
+
+    @patch("main.reddit")
+    def test_get_images_above_min_score(self, mock_reddit):
+        # Set up mock subreddit posts with different scores
+        mock_subreddit = MagicMock()
+        mock_subreddit.hot.return_value = [
+            MagicMock(score=MIN_SCORE + 1, url="http://image1.jpg"),
+            MagicMock(score=MIN_SCORE + 10, url="http://image2.png"),
+            MagicMock(score=MIN_SCORE - 1, url="http://image3.jpg"),  # Should be excluded
+        ]
+        mock_reddit.subreddit.return_value = mock_subreddit
+
+        images = get_images("some_subreddit")
+
+        # Assert that only images above MIN_SCORE are included
+        self.assertEqual(len(images), 2)
+        self.assertNotIn("http://image3.jpg", images)
+
+    @patch("main.reddit")
+    def test_get_images_only_images(self, mock_reddit):
+        # Set up mock posts with different file types
+        mock_subreddit = MagicMock()
+        mock_subreddit.hot.return_value = [
+            MagicMock(score=MIN_SCORE + 5, url="http://image.jpg"),
+            MagicMock(score=MIN_SCORE + 5, url="http://not_image.gif"),
+        ]
+        mock_reddit.subreddit.return_value = mock_subreddit
+
+        images = get_images("some_subreddit")
+
+        # Assert only image URLs are returned
+        self.assertEqual(len(images), 1)
+        self.assertIn("http://image.jpg", images)
+
+    def test_invalid_min_score(self):
+        # Test if MIN_SCORE is set to None or an invalid type
+        with self.assertRaises(TypeError):
+            MIN_SCORE = "invalid_value"
+            get_images("some_subreddit")
+
+if __name__ == "__main__":
+    unittest.main()