♻️ Add states to release factory

kids-first · Apr 20, 2020 · 4d6eeef · 4d6eeef
1 parent 744a2a1
commit 4d6eeef
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/coordinator/api/factories/release.py b/coordinator/api/factories/release.py
@@ -1,4 +1,5 @@
 import factory
+from factory.fuzzy import FuzzyChoice
 
 from coordinator.api.models.release import Release
 from coordinator.api.factories.study import StudyFactory
@@ -14,6 +15,9 @@ class Meta:
     tasks = factory.RelatedFactory(
         "coordinator.api.factories.task.TaskFactory", "release"
     )
+    state = FuzzyChoice(
+        ["waiting", "failed", "canceled", "staged", "rejected", "published"]
+    )
 
     @factory.post_generation
     def studies(self, create, extracted, **kwargs):