def setup_subscription(cls, domain_name, software_plan):
generator.bootstrap_test_plans()
plan = DefaultProductPlan.get_default_plan_version(
edition=software_plan)
cls.account = BillingAccount.get_or_create_account_by_domain(
domain_name, created_by="automated-test" + cls.__name__)[0]
cls.subscription = Subscription.new_domain_subscription(
cls.account, domain_name, plan)
cls.subscription.is_active = True
cls.subscription.save()
def setUpClass(cls):
super(BaseInvoiceTestCase, cls).setUpClass()
generator.bootstrap_test_plans() # TODO - only call for subclasses that actually need the test plans
cls.billing_contact = generator.create_arbitrary_web_user_name()
cls.dimagi_user = generator.create_arbitrary_web_user_name(is_dimagi=True)
cls.currency = generator.init_default_currency()
cls.account = generator.billing_account(
cls.dimagi_user, cls.billing_contact)
cls.domain = generator.arbitrary_domain()
cls.subscription_length = 15 # months
subscription_start_date = datetime.date(2016, 2, 23)
subscription_end_date = add_months_to_date(subscription_start_date, cls.subscription_length)
cls.subscription = generator.generate_domain_subscription(
cls.account,
cls.domain,
date_start=subscription_start_date,
date_end=subscription_end_date,
)
def setUpClass(cls):
super(TestDomainInvoiceFactory, cls).setUpClass()
generator.bootstrap_test_plans()