add unit tests to Canonical CLA

tests/cla/test_routes.py

@@ -0,0 +1,93 @@
+import unittest
+from unittest.mock import patch, MagicMock
+from webapp.app import app
+import flask
+
+
+class TestCLARoutes(unittest.TestCase):
+    def setUp(self):
+        self.client = app.test_client()
+        self.client.testing = True
+
+    @patch("requests.request")
+    def test_canonical_cla_api_proxy(self, mock_request):
+        mock_response = MagicMock()
+        mock_response.content = b"Test content"
+        mock_response.headers = {"Content-Type": "application/json"}
+        mock_response.status_code = 200
+        mock_request.return_value = mock_response
+        # https://example.com/api -> aHR0cHM6Ly9leGFtcGxlLmNvbS9hcGk=
+        response = self.client.get(
+            "/legal/contributors/agreement/api?request_url=aHR0cHM6Ly9leGFtcGxlLmNvbS9hcGk="
+        )
+
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(response.data, b"Test content")
+        self.assertEqual(response.headers["Content-Type"], "application/json")
+        self.assertEqual(response.headers["Cache-Control"], "no-store")
+
+    def test_canonical_cla_api_github_logout(self):
+        response = self.client.get(
+            "/legal/contributors/agreement/api/github/logout"
+        )
+
+        self.assertEqual(response.status_code, 302)
+        self.assertEqual(response.location, "/legal/contributors/agreement")
+        self.assertIn("github_oauth2_session", response.headers["Set-Cookie"])
+        self.assertEqual(response.headers["Cache-Control"], "no-store")
+
+    @patch("webapp.canonical_cla.views.get_query_param")
+    def test_canonical_cla_api_github_login(self, mock_get_query_param):
+        mock_get_query_param.side_effect = [
+            "https://example.com/agreement",  # agreement_url
+            "test_access_token",  # access_token
+            None,  # github_error
+        ]
+
+        response = self.client.get(
+            "/legal/contributors/agreement/api/github/login"
+        )
+
+        self.assertEqual(response.status_code, 302)
+        self.assertEqual(
+            response.location, "https://example.com/agreement?github_error="
+        )
+        self.assertIn("github_oauth2_session", response.headers["Set-Cookie"])
+        self.assertEqual(response.headers["Cache-Control"], "no-store")
+
+    def test_canonical_cla_api_launchpad_logout(self):
+        response = self.client.get(
+            "/legal/contributors/agreement/api/launchpad/logout"
+        )
+
+        self.assertEqual(response.status_code, 302)
+        self.assertEqual(response.location, "/legal/contributors/agreement")
+        self.assertIn(
+            "launchpad_oauth_session", response.headers["Set-Cookie"]
+        )
+        self.assertEqual(response.headers["Cache-Control"], "no-store")
+
+    @patch("webapp.canonical_cla.views.get_query_param")
+    def test_canonical_cla_api_launchpad_login(self, mock_get_query_param):
+        mock_get_query_param.side_effect = [
+            "https://example.com/agreement",  # agreement_url
+            "test_access_token",  # access_token
+            None,  # launchpad_error
+        ]
+
+        response = self.client.get(
+            "/legal/contributors/agreement/api/launchpad/login"
+        )
+
+        self.assertEqual(response.status_code, 302)
+        self.assertEqual(
+            response.location, "https://example.com/agreement?launchpad_error="
+        )
+        self.assertIn(
+            "launchpad_oauth_session", response.headers["Set-Cookie"]
+        )
+        self.assertEqual(response.headers["Cache-Control"], "no-store")
+
+
+if __name__ == "__main__":
+    unittest.main()