|
13 | 13 | import factory
|
14 | 14 |
|
15 | 15 | from warehouse.oidc.models import (
|
| 16 | + ActiveStatePublisher, |
16 | 17 | GitHubPublisher,
|
17 | 18 | GooglePublisher,
|
| 19 | + PendingActiveStatePublisher, |
18 | 20 | PendingGitHubPublisher,
|
19 | 21 | PendingGooglePublisher,
|
20 | 22 | )
|
@@ -67,3 +69,37 @@ class Meta:
|
67 | 69 | email = factory.Faker("safe_email")
|
68 | 70 | sub = factory.Faker("pystr", max_chars=12)
|
69 | 71 | added_by = factory.SubFactory(UserFactory)
|
| 72 | + |
| 73 | + |
| 74 | +class ActiveStatePublisherFactory(WarehouseFactory): |
| 75 | + class Meta: |
| 76 | + model = ActiveStatePublisher |
| 77 | + |
| 78 | + id = factory.Faker("uuid4", cast_to=None) |
| 79 | + sub = factory.Faker("pystr", max_chars=12) |
| 80 | + organization_id = factory.Faker("uuid4") |
| 81 | + organization_url_name = factory.Faker("pystr", max_chars=12) |
| 82 | + project_id = factory.Faker("uuid4") |
| 83 | + activestate_project_name = factory.Faker("pystr", max_chars=12) |
| 84 | + project_path = f"{organization_url_name}/{activestate_project_name}" |
| 85 | + user_id = factory.Faker("uuid4") |
| 86 | + branch_id = factory.Faker("uuid4") |
| 87 | + project_visibility = factory.Faker("random_element", elements=[True, False]) |
| 88 | + sub = factory.Faker("pystr", max_chars=12) |
| 89 | + |
| 90 | + |
| 91 | +class PendingActiveStatePublisherFactory(WarehouseFactory): |
| 92 | + class Meta: |
| 93 | + model = PendingActiveStatePublisher |
| 94 | + |
| 95 | + id = factory.Faker("uuid4", cast_to=None) |
| 96 | + sub = factory.Faker("pystr", max_chars=12) |
| 97 | + project_name = factory.Faker("pystr", max_chars=12) |
| 98 | + organization_id = factory.Faker("uuid4") |
| 99 | + organization_url_name = factory.Faker("pystr", max_chars=12) |
| 100 | + project_id = factory.Faker("uuid4") |
| 101 | + activestate_project_name = factory.Faker("pystr", max_chars=12) |
| 102 | + project_path = f"{organization_url_name}/{activestate_project_name}" |
| 103 | + user_id = factory.Faker("uuid4") |
| 104 | + project_visibility = factory.Faker("random_element", elements=[True, False]) |
| 105 | + added_by = factory.SubFactory(UserFactory) |
0 commit comments