def factory_product_model(org_id: str,
product_code: str = 'PPR'):
"""Produce a templated product model."""
subscription = ProductSubscriptionModel(org_id=org_id, product_code=product_code)
subscription.save()
return subscription
def factory_product_model(org_id: str,
product_code: str = 'PPR',
product_role_codes: list = ['search']
):
"""Produce a templated product model."""
subscription = ProductSubscriptionModel(org_id=org_id, product_code=product_code)
subscription.save()
# Save product roles
for role_code in product_role_codes:
product_role_code = ProductRoleCodeModel.find_by_code_and_product_code(role_code, product_code)
product_role = ProductSubscriptionRoleModel(product_subscription_id=subscription.id,
product_role_id=product_role_code.id)
product_role.save()
return subscription