Add OIDC ActiveStatePublisher modeling

Author: th3coop

tests/common/db/oidc.py

@@ -14,9 +14,11 @@
 import pretend
 
 from warehouse.oidc.models import (
+    ActiveStatePublisher,
     GitHubPublisher,
     GooglePublisher,
     OIDCPublisher,
+    PendingActiveStatePublisher,
     PendingGitHubPublisher,
     PendingGooglePublisher,
     PendingOIDCPublisher,
@@ -86,3 +88,37 @@ class Meta:
     email = factory.Faker("safe_email")
     sub = factory.Faker("pystr", max_chars=12)
     added_by = factory.SubFactory(UserFactory)
+
+
+class ActiveStatePublisherFactory(WarehouseFactory):
+    class Meta:
+        model = ActiveStatePublisher
+
+    id = factory.Faker("uuid4", cast_to=None)
+    sub = factory.Faker("pystr", max_chars=12)
+    organization_id = factory.Faker("uuid4")
+    organization_url_name = factory.Faker("pystr", max_chars=12)
+    project_id = factory.Faker("uuid4")
+    activestate_project_name = factory.Faker("pystr", max_chars=12)
+    project_path = f"{organization_url_name}/{activestate_project_name}"
+    user_id = factory.Faker("uuid4")
+    branch_id = factory.Faker("uuid4")
+    project_visibility = factory.Faker("random_element", elements=[True, False])
+    sub = factory.Faker("pystr", max_chars=12)
+
+
+class PendingActiveStatePublisherFactory(WarehouseFactory):
+    class Meta:
+        model = PendingActiveStatePublisher
+
+    id = factory.Faker("uuid4", cast_to=None)
+    sub = factory.Faker("pystr", max_chars=12)
+    project_name = factory.Faker("pystr", max_chars=12)
+    organization_id = factory.Faker("uuid4")
+    organization_url_name = factory.Faker("pystr", max_chars=12)
+    project_id = factory.Faker("uuid4")
+    activestate_project_name = factory.Faker("pystr", max_chars=12)
+    project_path = f"{organization_url_name}/{activestate_project_name}"
+    user_id = factory.Faker("uuid4")
+    project_visibility = factory.Faker("random_element", elements=[True, False])
+    added_by = factory.SubFactory(UserFactory)

tests/conftest.py

@@ -351,6 +351,18 @@ def oidc_service(db_session):
     )
 
 
+# @pytest.fixture
+# def activestate_oidc_service(db_session):
+#     return oidc_services.NullOIDCPublisherService(
+#         db_session,
+#         pretend.stub(),
+#         ACTIVESTATE_OIDC_ISSUER_URL,
+#         pretend.stub(),
+#         pretend.stub(),
+#         pretend.stub(),
+#     )
+
+
 @pytest.fixture
 def macaroon_service(db_session):
     return macaroon_services.DatabaseMacaroonService(db_session)