Limit widget to supported widget types in test

the-deep · Nov 7, 2023 · 55ded41 · 55ded41
1 parent 42da67d
commit 55ded41
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/apps/analysis_framework/factories.py b/apps/analysis_framework/factories.py
@@ -8,6 +8,7 @@
     Widget,
     Filter,
 )
+from .widgets.store import widget_store
 
 
 class AnalysisFrameworkFactory(DjangoModelFactory):
@@ -28,7 +29,7 @@ class Meta:
 class WidgetFactory(DjangoModelFactory):
     title = factory.Sequence(lambda n: f'Widget-{n}')
     key = factory.Sequence(lambda n: f'widget-key-{n}')
-    widget_id = fuzzy.FuzzyChoice(Widget.WidgetType.choices, getter=lambda c: c[0])
+    widget_id = fuzzy.FuzzyChoice(widget_store.keys())
     properties = {}
     version = 1