def get_items_by_customer_code(): """Helper method that uses the /customer/list API of endpoint to get a dictionary of the items available to each customer keyed by code.""" plans = {plan.code: plan for plan in Plan.all()} customers = Customer.list() items = {} for customer in customers: # Grab the items for each customer from the plan of the subscription items[int(customer.code)] = {item.name: item.quantity_included \ for item in plans[customer.subscription.plan.code].items} return items
def test_plans_parsing(self): responses.add(responses.POST, Plan.build_url('/plans/get'), body=self.read_fixture('plans.xml'), content_type='application/xml') plans = Plan.all() assert len(plans) == 2 plan = plans[0] assert plan.id == '6b0d13f4-6bef-102e-b098-40402145ee8b' assert plan.code == 'FREE_MONTHLY' assert plan.recurring_charge_amount == 0.0 assert plan.name == 'Free Monthly' assert plan.billing_frequency == 'monthly' assert plan.billing_frequency_quantity == 1 assert plan.setup_charge_code == None assert plan.is_active == 1 assert plan.is_free == 1 assert plan.billing_frequency_unit == 'months' assert plan.initial_bill_count_unit == 'months' assert plan.setup_charge_amount == 0.0 assert plan.billing_frequency_per == 'month' assert plan.trial_days == 0 assert plan.created_datetime == \ arrow.get('2011-01-07T20:46:43+00:00').datetime assert plan.recurring_charge_code == 'FREE_MONTHLY_RECURRING' assert plan.initial_bill_count == 1 assert plan.description == 'A free monthly plan' plan = plans[1] assert plan.id == '11af9cfc-6bf2-102e-b098-40402145ee8b' assert plan.code == 'PAID_MONTHLY' assert plan.recurring_charge_amount == 20.0 assert plan.name == 'Paid Monthly' assert plan.billing_frequency == 'monthly' assert plan.billing_frequency_quantity == 1 assert plan.setup_charge_code == None assert plan.is_active == 1 assert plan.is_free == 0 assert plan.billing_frequency_unit == 'months' assert plan.initial_bill_count_unit == 'months' assert plan.setup_charge_amount == 0.0 assert plan.billing_frequency_per == 'month' assert plan.trial_days == 0 assert plan.created_datetime == \ arrow.get('2011-01-07T21:05:42+00:00').datetime assert plan.recurring_charge_code == 'PAID_MONTHLY_RECURRING' assert plan.initial_bill_count == 1 assert plan.description == None
def test_plan_with_items_parsing(self): responses.add(responses.POST, Plan.build_url('/plans/get'), body=self.read_fixture('plans_with_items.xml'), content_type='application/xml') plans = Plan.all() assert len(plans) == 3 plan = plans[0] assert plan.id == '6b0d13f4-6bef-102e-b098-40402145ee8b' assert plan.code == 'FREE_MONTHLY' assert plan.recurring_charge_amount == 0.0 assert plan.name == 'Free Monthly' assert plan.billing_frequency == 'monthly' assert plan.billing_frequency_quantity == 1 assert plan.setup_charge_code == None assert plan.is_active == 1 assert plan.is_free == 1 assert plan.billing_frequency_unit == 'months' assert plan.initial_bill_count_unit == 'months' assert plan.setup_charge_amount == 0.0 assert plan.billing_frequency_per == 'month' assert plan.trial_days == 0 assert plan.created_datetime == \ arrow.get('2011-01-07T20:46:43+00:00').datetime assert plan.recurring_charge_code == 'FREE_MONTHLY_RECURRING' assert plan.initial_bill_count == 1 assert plan.description == 'A free monthly plan' item = plan.items[0] assert item.id == 'd19b4970-6e5a-102e-b098-40402145ee8b' assert item.code == 'MONTHLY_ITEM' assert item.quantity_included == 0 assert item.is_periodic == 0 assert item.overage_amount == 0.00 assert item.created_datetime == \ arrow.get('2011-01-10T22:40:34+00:00').datetime item = plan.items[1] assert item.id == 'd19ef2f0-6e5a-102e-b098-40402145ee8b' assert item.code == 'ONCE_ITEM' assert item.quantity_included == 0 assert item.is_periodic == 0 assert item.overage_amount == 0.00 assert item.created_datetime == \ arrow.get('2011-01-10T22:40:34+00:00').datetime plan = plans[1] assert plan.id == 'd19974a6-6e5a-102e-b098-40402145ee8b' assert plan.code == 'TRACKED_MONTHLY' assert plan.recurring_charge_amount == 10.0 assert plan.name == 'Tracked Monthly' assert plan.billing_frequency == 'monthly' assert plan.billing_frequency_quantity == 1 assert plan.setup_charge_code == None assert plan.is_active == 1 assert plan.is_free == 0 assert plan.billing_frequency_unit == 'months' assert plan.initial_bill_count_unit == 'months' assert plan.setup_charge_amount == 0.0 assert plan.billing_frequency_per == 'month' assert plan.trial_days == 0 assert plan.created_datetime == \ arrow.get('2011-01-10T22:40:34+00:00').datetime assert plan.recurring_charge_code == 'TRACKED_MONTHLY_RECURRING' assert plan.initial_bill_count == 1 assert plan.description == None assert len(plan.items) == 2 item = plan.items[0] assert item.id == 'd19b4970-6e5a-102e-b098-40402145ee8b' assert item.code == 'MONTHLY_ITEM' assert item.quantity_included == 2 assert item.is_periodic == 1 assert item.overage_amount == 10.00 assert item.created_datetime == \ arrow.get('2011-01-10T22:40:34+00:00').datetime item = plan.items[1] assert item.id == 'd19ef2f0-6e5a-102e-b098-40402145ee8b' assert item.code == 'ONCE_ITEM' assert item.quantity_included == 0 assert item.is_periodic == 0 assert item.overage_amount == 10.00 assert item.created_datetime == \ arrow.get('2011-01-10T22:40:34+00:00').datetime plan = plans[2] assert plan.id == '11af9cfc-6bf2-102e-b098-40402145ee8b' assert plan.code == 'PAID_MONTHLY' assert plan.recurring_charge_amount == 20.0 assert plan.name == 'Paid Monthly' assert plan.billing_frequency == 'monthly' assert plan.billing_frequency_quantity == 1 assert plan.setup_charge_code == None assert plan.is_active == 1 assert plan.is_free == 0 assert plan.billing_frequency_unit == 'months' assert plan.initial_bill_count_unit == 'months' assert plan.setup_charge_amount == 0.0 assert plan.billing_frequency_per == 'month' assert plan.trial_days == 0 assert plan.created_datetime == \ arrow.get('2011-01-07T21:05:42+00:00').datetime assert plan.recurring_charge_code == 'PAID_MONTHLY_RECURRING' assert plan.initial_bill_count == 1 assert plan.description == None assert len(plan.items) == 2 item = plan.items[0] assert item.id == 'd19b4970-6e5a-102e-b098-40402145ee8b' assert item.code == 'MONTHLY_ITEM' assert item.quantity_included == 0 assert item.is_periodic == 0 assert item.overage_amount == 0.00 assert item.created_datetime == \ arrow.get('2011-01-10T22:40:34+00:00').datetime item = plan.items[1] assert item.id == 'd19ef2f0-6e5a-102e-b098-40402145ee8b' assert item.code == 'ONCE_ITEM' assert item.quantity_included == 0 assert item.is_periodic == 0 assert item.overage_amount == 0.00 assert item.created_datetime == \ arrow.get('2011-01-10T22:40:34+00:00').datetime