From 46f9f4d88e4ba522edd590bbf42e6d3c74e55a1e Mon Sep 17 00:00:00 2001 From: Marc Date: Thu, 25 Sep 2014 18:21:17 +0000 Subject: [PATCH] Refactored orders.tests.functional_tests --- orchestra/apps/orders/models.py | 1 - .../orders/tests/functional_tests/__init__.py | 0 .../orders/tests/functional_tests/tests.py | 595 ------------------ orchestra/apps/services/handlers.py | 2 + .../tests/functional_tests/__init__.py | 16 + .../tests/functional_tests/test_domain.py | 138 ++++ .../tests/functional_tests/test_ftp.py | 103 +++ .../tests/functional_tests/test_job.py | 52 ++ .../tests/functional_tests/test_mailbox.py | 133 ++++ .../tests/functional_tests/test_plan.py | 37 ++ .../tests/functional_tests/test_traffic.py | 157 +++++ 11 files changed, 638 insertions(+), 596 deletions(-) delete mode 100644 orchestra/apps/orders/tests/functional_tests/__init__.py delete mode 100644 orchestra/apps/orders/tests/functional_tests/tests.py create mode 100644 orchestra/apps/services/tests/functional_tests/__init__.py create mode 100644 orchestra/apps/services/tests/functional_tests/test_domain.py create mode 100644 orchestra/apps/services/tests/functional_tests/test_ftp.py create mode 100644 orchestra/apps/services/tests/functional_tests/test_job.py create mode 100644 orchestra/apps/services/tests/functional_tests/test_mailbox.py create mode 100644 orchestra/apps/services/tests/functional_tests/test_plan.py create mode 100644 orchestra/apps/services/tests/functional_tests/test_traffic.py diff --git a/orchestra/apps/orders/models.py b/orchestra/apps/orders/models.py index 6a7b2eda..be69d778 100644 --- a/orchestra/apps/orders/models.py +++ b/orchestra/apps/orders/models.py @@ -32,7 +32,6 @@ class OrderQuerySet(models.QuerySet): group_by = queryset.group_by def bill(self, **options): - # TODO classmethod? bills = [] bill_backend = Order.get_bill_backend() qs = self.select_related('account', 'service') diff --git a/orchestra/apps/orders/tests/functional_tests/__init__.py b/orchestra/apps/orders/tests/functional_tests/__init__.py deleted file mode 100644 index e69de29b..00000000 diff --git a/orchestra/apps/orders/tests/functional_tests/tests.py b/orchestra/apps/orders/tests/functional_tests/tests.py deleted file mode 100644 index 2dde24ea..00000000 --- a/orchestra/apps/orders/tests/functional_tests/tests.py +++ /dev/null @@ -1,595 +0,0 @@ -import datetime -import decimal -import sys - -from dateutil.relativedelta import relativedelta -from django.contrib.contenttypes.models import ContentType -from django.db.models import F -from django.utils import timezone -from freezegun import freeze_time - -from orchestra.apps.accounts.models import Account -from orchestra.apps.mails.models import Mailbox -from orchestra.apps.miscellaneous.models import MiscService, Miscellaneous -from orchestra.apps.resources.models import Resource, ResourceData, MonitorData -from orchestra.apps.services.models import Service, Plan -from orchestra.apps.services import settings as services_settings -from orchestra.apps.users.models import User -from orchestra.utils.tests import BaseTestCase, random_ascii - - -class BaseBillingTest(BaseTestCase): - def create_account(self): - account = Account.objects.create() - user = User.objects.create_user(username='account_%s' % random_ascii(5), account=account) - account.user = user - account.save() - return account - - -class FTPBillingTest(BaseBillingTest): - def create_ftp_service(self): - return Service.objects.create( - description="FTP Account", - content_type=ContentType.objects.get_for_model(User), - match='not user.is_main and user.has_posix()', - billing_period=Service.ANUAL, - billing_point=Service.FIXED_DATE, - is_fee=False, - metric='', - pricing_period=Service.NEVER, - rate_algorithm=Service.STEP_PRICE, - on_cancel=Service.COMPENSATE, - payment_style=Service.PREPAY, - tax=0, - nominal_price=10, - ) - - def create_ftp(self, account=None): - if not account: - account = self.create_account() - username = '%s_ftp' % random_ascii(10) - user = User.objects.create_user(username=username, account=account) - POSIX = user._meta.get_field_by_name('posix')[0].model - POSIX.objects.create(user=user) - return user - - def test_ftp_account_1_year_fiexed(self): - service = self.create_ftp_service() - user = self.create_ftp() - bp = timezone.now().date() + relativedelta(years=1) - bills = service.orders.bill(billing_point=bp, fixed_point=True) - self.assertEqual(10, bills[0].get_total()) - - def test_ftp_account_2_year_fiexed(self): - service = self.create_ftp_service() - user = self.create_ftp() - bp = timezone.now().date() + relativedelta(years=2) - bills = service.orders.bill(billing_point=bp, fixed_point=True) - self.assertEqual(20, bills[0].get_total()) - - def test_ftp_account_6_month_fixed(self): - service = self.create_ftp_service() - self.create_ftp() - bp = timezone.now().date() + relativedelta(months=6) - bills = service.orders.bill(billing_point=bp, fixed_point=True) - self.assertEqual(5, bills[0].get_total()) - - def test_ftp_account_next_billing_point(self): - service = self.create_ftp_service() - self.create_ftp() - now = timezone.now() - bp_month = services_settings.SERVICES_SERVICE_ANUAL_BILLING_MONTH - if now.month > bp_month: - bp = datetime.datetime(year=now.year+1, month=bp_month, - day=1, tzinfo=timezone.get_current_timezone()) - else: - bp = datetime.datetime(year=now.year, month=bp_month, - day=1, tzinfo=timezone.get_current_timezone()) - bills = service.orders.bill(billing_point=now, fixed_point=False) - size = decimal.Decimal((bp - now).days)/365 - error = decimal.Decimal(0.05) - self.assertGreater(10*size+error*(10*size), bills[0].get_total()) - self.assertLess(10*size-error*(10*size), bills[0].get_total()) - - def test_ftp_account_with_compensation(self): - account = self.create_account() - service = self.create_ftp_service() - user = self.create_ftp(account=account) - first_bp = timezone.now().date() + relativedelta(years=2) - bills = service.orders.bill(billing_point=first_bp, fixed_point=True) - self.assertEqual(1, service.orders.active().count()) - user.delete() - self.assertEqual(0, service.orders.active().count()) - user = self.create_ftp(account=account) - self.assertEqual(1, service.orders.active().count()) - self.assertEqual(2, service.orders.count()) - bp = timezone.now().date() + relativedelta(years=1) - bills = service.orders.bill(billing_point=bp, fixed_point=True, new_open=True) - discount = bills[0].lines.order_by('id')[0].sublines.get() - self.assertEqual(decimal.Decimal(-20), discount.total) - order = service.orders.order_by('id').first() - self.assertEqual(order.cancelled_on, order.billed_until) - order = service.orders.order_by('-id').first() - self.assertEqual(first_bp, order.billed_until) - self.assertEqual(decimal.Decimal(0), bills[0].get_total()) - - def test_ftp_account_with_rates(self): - pass - - -class DomainBillingTest(BaseBillingTest): - def create_domain_service(self): - service = Service.objects.create( - description="Domain .ES", - content_type=ContentType.objects.get_for_model(Miscellaneous), - match="miscellaneous.is_active and miscellaneous.service.name.lower() == 'domain .es'", - billing_period=Service.ANUAL, - billing_point=Service.ON_REGISTER, - is_fee=False, - metric='', - pricing_period=Service.BILLING_PERIOD, - rate_algorithm=Service.STEP_PRICE, - on_cancel=Service.NOTHING, - payment_style=Service.PREPAY, - tax=0, - nominal_price=10 - ) - plan = Plan.objects.create(is_default=True, name='Default') - service.rates.create(plan=plan, quantity=1, price=0) - service.rates.create(plan=plan, quantity=2, price=10) - service.rates.create(plan=plan, quantity=4, price=9) - service.rates.create(plan=plan, quantity=6, price=6) - return service - - def create_domain(self, account=None): - if not account: - account = self.create_account() - domain_name = '%s.es' % random_ascii(10) - domain_service, __ = MiscService.objects.get_or_create(name='domain .es', description='Domain .ES') - return Miscellaneous.objects.create(service=domain_service, description=domain_name, account=account) - - def test_domain(self): - service = self.create_domain_service() - account = self.create_account() - self.create_domain(account=account) - bills = service.orders.bill() - self.assertEqual(0, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill() - self.assertEqual(10, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill() - self.assertEqual(20, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill() - self.assertEqual(29, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill() - self.assertEqual(38, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill() - self.assertEqual(44, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill() - self.assertEqual(50, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill() - self.assertEqual(56, bills[0].get_total()) - - def test_domain_proforma(self): - service = self.create_domain_service() - account = self.create_account() - self.create_domain(account=account) - bills = service.orders.bill(proforma=True, new_open=True) - self.assertEqual(0, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(proforma=True, new_open=True) - self.assertEqual(10, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(proforma=True, new_open=True) - self.assertEqual(20, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(proforma=True, new_open=True) - self.assertEqual(29, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(proforma=True, new_open=True) - self.assertEqual(38, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(proforma=True, new_open=True) - self.assertEqual(44, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(proforma=True, new_open=True) - self.assertEqual(50, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(proforma=True, new_open=True) - self.assertEqual(56, bills[0].get_total()) - - def test_domain_cumulative(self): - service = self.create_domain_service() - account = self.create_account() - self.create_domain(account=account) - bills = service.orders.bill(proforma=True) - self.assertEqual(0, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(proforma=True) - self.assertEqual(10, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(proforma=True) - self.assertEqual(30, bills[0].get_total()) - - def test_domain_new_open(self): - service = self.create_domain_service() - account = self.create_account() - self.create_domain(account=account) - bills = service.orders.bill(new_open=True) - self.assertEqual(0, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(new_open=True) - self.assertEqual(10, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(new_open=True) - self.assertEqual(10, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(new_open=True) - self.assertEqual(9, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(new_open=True) - self.assertEqual(9, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(new_open=True) - self.assertEqual(6, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(new_open=True) - self.assertEqual(6, bills[0].get_total()) - self.create_domain(account=account) - bills = service.orders.bill(new_open=True) - self.assertEqual(6, bills[0].get_total()) - - -class BaseTrafficBillingTest(BaseBillingTest): - METRIC = 'account.resources.traffic.used' - - def create_traffic_service(self): - service = Service.objects.create( - description="Traffic", - content_type=ContentType.objects.get_for_model(Account), - match="account.is_active", - billing_period=Service.MONTHLY, - billing_point=Service.FIXED_DATE, - is_fee=False, - metric=self.METRIC, - pricing_period=Service.BILLING_PERIOD, - rate_algorithm=Service.STEP_PRICE, - on_cancel=Service.NOTHING, - payment_style=Service.POSTPAY, - tax=0, - nominal_price=10 - ) - plan = Plan.objects.create(is_default=True, name='Default') - service.rates.create(plan=plan, quantity=1, price=0) - service.rates.create(plan=plan, quantity=11, price=10) - return service - - def create_traffic_resource(self): - self.resource = Resource.objects.create( - name='traffic', - content_type=ContentType.objects.get_for_model(Account), - period=Resource.MONTHLY_SUM, - verbose_name='Account Traffic', - unit='GB', - scale=10**9, - ondemand=True, - monitors='FTPTraffic', - ) - return self.resource - - def report_traffic(self, account, date, value): - ct = ContentType.objects.get_for_model(Account) - object_id = account.pk - MonitorData.objects.create(monitor='FTPTraffic', content_object=account.user, value=value, date=date) - data = ResourceData.get_or_create(account, self.resource) - data.update() - - -class TrafficBillingTest(BaseTrafficBillingTest): - def test_traffic(self): - service = self.create_traffic_service() - resource = self.create_traffic_resource() - account = self.create_account() - - self.report_traffic(account, timezone.now(), 10**9) - bills = service.orders.bill(commit=False) - self.assertEqual([(account, [])], bills) - - # move into the past - delta = datetime.timedelta(days=60) - date = (timezone.now()-delta).date() - order = service.orders.get() - order.registered_on = date - order.save() - - self.report_traffic(account, date, 10**9*9) - order.metrics.update(updated_on=F('updated_on')-delta) - bills = service.orders.bill(proforma=True) - self.assertEqual(0, bills[0].get_total()) - - self.report_traffic(account, date, 10**10*9) - metric = order.metrics.latest() - metric.updated_on -= delta - metric.save() - bills = service.orders.bill(proforma=True) - self.assertEqual(900, bills[0].get_total()) - - def test_multiple_traffics(self): - service = self.create_traffic_service() - resource = self.create_traffic_resource() - account1 = self.create_account() - account2 = self.create_account() - # TODO - - -class TrafficPrepayBillingTest(BaseTrafficBillingTest): - METRIC = "max((account.resources.traffic.used or 0) - getattr(account.miscellaneous.filter(is_active=True, service__name='traffic prepay').last(), 'amount', 0), 0)" - - def create_prepay_service(self): - service = Service.objects.create( - description="Traffic prepay", - content_type=ContentType.objects.get_for_model(Miscellaneous), - match="miscellaneous.is_active and miscellaneous.service.name.lower() == 'traffic prepay'", - billing_period=Service.MONTHLY, - # make sure full months are always paid - billing_point=Service.ON_REGISTER, - is_fee=False, - metric="miscellaneous.amount", - pricing_period=Service.NEVER, - rate_algorithm=Service.STEP_PRICE, - on_cancel=Service.NOTHING, - payment_style=Service.PREPAY, - tax=0, - nominal_price=50 - ) - return service - - def create_prepay(self, amount, account=None): - if not account: - account = self.create_account() - name = 'traffic prepay' - service, __ = MiscService.objects.get_or_create(name='traffic prepay', description='Traffic prepay', has_amount=True) - return Miscellaneous.objects.create(service=service, description=name, account=account, amount=amount) - - def test_traffic_prepay(self): - service = self.create_traffic_service() - prepay_service = self.create_prepay_service() - account = self.create_account() - self.create_traffic_resource() - now = timezone.now() - - prepay = self.create_prepay(10, account=account) - bill = account.orders.bill(proforma=True)[0] - self.assertEqual(10*50, bill.get_total()) - - self.report_traffic(account, timezone.now(), 10**10) - with freeze_time(now+relativedelta(months=1)): - bill = account.orders.bill(proforma=True, new_open=True)[0] - self.assertEqual(2*10*50 + 0*10, bill.get_total()) - - # TODO dateutils.relativedelta is buggy with fakedatetime - # TODO RuntimeWarning: DateTimeField MetricStorage.updated_on received a naive - self.report_traffic(account, timezone.now(), 10**10) - with freeze_time(now+relativedelta(months=1)): - bill = account.orders.bill(proforma=True, new_open=True)[0] - self.assertEqual(2*10*50 + 0*10, bill.get_total()) - - self.report_traffic(account, timezone.now(), 10**10) - with freeze_time(now+relativedelta(months=1)): - bill = account.orders.bill(proforma=True, new_open=True)[0] - self.assertEqual(2*10*50 + (30-10-10)*10, bill.get_total()) - - with freeze_time(now+relativedelta(months=2)): - self.report_traffic(account, timezone.now(), 10**11) - with freeze_time(now+relativedelta(months=1)): - bill = account.orders.bill(proforma=True, new_open=True)[0] - self.assertEqual(2*10*50 + (30-10-10)*10, bill.get_total()) - - with freeze_time(now+relativedelta(months=3)): - bill = account.orders.bill(proforma=True, new_open=True)[0] - self.assertEqual(4*10*50 + (30-10-10)*10 + (100-10-10)*10, bill.get_total()) - - # TODO metric on the current day! how to solve it consistently? - # TODO prepay doesnt allow for discount - - - -class MailboxBillingTest(BaseBillingTest): - def create_mailbox_service(self): - service = Service.objects.create( - description="Mailbox", - content_type=ContentType.objects.get_for_model(Mailbox), - match="True", - billing_period=Service.ANUAL, - billing_point=Service.FIXED_DATE, - is_fee=False, - metric='', - pricing_period=Service.NEVER, - rate_algorithm=Service.STEP_PRICE, - on_cancel=Service.COMPENSATE, - payment_style=Service.PREPAY, - tax=0, - nominal_price=10 - ) - plan = Plan.objects.create(is_default=True, name='Default') - service.rates.create(plan=plan, quantity=1, price=0) - service.rates.create(plan=plan, quantity=5, price=10) - return service - - def create_mailbox_disk_service(self): - service = Service.objects.create( - description="Mailbox disk", - content_type=ContentType.objects.get_for_model(Mailbox), - match="True", - billing_period=Service.ANUAL, - billing_point=Service.FIXED_DATE, - is_fee=False, - metric='max((mailbox.resources.disk.allocated or 0) -1, 0)', - pricing_period=Service.NEVER, - rate_algorithm=Service.STEP_PRICE, - on_cancel=Service.DISCOUNT, - payment_style=Service.PREPAY, - tax=0, - nominal_price=10 - ) - plan = Plan.objects.create(is_default=True, name='Default') - service.rates.create(plan=plan, quantity=1, price=10) - return service - - def create_disk_resource(self): - self.resource = Resource.objects.create( - name='disk', - content_type=ContentType.objects.get_for_model(Mailbox), - period=Resource.LAST, - verbose_name='Mailbox disk', - unit='GB', - scale=10**9, - ondemand=False, - monitors='MaildirDisk', - ) - return self.resource - - def allocate_disk(self, mailbox, value): - data = ResourceData.get_or_create(mailbox, self.resource) - data.allocated = value - data.save() - - def create_mailbox(self, account=None): - if not account: - account = self.create_account() - mailbox_name = '%s@orchestra.lan' % random_ascii(10) - return Mailbox.objects.create(name=mailbox_name, account=account) - - def test_mailbox_size(self): - service = self.create_mailbox_service() - disk_service = self.create_mailbox_disk_service() - self.create_disk_resource() - account = self.create_account() - mailbox = self.create_mailbox(account=account) - self.allocate_disk(mailbox, 10) - bill = service.orders.bill()[0] - self.assertEqual(0, bill.get_total()) - bp = timezone.now().date() + relativedelta(years=1) - bill = disk_service.orders.bill(billing_point=bp, fixed_point=True)[0] - self.assertEqual(90, bill.get_total()) - mailbox = self.create_mailbox(account=account) - mailbox = self.create_mailbox(account=account) - mailbox = self.create_mailbox(account=account) - mailbox = self.create_mailbox(account=account) - mailbox = self.create_mailbox(account=account) - mailbox = self.create_mailbox(account=account) - bill = service.orders.bill(billing_point=bp, fixed_point=True)[0] - self.assertEqual(120, bill.get_total()) - - def test_mailbox_size_with_changes(self): - service = self.create_mailbox_disk_service() - self.create_disk_resource() - account = self.create_account() - mailbox = self.create_mailbox(account=account) - now = timezone.now() - bp = now.date() + relativedelta(years=1) - options = dict(billing_point=bp, fixed_point=True, proforma=True, new_open=True) - - self.allocate_disk(mailbox, 10) - bill = service.orders.bill(**options).pop() - self.assertEqual(9*10, bill.get_total()) - - with freeze_time(now+relativedelta(months=6)): - self.allocate_disk(mailbox, 20) - bill = service.orders.bill(**options).pop() - total = 9*10*0.5 + 19*10*0.5 - self.assertEqual(total, bill.get_total()) - - with freeze_time(now+relativedelta(months=9)): - self.allocate_disk(mailbox, 30) - bill = service.orders.bill(**options).pop() - total = 9*10*0.5 + 19*10*0.25 + 29*10*0.25 - self.assertEqual(total, bill.get_total()) - - with freeze_time(now+relativedelta(years=1)): - self.allocate_disk(mailbox, 10) - bill = service.orders.bill(**options).pop() - total = 9*10*0.5 + 19*10*0.25 + 29*10*0.25 - self.assertEqual(total, bill.get_total()) - - -class JobBillingTest(BaseBillingTest): - def create_job_service(self): - service = Service.objects.create( - description="Random job", - content_type=ContentType.objects.get_for_model(Miscellaneous), - match="miscellaneous.is_active and miscellaneous.service.name.lower() == 'job'", - billing_period=Service.NEVER, - billing_point=Service.ON_REGISTER, - is_fee=False, - metric='miscellaneous.amount', - pricing_period=Service.BILLING_PERIOD, - rate_algorithm=Service.MATCH_PRICE, - on_cancel=Service.NOTHING, - payment_style=Service.POSTPAY, - tax=0, - nominal_price=20 - ) - plan = Plan.objects.create(is_default=True, name='Default') - service.rates.create(plan=plan, quantity=1, price=20) - service.rates.create(plan=plan, quantity=10, price=15) - return service - - def create_job(self, amount, account=None): - if not account: - account = self.create_account() - description = 'Random Job %s' % random_ascii(10) - service, __ = MiscService.objects.get_or_create(name='job', description=description, has_amount=True) - return Miscellaneous.objects.create(service=service, description=description, account=account, amount=amount) - - def test_job(self): - service = self.create_job_service() - account = self.create_account() - - job = self.create_job(5, account=account) - bill = service.orders.bill()[0] - self.assertEqual(5*20, bill.get_total()) - - job = self.create_job(100, account=account) - bill = service.orders.bill(new_open=True)[0] - self.assertEqual(100*15, bill.get_total()) - - -class PlanBillingTest(BaseBillingTest): - def create_plan_service(self): - service = Service.objects.create( - description="Association membership fee", - content_type=ContentType.objects.get_for_model(Miscellaneous), - match="account.is_active and account.type == 'ASSOCIATION'", - billing_period=Service.ANUAL, - billing_point=Service.FIXED_DATE, - is_fee=True, - metric='', - pricing_period=Service.BILLING_PERIOD, - rate_algorithm=Service.STEP_PRICE, - on_cancel=Service.DISCOUNT, - payment_style=Service.PREPAY, - tax=0, - nominal_price=20 - ) - return service - - def create_plan(self): - if not account: - account = self.create_account() - domain_name = '%s.es' % random_ascii(10) - domain_service, __ = MiscService.objects.get_or_create(name='domain .es', description='Domain .ES') - return Miscellaneous.objects.create(service=domain_service, description=domain_name, account=account) - - def test_plan(self): - pass - - -# TODO web disk size diff --git a/orchestra/apps/services/handlers.py b/orchestra/apps/services/handlers.py index 5c4a4686..00c15df9 100644 --- a/orchestra/apps/services/handlers.py +++ b/orchestra/apps/services/handlers.py @@ -454,7 +454,9 @@ class ServiceHandler(plugins.Plugin): else: lines = self.bill_with_metric(orders, account, **options) if options.get('commit', True): + now = timezone.now().date() for line in lines: + line.order.billed_on = now line.order.billed_until = line.order.new_billed_until line.order.save() return lines diff --git a/orchestra/apps/services/tests/functional_tests/__init__.py b/orchestra/apps/services/tests/functional_tests/__init__.py new file mode 100644 index 00000000..feac27d8 --- /dev/null +++ b/orchestra/apps/services/tests/functional_tests/__init__.py @@ -0,0 +1,16 @@ +from orchestra.apps.accounts.models import Account +from orchestra.apps.users.models import User +from orchestra.utils.tests import BaseTestCase, random_ascii + + +class BaseBillingTest(BaseTestCase): + def create_account(self): + account = Account.objects.create() + username = 'account_%s' % random_ascii(5) + user = User.objects.create_user(username=username, account=account) + account.user = user + account.save() + return account + + +# TODO web disk diff --git a/orchestra/apps/services/tests/functional_tests/test_domain.py b/orchestra/apps/services/tests/functional_tests/test_domain.py new file mode 100644 index 00000000..ba760383 --- /dev/null +++ b/orchestra/apps/services/tests/functional_tests/test_domain.py @@ -0,0 +1,138 @@ +from django.contrib.contenttypes.models import ContentType + +from orchestra.apps.miscellaneous.models import MiscService, Miscellaneous +from orchestra.utils.tests import random_ascii + +from ...models import Service, Plan + +from . import BaseBillingTest + + +class DomainBillingTest(BaseBillingTest): + def create_domain_service(self): + service = Service.objects.create( + description="Domain .ES", + content_type=ContentType.objects.get_for_model(Miscellaneous), + match="miscellaneous.is_active and miscellaneous.service.name.lower() == 'domain .es'", + billing_period=Service.ANUAL, + billing_point=Service.ON_REGISTER, + is_fee=False, + metric='', + pricing_period=Service.BILLING_PERIOD, + rate_algorithm=Service.STEP_PRICE, + on_cancel=Service.NOTHING, + payment_style=Service.PREPAY, + tax=0, + nominal_price=10 + ) + plan = Plan.objects.create(is_default=True, name='Default') + service.rates.create(plan=plan, quantity=1, price=0) + service.rates.create(plan=plan, quantity=2, price=10) + service.rates.create(plan=plan, quantity=4, price=9) + service.rates.create(plan=plan, quantity=6, price=6) + return service + + def create_domain(self, account=None): + if not account: + account = self.create_account() + domain_name = '%s.es' % random_ascii(10) + domain_service, __ = MiscService.objects.get_or_create(name='domain .es', description='Domain .ES') + return Miscellaneous.objects.create(service=domain_service, description=domain_name, account=account) + + def test_domain(self): + service = self.create_domain_service() + account = self.create_account() + self.create_domain(account=account) + bills = account.orders.bill() + self.assertEqual(0, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill() + self.assertEqual(10, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill() + self.assertEqual(20, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill() + self.assertEqual(29, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill() + self.assertEqual(38, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill() + self.assertEqual(44, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill() + self.assertEqual(50, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill() + self.assertEqual(56, bills[0].get_total()) + + def test_domain_proforma(self): + service = self.create_domain_service() + account = self.create_account() + self.create_domain(account=account) + bills = account.orders.bill(proforma=True, new_open=True) + self.assertEqual(0, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(proforma=True, new_open=True) + self.assertEqual(10, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(proforma=True, new_open=True) + self.assertEqual(20, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(proforma=True, new_open=True) + self.assertEqual(29, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(proforma=True, new_open=True) + self.assertEqual(38, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(proforma=True, new_open=True) + self.assertEqual(44, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(proforma=True, new_open=True) + self.assertEqual(50, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(proforma=True, new_open=True) + self.assertEqual(56, bills[0].get_total()) + + def test_domain_cumulative(self): + service = self.create_domain_service() + account = self.create_account() + self.create_domain(account=account) + bills = account.orders.bill(proforma=True) + self.assertEqual(0, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(proforma=True) + self.assertEqual(10, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(proforma=True) + self.assertEqual(30, bills[0].get_total()) + + def test_domain_new_open(self): + service = self.create_domain_service() + account = self.create_account() + self.create_domain(account=account) + bills = account.orders.bill(new_open=True) + self.assertEqual(0, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(new_open=True) + self.assertEqual(10, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(new_open=True) + self.assertEqual(10, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(new_open=True) + self.assertEqual(9, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(new_open=True) + self.assertEqual(9, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(new_open=True) + self.assertEqual(6, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(new_open=True) + self.assertEqual(6, bills[0].get_total()) + self.create_domain(account=account) + bills = account.orders.bill(new_open=True) + self.assertEqual(6, bills[0].get_total()) + diff --git a/orchestra/apps/services/tests/functional_tests/test_ftp.py b/orchestra/apps/services/tests/functional_tests/test_ftp.py new file mode 100644 index 00000000..229492b7 --- /dev/null +++ b/orchestra/apps/services/tests/functional_tests/test_ftp.py @@ -0,0 +1,103 @@ +import decimal +import datetime + +from dateutil.relativedelta import relativedelta +from django.contrib.contenttypes.models import ContentType +from django.utils import timezone + +from orchestra.apps.users.models import User +from orchestra.utils.tests import random_ascii + +from ... import settings +from ...models import Service + +from . import BaseBillingTest + + +class FTPBillingTest(BaseBillingTest): + def create_ftp_service(self): + return Service.objects.create( + description="FTP Account", + content_type=ContentType.objects.get_for_model(User), + match='not user.is_main and user.has_posix()', + billing_period=Service.ANUAL, + billing_point=Service.FIXED_DATE, + is_fee=False, + metric='', + pricing_period=Service.NEVER, + rate_algorithm=Service.STEP_PRICE, + on_cancel=Service.COMPENSATE, + payment_style=Service.PREPAY, + tax=0, + nominal_price=10, + ) + + def create_ftp(self, account=None): + if not account: + account = self.create_account() + username = '%s_ftp' % random_ascii(10) + user = User.objects.create_user(username=username, account=account) + POSIX = user._meta.get_field_by_name('posix')[0].model + POSIX.objects.create(user=user) + return user + + def test_ftp_account_1_year_fiexed(self): + service = self.create_ftp_service() + user = self.create_ftp() + bp = timezone.now().date() + relativedelta(years=1) + bills = service.orders.bill(billing_point=bp, fixed_point=True) + self.assertEqual(10, bills[0].get_total()) + + def test_ftp_account_2_year_fiexed(self): + service = self.create_ftp_service() + user = self.create_ftp() + bp = timezone.now().date() + relativedelta(years=2) + bills = service.orders.bill(billing_point=bp, fixed_point=True) + self.assertEqual(20, bills[0].get_total()) + + def test_ftp_account_6_month_fixed(self): + service = self.create_ftp_service() + self.create_ftp() + bp = timezone.now().date() + relativedelta(months=6) + bills = service.orders.bill(billing_point=bp, fixed_point=True) + self.assertEqual(5, bills[0].get_total()) + + def test_ftp_account_next_billing_point(self): + service = self.create_ftp_service() + self.create_ftp() + now = timezone.now().date() + bp_month = settings.SERVICES_SERVICE_ANUAL_BILLING_MONTH + if now.month > bp_month: + bp = datetime.date(year=now.year+1, month=bp_month, day=1) + else: + bp = datetime.date(year=now.year, month=bp_month, day=1) + bills = service.orders.bill(billing_point=now, fixed_point=False) + size = decimal.Decimal((bp - now).days)/365 + error = decimal.Decimal(0.05) + self.assertGreater(10*size+error*(10*size), bills[0].get_total()) + self.assertLess(10*size-error*(10*size), bills[0].get_total()) + + def test_ftp_account_with_compensation(self): + account = self.create_account() + service = self.create_ftp_service() + user = self.create_ftp(account=account) + first_bp = timezone.now().date() + relativedelta(years=2) + bills = account.orders.bill(billing_point=first_bp, fixed_point=True) + self.assertEqual(1, account.orders.active().count()) + user.delete() + self.assertEqual(0, account.orders.active().count()) + user = self.create_ftp(account=account) + self.assertEqual(1, account.orders.active().count()) + self.assertEqual(2, account.orders.count()) + bp = timezone.now().date() + relativedelta(years=1) + bills = account.orders.bill(billing_point=bp, fixed_point=True, new_open=True) + discount = bills[0].lines.order_by('id')[0].sublines.get() + self.assertEqual(decimal.Decimal(-20), discount.total) + order = account.orders.order_by('id').first() + self.assertEqual(order.cancelled_on, order.billed_until) + order = account.orders.order_by('-id').first() + self.assertEqual(first_bp, order.billed_until) + self.assertEqual(decimal.Decimal(0), bills[0].get_total()) + + def test_ftp_account_with_rates(self): + pass diff --git a/orchestra/apps/services/tests/functional_tests/test_job.py b/orchestra/apps/services/tests/functional_tests/test_job.py new file mode 100644 index 00000000..a54071fe --- /dev/null +++ b/orchestra/apps/services/tests/functional_tests/test_job.py @@ -0,0 +1,52 @@ +from django.contrib.contenttypes.models import ContentType + +from orchestra.apps.miscellaneous.models import MiscService, Miscellaneous +from orchestra.utils.tests import random_ascii + +from ...models import Service, Plan + +from . import BaseBillingTest + + +class JobBillingTest(BaseBillingTest): + def create_job_service(self): + service = Service.objects.create( + description="Random job", + content_type=ContentType.objects.get_for_model(Miscellaneous), + match="miscellaneous.is_active and miscellaneous.service.name.lower() == 'job'", + billing_period=Service.NEVER, + billing_point=Service.ON_REGISTER, + is_fee=False, + metric='miscellaneous.amount', + pricing_period=Service.BILLING_PERIOD, + rate_algorithm=Service.MATCH_PRICE, + on_cancel=Service.NOTHING, + payment_style=Service.POSTPAY, + tax=0, + nominal_price=20 + ) + plan = Plan.objects.create(is_default=True, name='Default') + service.rates.create(plan=plan, quantity=1, price=20) + service.rates.create(plan=plan, quantity=10, price=15) + return service + + def create_job(self, amount, account=None): + if not account: + account = self.create_account() + description = 'Random Job %s' % random_ascii(10) + service, __ = MiscService.objects.get_or_create(name='job', description=description, has_amount=True) + return Miscellaneous.objects.create(service=service, description=description, account=account, amount=amount) + + def test_job(self): + service = self.create_job_service() + account = self.create_account() + + job = self.create_job(5, account=account) + bill = account.orders.bill()[0] + self.assertEqual(5*20, bill.get_total()) + + job = self.create_job(100, account=account) + bill = account.orders.bill(new_open=True)[0] + self.assertEqual(100*15, bill.get_total()) + + diff --git a/orchestra/apps/services/tests/functional_tests/test_mailbox.py b/orchestra/apps/services/tests/functional_tests/test_mailbox.py new file mode 100644 index 00000000..3060952d --- /dev/null +++ b/orchestra/apps/services/tests/functional_tests/test_mailbox.py @@ -0,0 +1,133 @@ +from dateutil.relativedelta import relativedelta +from django.contrib.contenttypes.models import ContentType +from django.utils import timezone +from freezegun import freeze_time + +from orchestra.apps.mails.models import Mailbox +from orchestra.apps.resources.models import Resource, ResourceData, MonitorData +from orchestra.utils.tests import random_ascii + +from ...models import Service, Plan + +from . import BaseBillingTest + + +class MailboxBillingTest(BaseBillingTest): + def create_mailbox_service(self): + service = Service.objects.create( + description="Mailbox", + content_type=ContentType.objects.get_for_model(Mailbox), + match="True", + billing_period=Service.ANUAL, + billing_point=Service.FIXED_DATE, + is_fee=False, + metric='', + pricing_period=Service.NEVER, + rate_algorithm=Service.STEP_PRICE, + on_cancel=Service.COMPENSATE, + payment_style=Service.PREPAY, + tax=0, + nominal_price=10 + ) + plan = Plan.objects.create(is_default=True, name='Default') + service.rates.create(plan=plan, quantity=1, price=0) + service.rates.create(plan=plan, quantity=5, price=10) + return service + + def create_mailbox_disk_service(self): + service = Service.objects.create( + description="Mailbox disk", + content_type=ContentType.objects.get_for_model(Mailbox), + match="True", + billing_period=Service.ANUAL, + billing_point=Service.FIXED_DATE, + is_fee=False, + metric='max((mailbox.resources.disk.allocated or 0) -1, 0)', + pricing_period=Service.NEVER, + rate_algorithm=Service.STEP_PRICE, + on_cancel=Service.DISCOUNT, + payment_style=Service.PREPAY, + tax=0, + nominal_price=10 + ) + plan = Plan.objects.create(is_default=True, name='Default') + service.rates.create(plan=plan, quantity=1, price=10) + return service + + def create_disk_resource(self): + self.resource = Resource.objects.create( + name='disk', + content_type=ContentType.objects.get_for_model(Mailbox), + period=Resource.LAST, + verbose_name='Mailbox disk', + unit='GB', + scale=10**9, + ondemand=False, + monitors='MaildirDisk', + ) + return self.resource + + def allocate_disk(self, mailbox, value): + data = ResourceData.get_or_create(mailbox, self.resource) + data.allocated = value + data.save() + + def create_mailbox(self, account=None): + if not account: + account = self.create_account() + mailbox_name = '%s@orchestra.lan' % random_ascii(10) + return Mailbox.objects.create(name=mailbox_name, account=account) + + def test_mailbox_size(self): + service = self.create_mailbox_service() + disk_service = self.create_mailbox_disk_service() + self.create_disk_resource() + account = self.create_account() + mailbox = self.create_mailbox(account=account) + self.allocate_disk(mailbox, 10) + bill = service.orders.bill()[0] + self.assertEqual(0, bill.get_total()) + bp = timezone.now().date() + relativedelta(years=1) + bill = disk_service.orders.bill(billing_point=bp, fixed_point=True)[0] + self.assertEqual(90, bill.get_total()) + mailbox = self.create_mailbox(account=account) + mailbox = self.create_mailbox(account=account) + mailbox = self.create_mailbox(account=account) + mailbox = self.create_mailbox(account=account) + mailbox = self.create_mailbox(account=account) + mailbox = self.create_mailbox(account=account) + bill = service.orders.bill(billing_point=bp, fixed_point=True)[0] + self.assertEqual(120, bill.get_total()) + + def test_mailbox_size_with_changes(self): + service = self.create_mailbox_disk_service() + self.create_disk_resource() + account = self.create_account() + mailbox = self.create_mailbox(account=account) + now = timezone.now() + bp = now.date() + relativedelta(years=1) + options = dict(billing_point=bp, fixed_point=True, proforma=True, new_open=True) + + self.allocate_disk(mailbox, 10) + bill = service.orders.bill(**options).pop() + self.assertEqual(9*10, bill.get_total()) + + with freeze_time(now+relativedelta(months=6)): + self.allocate_disk(mailbox, 20) + bill = service.orders.bill(**options).pop() + total = 9*10*0.5 + 19*10*0.5 + self.assertEqual(total, bill.get_total()) + + with freeze_time(now+relativedelta(months=9)): + self.allocate_disk(mailbox, 30) + bill = service.orders.bill(**options).pop() + total = 9*10*0.5 + 19*10*0.25 + 29*10*0.25 + self.assertEqual(total, bill.get_total()) + + with freeze_time(now+relativedelta(years=1)): + self.allocate_disk(mailbox, 10) + bill = service.orders.bill(**options).pop() + total = 9*10*0.5 + 19*10*0.25 + 29*10*0.25 + self.assertEqual(total, bill.get_total()) + + # TODO recharge missing stuff diff --git a/orchestra/apps/services/tests/functional_tests/test_plan.py b/orchestra/apps/services/tests/functional_tests/test_plan.py new file mode 100644 index 00000000..79cac8cc --- /dev/null +++ b/orchestra/apps/services/tests/functional_tests/test_plan.py @@ -0,0 +1,37 @@ +from django.contrib.contenttypes.models import ContentType + +from orchestra.apps.miscellaneous.models import MiscService, Miscellaneous + +from ...models import Service, Plan + +from . import BaseBillingTest + + +class PlanBillingTest(BaseBillingTest): + def create_plan_service(self): + service = Service.objects.create( + description="Association membership fee", + content_type=ContentType.objects.get_for_model(Miscellaneous), + match="account.is_active and account.type == 'ASSOCIATION'", + billing_period=Service.ANUAL, + billing_point=Service.FIXED_DATE, + is_fee=True, + metric='', + pricing_period=Service.BILLING_PERIOD, + rate_algorithm=Service.STEP_PRICE, + on_cancel=Service.DISCOUNT, + payment_style=Service.PREPAY, + tax=0, + nominal_price=20 + ) + return service + + def create_plan(self): + if not account: + account = self.create_account() + domain_name = '%s.es' % random_ascii(10) + domain_service, __ = MiscService.objects.get_or_create(name='domain .es', description='Domain .ES') + return Miscellaneous.objects.create(service=domain_service, description=domain_name, account=account) + + def test_plan(self): + pass diff --git a/orchestra/apps/services/tests/functional_tests/test_traffic.py b/orchestra/apps/services/tests/functional_tests/test_traffic.py new file mode 100644 index 00000000..b6dd17e2 --- /dev/null +++ b/orchestra/apps/services/tests/functional_tests/test_traffic.py @@ -0,0 +1,157 @@ +import datetime + +from dateutil.relativedelta import relativedelta +from django.contrib.contenttypes.models import ContentType +from django.utils import timezone +from freezegun import freeze_time + +from orchestra.apps.accounts.models import Account +from orchestra.apps.miscellaneous.models import MiscService, Miscellaneous +from orchestra.apps.resources.models import Resource, ResourceData, MonitorData + +from ...models import Service, Plan + +from . import BaseBillingTest + + +class BaseTrafficBillingTest(BaseBillingTest): + METRIC = 'account.resources.traffic.used' + + def create_traffic_service(self): + service = Service.objects.create( + description="Traffic", + content_type=ContentType.objects.get_for_model(Account), + match="account.is_active", + billing_period=Service.MONTHLY, + billing_point=Service.FIXED_DATE, + is_fee=False, + metric=self.METRIC, + pricing_period=Service.BILLING_PERIOD, + rate_algorithm=Service.STEP_PRICE, + on_cancel=Service.NOTHING, + payment_style=Service.POSTPAY, + tax=0, + nominal_price=10 + ) + plan = Plan.objects.create(is_default=True, name='Default') + service.rates.create(plan=plan, quantity=1, price=0) + service.rates.create(plan=plan, quantity=11, price=10) + return service + + def create_traffic_resource(self): + self.resource = Resource.objects.create( + name='traffic', + content_type=ContentType.objects.get_for_model(Account), + period=Resource.MONTHLY_SUM, + verbose_name='Account Traffic', + unit='GB', + scale=10**9, + ondemand=True, + monitors='FTPTraffic', + ) + return self.resource + + def report_traffic(self, account, value): + ct = ContentType.objects.get_for_model(Account) + object_id = account.pk + MonitorData.objects.create(monitor='FTPTraffic', content_object=account.user, value=value, date=timezone.now()) + data = ResourceData.get_or_create(account, self.resource) + data.update() + + +class TrafficBillingTest(BaseTrafficBillingTest): + def test_traffic(self): + service = self.create_traffic_service() + resource = self.create_traffic_resource() + account = self.create_account() + now = timezone.now().date() + + self.report_traffic(account, 10**9) + bill = account.orders.bill(commit=False)[0] + self.assertEqual((account, []), bill) + self.report_traffic(account, 10**9*9) + + with freeze_time(now+relativedelta(months=1)): + bill = account.orders.bill(proforma=True)[0] + self.report_traffic(account, 10**10*9) + self.assertEqual(0, bill.get_total()) + + with freeze_time(now+relativedelta(months=3)): + bill = account.orders.bill(proforma=True)[0] + self.assertEqual((90-10)*10, bill.get_total()) + + def test_multiple_traffics(self): + service = self.create_traffic_service() + resource = self.create_traffic_resource() + account1 = self.create_account() + account2 = self.create_account() + # TODO + + +class TrafficPrepayBillingTest(BaseTrafficBillingTest): + METRIC = "max((account.resources.traffic.used or 0) - getattr(account.miscellaneous.filter(is_active=True, service__name='traffic prepay').last(), 'amount', 0), 0)" + + def create_prepay_service(self): + service = Service.objects.create( + description="Traffic prepay", + content_type=ContentType.objects.get_for_model(Miscellaneous), + match="miscellaneous.is_active and miscellaneous.service.name.lower() == 'traffic prepay'", + billing_period=Service.MONTHLY, + # make sure full months are always paid + billing_point=Service.ON_REGISTER, + is_fee=False, + metric="miscellaneous.amount", + pricing_period=Service.NEVER, + rate_algorithm=Service.STEP_PRICE, + on_cancel=Service.NOTHING, + payment_style=Service.PREPAY, + tax=0, + nominal_price=50 + ) + return service + + def create_prepay(self, amount, account=None): + if not account: + account = self.create_account() + name = 'traffic prepay' + service, __ = MiscService.objects.get_or_create(name='traffic prepay', description='Traffic prepay', has_amount=True) + return Miscellaneous.objects.create(service=service, description=name, account=account, amount=amount) + + def test_traffic_prepay(self): + service = self.create_traffic_service() + prepay_service = self.create_prepay_service() + account = self.create_account() + self.create_traffic_resource() + now = timezone.now() + + prepay = self.create_prepay(10, account=account) + bill = account.orders.bill(proforma=True)[0] + self.assertEqual(10*50, bill.get_total()) + + self.report_traffic(account, 10**10) + with freeze_time(now+relativedelta(months=1)): + bill = account.orders.bill(proforma=True, new_open=True)[0] + self.assertEqual(2*10*50 + 0*10, bill.get_total()) + + # TODO dateutils.relativedelta is buggy with fakedatetime + # TODO RuntimeWarning: DateTimeField MetricStorage.updated_on received a naive + self.report_traffic(account, 10**10) + with freeze_time(now+relativedelta(months=1)): + bill = account.orders.bill(proforma=True, new_open=True)[0] + self.assertEqual(2*10*50 + 0*10, bill.get_total()) + + self.report_traffic(account, 10**10) + with freeze_time(now+relativedelta(months=1)): + bill = account.orders.bill(proforma=True, new_open=True)[0] + self.assertEqual(2*10*50 + (30-10-10)*10, bill.get_total()) + + with freeze_time(now+relativedelta(months=2)): + self.report_traffic(account, 10**11) + with freeze_time(now+relativedelta(months=1)): + bill = account.orders.bill(proforma=True, new_open=True)[0] + self.assertEqual(2*10*50 + (30-10-10)*10, bill.get_total()) + + with freeze_time(now+relativedelta(months=3)): + bill = account.orders.bill(proforma=True, new_open=True)[0] + self.assertEqual(4*10*50 + (30-10-10)*10 + (100-10-10)*10, bill.get_total()) +