Refactored orders.tests.functional_tests
This commit is contained in:
Marc
parent
a3696b859a
commit
46f9f4d88e
|
|
@ -32,7 +32,6 @@ class OrderQuerySet(models.QuerySet):
|
||||||
group_by = queryset.group_by
|
group_by = queryset.group_by
|
||||||
|
|
||||||
def bill(self, **options):
|
def bill(self, **options):
|
||||||
# TODO classmethod?
|
|
||||||
bills = []
|
bills = []
|
||||||
bill_backend = Order.get_bill_backend()
|
bill_backend = Order.get_bill_backend()
|
||||||
qs = self.select_related('account', 'service')
|
qs = self.select_related('account', 'service')
|
||||||
|
|
|
||||||
|
|
@ -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
|
|
||||||
|
|
@ -454,7 +454,9 @@ class ServiceHandler(plugins.Plugin):
|
||||||
else:
|
else:
|
||||||
lines = self.bill_with_metric(orders, account, **options)
|
lines = self.bill_with_metric(orders, account, **options)
|
||||||
if options.get('commit', True):
|
if options.get('commit', True):
|
||||||
|
now = timezone.now().date()
|
||||||
for line in lines:
|
for line in lines:
|
||||||
|
line.order.billed_on = now
|
||||||
line.order.billed_until = line.order.new_billed_until
|
line.order.billed_until = line.order.new_billed_until
|
||||||
line.order.save()
|
line.order.save()
|
||||||
return lines
|
return lines
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
@ -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())
|
||||||
|
|
||||||
|
|
@ -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
|
||||||
|
|
@ -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())
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -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
|
||||||
|
|
@ -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
|
||||||
|
|
@ -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())
|
||||||
|
|
||||||
Loading…
Reference in New Issue