Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(python-sdk): MAX-28 - test preview allocations - happy path #16

Merged
merged 1 commit into from
Jan 30, 2024
Merged

feat(python-sdk): MAX-28 - test preview allocations - happy path #16

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
228 changes: 228 additions & 0 deletions tests/test_subscription_components.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,228 @@
from advancedbilling.models.allocate_components import AllocateComponents
from advancedbilling.models.allocation import Allocation
from advancedbilling.models.allocation_preview import AllocationPreview
from advancedbilling.models.allocation_preview_item import AllocationPreviewItem
from advancedbilling.models.component import Component
from advancedbilling.models.create_customer_request import CreateCustomerRequest
from advancedbilling.models.create_on_off_component import CreateOnOffComponent
from advancedbilling.models.create_or_update_product_request import CreateOrUpdateProductRequest
from advancedbilling.models.create_payment_profile_request import CreatePaymentProfileRequest
from advancedbilling.models.create_product_family_request import CreateProductFamilyRequest
from advancedbilling.models.create_quantity_based_component import CreateQuantityBasedComponent
from advancedbilling.models.create_subscription_request import CreateSubscriptionRequest
from advancedbilling.models.credit_card_payment_profile import CreditCardPaymentProfile
from advancedbilling.models.customer import Customer
from advancedbilling.models.interval_unit import IntervalUnit
from advancedbilling.models.payment_type import PaymentType
from advancedbilling.models.preview_allocations_request import PreviewAllocationsRequest
from advancedbilling.models.pricing_scheme import PricingScheme
from advancedbilling.models.product import Product
from advancedbilling.models.product_family import ProductFamily
from advancedbilling.models.subscription import Subscription
from tests.base import TestBase


class TestSubscriptionComponents(TestBase):
def test_preview_allocations_subscription_and_on_off_component_and_quantity_based_component_then_preview_allocations_and_allocate_components(
self):
product_family: ProductFamily = self.get_product_families_controller().create_product_family(
CreateProductFamilyRequest.from_dictionary(
{
"product_family": {
"name": "TestSubscriptionComponents_Product_Family_Name_1",
"description": "TestSubscriptionComponents_Product_Family_Description_1",
}
}
)
).product_family
product: Product = self.get_products_controller().create_product(
product_family.id,
CreateOrUpdateProductRequest.from_dictionary(
{
"product":
{
"name": "TestSubscriptionComponents_Product_Name_1",
"handle": "testsubscriptioncomponents_product_handle_1",
"description": "TestSubscriptionComponents_Product_Description_1",
"interval": 1,
"price_in_cents": 10,
"interval_unit": IntervalUnit.DAY
}
}
)).product
customer: Customer = self.get_customers_controller().create_customer(CreateCustomerRequest.from_dictionary(
{
"customer": {
"first_name": "TestSubscriptionComponents_Customer_FirstName_1",
"last_name": "TestSubscriptionComponents_Customer_LastName_1",
"email": "[email protected]",
"cc_emails": ["[email protected]"],
"organization": "TestSubscriptionComponents_CustomerOrganization_1",
"reference": "TestSubscriptionComponents_CustomerReference_1",
"address": "test address",
"address_2": "test address two",
"city": "Ohio",
"state": "TX",
"zip": "test zip",
"country": "US",
"phone": "+00 123 456 789",
"tax_exempt": False,
"vat_number": "test vat number",
"parent_id": None,
"locale": None
}
}
)).customer
payment_profile: CreditCardPaymentProfile = self.get_payment_profiles_controller().create_payment_profile(
CreatePaymentProfileRequest.from_dictionary(
{
"payment_profile": {
"customer_id": customer.id,
"payment_type": PaymentType.CREDIT_CARD,
"expiration_month": 12,
"expiration_year": 2027,
"full_number": "4111111111111111"
}
}
)
).payment_profile
on_off_component: Component = self.get_components_controller().create_on_off_component(
product_family.id,
CreateOnOffComponent.from_dictionary(
{
"on_off_component": {
"name": "TestSubscriptionComponents_On_Off_Component_Name_1",
'unit_price': "1.0",
"prices": [
{
"unit_price": "0.10",
"starting_quantity": "0"
}
],
}
}
)).component
quantity_based_component: Component = self.get_components_controller().create_quantity_based_component(
product_family.id,
CreateQuantityBasedComponent.from_dictionary(
{
"quantity_based_component": {
"name": "TestSubscriptionComponents_QuantityBased_Component_Name_1",
"unit_name": "Component",
"unit_price": "1.0",
"allow_fractional_quantities": True,
"pricing_scheme": PricingScheme.PER_UNIT
}
}
)
).component
quantity_based_component_two: Component = self.get_components_controller().create_quantity_based_component(
product_family.id,
CreateQuantityBasedComponent.from_dictionary(
{
"quantity_based_component": {
"name": "TestSubscriptionComponents_QuantityBased_Component_Name_2",
"unit_name": "Component",
"unit_price": "1.0",
"allow_fractional_quantities": True,
"pricing_scheme": PricingScheme.PER_UNIT
}
}
)
).component
subscription: Subscription = self.get_subscriptions_controller().create_subscription(
CreateSubscriptionRequest.from_dictionary(
{
"subscription": {
"product_id": product.id,
'customer_id': customer.id,
"dunning_communication_delay_enabled": False,
"payment_collection_method": "automatic",
"skip_billing_manifest_taxes": False,
"payment_profile_id": payment_profile.id,
}
}
)).subscription

# THEN
allocation: AllocationPreview = self.get_subscription_components_controller().preview_allocations(
subscription.id,
PreviewAllocationsRequest.from_dictionary(
{
"allocations": [
{
"component_id": on_off_component.id,
"memo": "TestSubscriptionComponents_On_Off_Component_memo_1",
"quantity": "1"
},
{
"component_id": quantity_based_component.id,
"memo": "TestSubscriptionComponents_QuantityBased_Component_memo_1",
"quantity": "10.5"
},
{
"component_id": quantity_based_component_two.id,
"memo": "TestSubscriptionComponents_QuantityBased_Component_memo_2",
"quantity": 1
}
]
}
)).allocation_preview

assert "upgrade" == allocation.direction
assert 3 == len(allocation.allocations)

allocations: AllocationPreview = allocation.allocations
on_off_component_allocation: AllocationPreviewItem = allocations[0]
quantity_based_component_allocation: AllocationPreviewItem = allocations[1]
quantity_based_component_two_allocation: AllocationPreviewItem = allocations[2]

assert on_off_component.id == on_off_component_allocation.component_id
assert 1 == on_off_component_allocation.quantity

assert quantity_based_component.id == quantity_based_component_allocation.component_id
assert "10.5" == quantity_based_component_allocation.quantity

assert quantity_based_component_two.id == quantity_based_component_two_allocation.component_id
assert '1.0' == quantity_based_component_two_allocation.quantity

# AND

allocated_components = self.get_subscription_components_controller().allocate_components(
subscription.id,
AllocateComponents.from_dictionary({
"proration_upgrade_scheme": "prorate-attempt-capture",
"proration_downgrade_scheme": "no-prorate",
"allocations": [
{
"component_id": on_off_component.id,
"memo": "TestSubscriptionComponents_On_Off_Component_memo_1",
"quantity": "1"
},
{
"component_id": quantity_based_component.id,
"memo": "TestSubscriptionComponents_QuantityBased_Component_memo_1",
"quantity": "10.5"
},
{
"component_id": quantity_based_component_two.id,
"memo": "TestSubscriptionComponents_QuantityBased_Component_memo_2",
"quantity": 1
}
]
}))

assert 3 == len(allocated_components)

allocated_on_off_component: Allocation = allocated_components[0].allocation
allocated_quantity_based_component: Allocation = allocated_components[1].allocation
allocated_quantity_based_component_two: Allocation = allocated_components[2].allocation

assert on_off_component.id == allocated_on_off_component.component_id
assert on_off_component_allocation.quantity == allocated_on_off_component.quantity

assert quantity_based_component.id == allocated_quantity_based_component.component_id
assert quantity_based_component_allocation.quantity == allocated_quantity_based_component.quantity

assert quantity_based_component_two.id == allocated_quantity_based_component_two.component_id
assert allocated_quantity_based_component_two.quantity == allocated_quantity_based_component_two.quantity
Loading