Add Google PendingPublisher tests

Author: di

tests/unit/accounts/test_views.py

@@ -57,7 +57,7 @@
 from warehouse.events.tags import EventTag
 from warehouse.metrics.interfaces import IMetricsService
 from warehouse.oidc.interfaces import TooManyOIDCRegistrations
-from warehouse.oidc.models import PendingGitHubPublisher
+from warehouse.oidc.models import PendingGitHubPublisher, PendingGooglePublisher
 from warehouse.organizations.models import (
     OrganizationInvitation,
     OrganizationRole,
@@ -3431,6 +3431,11 @@ def test_manage_publishing_admin_disabled(self, monkeypatch, pyramid_request):
                 AdminFlagValue.DISALLOW_GITHUB_OIDC,
                 "GitHub",
             ),
+            (
+                "add_pending_google_oidc_publisher",
+                AdminFlagValue.DISALLOW_GOOGLE_OIDC,
+                "Google",
+            ),
         ],
     )
     def test_add_pending_oidc_publisher_admin_disabled(
@@ -3502,6 +3507,11 @@ def test_add_pending_oidc_publisher_admin_disabled(
                 AdminFlagValue.DISALLOW_GITHUB_OIDC,
                 "GitHub",
             ),
+            (
+                "add_pending_google_oidc_publisher",
+                AdminFlagValue.DISALLOW_GOOGLE_OIDC,
+                "Google",
+            ),
         ],
     )
     def test_add_pending_oidc_publisher_user_cannot_register(
@@ -3596,6 +3606,18 @@ def test_add_pending_oidc_publisher_user_cannot_register(
                 ),
                 PendingGitHubPublisher,
             ),
+            (
+                "add_pending_google_oidc_publisher",
+                AdminFlagValue.DISALLOW_GOOGLE_OIDC,
+                "Google",
+                lambda i, user_id: PendingGooglePublisher(
+                    project_name="some-project-name-" + str(i),
+                    email="some-email-" + str(i) + "@example.com",
+                    sub="some-sub",
+                    added_by_id=user_id,
+                ),
+                PendingGooglePublisher,
+            ),
         ],
     )
     def test_add_pending_github_oidc_publisher_too_many_already(
@@ -3663,6 +3685,10 @@ def test_add_pending_github_oidc_publisher_too_many_already(
                 "add_pending_github_oidc_publisher",
                 "GitHub",
             ),
+            (
+                "add_pending_google_oidc_publisher",
+                "Google",
+            ),
         ],
     )
     def test_add_pending_oidc_publisher_ratelimited(
@@ -3725,6 +3751,10 @@ def test_add_pending_oidc_publisher_ratelimited(
                 "add_pending_github_oidc_publisher",
                 "GitHub",
             ),
+            (
+                "add_pending_google_oidc_publisher",
+                "Google",
+            ),
         ],
     )
     def test_add_pending_oidc_publisher_invalid_form(
@@ -3814,6 +3844,23 @@ def test_add_pending_oidc_publisher_invalid_form(
                     }
                 ),
             ),
+            (
+                "add_pending_google_oidc_publisher",
+                "Google",
+                lambda user_id: PendingGooglePublisher(
+                    project_name="some-project-name",
+                    email="[email protected]",
+                    sub="some-sub",
+                    added_by_id=user_id,
+                ),
+                MultiDict(
+                    {
+                        "email": "[email protected]",
+                        "sub": "some-sub",
+                        "project_name": "some-project-name",
+                    }
+                ),
+            ),
         ],
     )
     def test_add_pending_oidc_publisher_already_exists(
@@ -3900,6 +3947,18 @@ def test_add_pending_oidc_publisher_already_exists(
                 ),
                 PendingGitHubPublisher,
             ),
+            (
+                "add_pending_google_oidc_publisher",
+                "Google",
+                MultiDict(
+                    {
+                        "email": "[email protected]",
+                        "sub": "some-sub",
+                        "project_name": "some-project-name",
+                    }
+                ),
+                PendingGooglePublisher,
+            ),
         ],
     )
     def test_add_pending_oidc_publisher(
@@ -4091,6 +4150,15 @@ def test_delete_pending_oidc_publisher_invalid_form(
                 ),
                 PendingGitHubPublisher,
             ),
+            (
+                lambda user_id: PendingGooglePublisher(
+                    project_name="some-project-name",
+                    email="[email protected]",
+                    sub="some-sub",
+                    added_by_id=user_id,
+                ),
+                PendingGooglePublisher,
+            ),
         ],
     )
     def test_delete_pending_oidc_publisher_not_found(
@@ -4142,6 +4210,15 @@ def test_delete_pending_oidc_publisher_not_found(
                 ),
                 PendingGitHubPublisher,
             ),
+            (
+                lambda user_id: PendingGooglePublisher(
+                    project_name="some-project-name",
+                    email="[email protected]",
+                    sub="some-sub",
+                    added_by_id=user_id,
+                ),
+                PendingGooglePublisher,
+            ),
         ],
     )
     def test_delete_pending_oidc_publisher_no_access(
@@ -4197,6 +4274,16 @@ def test_delete_pending_oidc_publisher_no_access(
                 ),
                 PendingGitHubPublisher,
             ),
+            (
+                "Google",
+                lambda user_id: PendingGooglePublisher(
+                    project_name="some-project-name",
+                    email="[email protected]",
+                    sub="some-sub",
+                    added_by_id=user_id,
+                ),
+                PendingGooglePublisher,
+            ),
         ],
     )
     def test_delete_pending_oidc_publisher(