Tests for Google and Github login

website/tests.py

@@ -1,5 +1,5 @@
 import os
-
+import unittest
 import chromedriver_autoinstaller
 from django.core.files.storage import default_storage
 from django.core.files.uploadedfile import SimpleUploadedFile
@@ -13,6 +13,11 @@
 
 from .models import Issue, IssueScreenshot
 
+from unittest.mock import patch
+from django.urls import reverse
+from django.test import RequestFactory, TestCase
+from .views import GoogleLogin, GithubLogin
+
 os.environ["DJANGO_LIVE_TEST_SERVER_ADDRESS"] = "localhost:8082"
 
 
@@ -145,3 +150,53 @@ def test_on_hide(self):
 
             if "hidden" not in filename:
                 self.assertFalse(True, "files rename failed")
+
+class TestGoogleLogin(TestCase):
+    def setUp(self):
+        self.factory = RequestFactory()
+
+    def test_callback_url(self):
+        view = GoogleLogin()
+        request = self.factory.get("/auth/google/")
+        request.session = {}
+        view.request = request
+
+        def mock_reverse(url_name):
+            if url_name == "google_callback":
+                return "http://example.com/accounts/google/login/callback/"
+            else:
+                raise ValueError("Unexpected URL name")
+
+        def mock_build_absolute_uri(uri):
+            return "http://example.com" + uri
+
+        with patch("django.urls.reverse", side_effect=mock_reverse):
+            with patch("django.http.HttpRequest.build_absolute_uri", side_effect=mock_build_absolute_uri):
+                callback_url = view.callback_url
+
+        self.assertEqual(callback_url, "http://example.com/accounts/google/login/callback/")
+
+class TestGithubLogin(TestCase):
+    def setUp(self):
+        self.factory = RequestFactory()
+
+    def test_callback_url(self):
+        view = GithubLogin()
+        request = self.factory.get("auth/github/")
+        request.session = {}
+        view.request = request
+
+        def mock_reverse(url_name):
+            if url_name == "github_callback":
+                return "http://example.com/accounts/github/login/callback/"
+            else:
+                raise ValueError("Unexpected URL name")
+
+        def mock_build_absolute_uri(uri):
+            return "http://example.com" + uri
+
+        with patch("django.urls.reverse", side_effect=mock_reverse):
+            with patch("django.http.HttpRequest.build_absolute_uri", side_effect=mock_build_absolute_uri):
+                callback_url = view.callback_url
+
+        self.assertEqual(callback_url, "http://example.com/accounts/github/login/callback/")