|
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,29 @@ 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 | + organization = factory.Faker("pystr", max_chars=12) |
| 80 | + activestate_project_name = factory.Faker("pystr", max_chars=12) |
| 81 | + actor = factory.Faker("pystr", max_chars=12) |
| 82 | + actor_id = factory.Faker("uuid4") |
| 83 | + ingredient = factory.Faker("pystr", max_chars=12) |
| 84 | + |
| 85 | + |
| 86 | +class PendingActiveStatePublisherFactory(WarehouseFactory): |
| 87 | + class Meta: |
| 88 | + model = PendingActiveStatePublisher |
| 89 | + |
| 90 | + id = factory.Faker("uuid4", cast_to=None) |
| 91 | + project_name = factory.Faker("pystr", max_chars=12) |
| 92 | + organization = factory.Faker("pystr", max_chars=12) |
| 93 | + activestate_project_name = factory.Faker("pystr", max_chars=12) |
| 94 | + actor = factory.Faker("pystr", max_chars=12) |
| 95 | + actor_id = factory.Faker("uuid4") |
| 96 | + added_by = factory.SubFactory(UserFactory) |
| 97 | + ingredient = factory.Faker("pystr", max_chars=12) |
0 commit comments