Add ManagedOpportunity in fixture

dimagi · Oct 29, 2024 · b389826 · b389826
1 parent 0f7c83f
commit b389826
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/commcare_connect/conftest.py b/commcare_connect/conftest.py
@@ -10,6 +10,7 @@
     PaymentUnitFactory,
 )
 from commcare_connect.organization.models import Organization
+from commcare_connect.program.tests.factories import ManagedOpportunityFactory
 from commcare_connect.users.models import User
 from commcare_connect.users.tests.factories import (
     ConnectIdUserLinkFactory,
@@ -52,7 +53,10 @@ def opportunity(request):
     verification_flags = getattr(request, "param", {}).get("verification_flags", {})
     opp_options = {"is_test": False}
     opp_options.update(getattr(request, "param", {}).get("opp_options", {}))
-    factory = OpportunityFactory(**opp_options)
+    if opp_options.get("managed", False):
+        factory = ManagedOpportunityFactory(**opp_options)
+    else:
+        factory = OpportunityFactory(**opp_options)
     OpportunityVerificationFlagsFactory(opportunity=factory, **verification_flags)
     return factory