Add Google Publisher tests

Author: di

tests/unit/manage/test_views.py

@@ -47,7 +47,7 @@
 from warehouse.manage.views import organizations as org_views
 from warehouse.metrics.interfaces import IMetricsService
 from warehouse.oidc.interfaces import TooManyOIDCRegistrations
-from warehouse.oidc.models import GitHubPublisher, OIDCPublisher
+from warehouse.oidc.models import GitHubPublisher, GooglePublisher, OIDCPublisher
 from warehouse.organizations.interfaces import IOrganizationService
 from warehouse.organizations.models import (
     OrganizationRoleType,
@@ -5905,6 +5905,21 @@ def test_manage_project_oidc_publishers_admin_disabled(
                     normalized_environment=publisher.environment,
                 ),
             ),
+            (
+                "add_google_oidc_publisher",
+                pretend.stub(
+                    id="fakeid",
+                    publisher_name="Google",
+                    publisher_url=lambda x=None: None,
+                    email="[email protected]",
+                    sub="some-sub",
+                ),
+                lambda publisher: pretend.stub(
+                    validate=pretend.call_recorder(lambda: True),
+                    email=pretend.stub(data=publisher.email),
+                    sub=pretend.stub(data=publisher.sub),
+                ),
+            ),
         ],
     )
     def test_add_oidc_publisher_preexisting(
@@ -6014,6 +6029,15 @@ def test_add_oidc_publisher_preexisting(
                 ),
                 pretend.stub(publisher_name="GitHub"),
             ),
+            (
+                "add_google_oidc_publisher",
+                pretend.stub(
+                    validate=pretend.call_recorder(lambda: True),
+                    email=pretend.stub(data="[email protected]"),
+                    sub=pretend.stub(data="some-sub"),
+                ),
+                "Google",
+            ),
         ],
     )
     def test_add_oidc_publisher_created(
@@ -6143,6 +6167,20 @@ def test_add_oidc_publisher_created(
                     }
                 ),
             ),
+            (
+                "add_google_oidc_publisher",
+                "Google",
+                GooglePublisher(
+                    email="[email protected]",
+                    sub="some-sub",
+                ),
+                MultiDict(
+                    {
+                        "email": "[email protected]",
+                        "sub": "some-sub",
+                    }
+                ),
+            ),
         ],
     )
     def test_add_oidc_publisher_already_registered_with_project(
@@ -6209,6 +6247,7 @@ def test_add_oidc_publisher_already_registered_with_project(
         "view_name, publisher_name",
         [
             ("add_github_oidc_publisher", "GitHub"),
+            ("add_google_oidc_publisher", "Google"),
         ],
     )
     def test_add_oidc_publisher_ratelimited(
@@ -6256,6 +6295,7 @@ def test_add_oidc_publisher_ratelimited(
         "view_name, publisher_name",
         [
             ("add_github_oidc_publisher", "GitHub"),
+            ("add_google_oidc_publisher", "Google"),
         ],
     )
     def test_add_oidc_publisher_admin_disabled(
@@ -6296,6 +6336,7 @@ def test_add_oidc_publisher_admin_disabled(
         "view_name, publisher_name",
         [
             ("add_github_oidc_publisher", "GitHub"),
+            ("add_google_oidc_publisher", "Google"),
         ],
     )
     def test_add_oidc_publisher_invalid_form(
@@ -6357,6 +6398,10 @@ def test_add_oidc_publisher_invalid_form(
                 workflow_filename="some-workflow-filename.yml",
                 environment="some-environment",
             ),
+            GooglePublisher(
+                email="[email protected]",
+                sub="some-sub",
+            ),
         ],
     )
     def test_delete_oidc_publisher_registered_to_multiple_projects(
@@ -6455,6 +6500,10 @@ def test_delete_oidc_publisher_registered_to_multiple_projects(
                 workflow_filename="some-workflow-filename.yml",
                 environment="some-environment",
             ),
+            GooglePublisher(
+                email="[email protected]",
+                sub="some-sub",
+            ),
         ],
     )
     def test_delete_oidc_publisher_entirely(self, monkeypatch, db_request, publisher):