def _devices_sample(self):
self.device1 = utils.get_or_create_device()
self.device2 = utils.get_or_create_device()
self.venture_device = utils.get_or_create_venture()
start = datetime.date(2013, 10, 8)
end = datetime.date(2013, 10, 20)
base_usage_types = models.UsageType.objects.filter(type='BU')
for i, device in enumerate([self.device1, self.device2], start=1):
for j, ut in enumerate(base_usage_types, start=1):
for k, day in enumerate(rrule.rrule(
rrule.DAILY,
dtstart=start,
until=end
)):
daily_usage = models.DailyUsage(
date=day,
pricing_venture=self.venture_device,
pricing_device=device,
value=10 * i,
type=ut,
)
if ut.by_warehouse:
daily_usage.warehouse = (
self.warehouses[k % len(self.warehouses)]
)
daily_usage.save()
def test_usages_count_price(self):
day = datetime.date(2013, 4, 25)
venture = models.Venture(venture_id=3)
venture.save()
usage_type = models.UsageType(name='waciki')
usage_type.save()
daily_usage = models.DailyUsage(
type=usage_type,
value=32,
date=day,
pricing_venture=venture,
)
daily_usage.save()
other_day = datetime.date(2013, 4, 24)
daily_usage = models.DailyUsage(
type=usage_type,
value=32,
date=other_day,
pricing_venture=venture,
)
daily_usage.save()
usage_price = models.UsagePrice(
start=day,
end=day,
price=4,
type=usage_type,
)
usage_price.save()
count, price = venture.get_usages_count_price(day, day, usage_type)
self.assertEquals(count, 32)
self.assertEquals(price, decimal.Decimal('128'))
day = datetime.date(2013, 4, 26)
count, price = venture.get_usages_count_price(day, day, usage_type)
self.assertEquals(count, 0)
self.assertEquals(price, decimal.Decimal('0'))
day = datetime.date(2013, 4, 24)
count, price = venture.get_usages_count_price(day, day, usage_type)
self.assertEquals(count, 32)
self.assertIsNone(price)
def test_total_cost(self):
# add regular usage type
regular_usage_type = models.UsageType(
name='RUsageType1',
symbol='rut1',
by_cost=True,
type='RU',
)
regular_usage_type.save()
models.DailyUsage(type=regular_usage_type,
pricing_venture=self.venture1,
value=100,
date=datetime.date(2013, 10, 11)).save()
models.DailyUsage(type=regular_usage_type,
pricing_venture=self.venture2,
value=300,
date=datetime.date(2013, 10, 12)).save()
models.ServiceUsageTypes(
usage_type=self.service_usage_type1,
service=self.service,
start=datetime.date(2013, 10, 1),
end=datetime.date(2013, 10, 30),
percent=100,
).save()
models.UsagePrice(type=regular_usage_type,
start=datetime.date(2013, 10, 1),
end=datetime.date(2013, 10, 30),
cost=1000).save()
self.service.regular_usage_types.add(regular_usage_type)
self.service.save()
self.assertEqual(
D(4490),
ServicePlugin.total_cost(datetime.date(2013, 10, 10),
datetime.date(2013, 10, 20), self.service,
False, [self.venture1]))
def test_usage_models(self):
type_ = models.UsageType(name='ziew')
type_.save()
price = models.UsagePrice(
type=type_,
price=3,
start=datetime.date(2013, 4, 24),
end=datetime.date(2013, 4, 26),
)
price.save()
usage = models.DailyUsage(
date=datetime.date(2013, 4, 25),
value=3,
type=type_,
)
usage.save()
self.assertEquals(usage.value, 3)
self.assertEquals(usage.type, type_)
self.assertEquals(price.price, 3)
self.assertEquals(type_.name, 'ziew')
def test_get_usages_no_price(self):
start = datetime.date(2013, 11, 8)
end = datetime.date(2013, 11, 22)
base_usage_types = models.UsageType.objects.filter(type='BU')
for i, ut in enumerate(base_usage_types, start=1):
days = rrule.rrule(rrule.DAILY, dtstart=start, until=end)
for j, day in enumerate(days, start=1):
for k, venture in enumerate(self.ventures, start=1):
daily_usage = models.DailyUsage(
date=day,
pricing_venture=venture,
value=10 * i * k,
type=ut,
)
if ut.by_warehouse:
daily_usage.warehouse = (
self.warehouses[j % len(self.warehouses)])
daily_usage.save()
result = UsagePlugin.costs(
start=datetime.date(2013, 11, 10),
end=datetime.date(2013, 11, 20),
usage_type=self.usage_type,
ventures=self.ventures_subset,
forecast=False,
no_price_msg=True,
)
self.assertEquals(
result,
{
1: {
'ut_1_count': 110.0, # 11 * 10
'ut_1_cost': _('No price'),
},
2: {
'ut_1_count': 220.0, # 11 * 20
'ut_1_cost': _('No price'),
}
})
def test_get_dailyusages(self):
# test if sum of usages per day if properly calculated
du = models.DailyUsage(
date=datetime.date(2013, 10, 11),
pricing_venture=self.venture2,
pricing_device=utils.get_or_create_device(),
value=100,
type=self.usage_type,
)
du.save()
result = UsagePlugin(
usage_type=self.usage_type,
start=datetime.date(2013, 10, 10),
end=datetime.date(2013, 10, 13),
ventures=self.ventures_subset,
type='dailyusages',
)
self.assertEquals(
result,
{
datetime.date(2013, 10, 10): {
self.venture1.id: 10,
self.venture2.id: 20,
},
datetime.date(2013, 10, 11): {
self.venture1.id: 10,
self.venture2.id: 120, # additional usage!
},
datetime.date(2013, 10, 12): {
self.venture1.id: 10,
self.venture2.id: 20,
},
datetime.date(2013, 10, 13): {
self.venture1.id: 10,
self.venture2.id: 20,
},
})
def setUp(self):
self.plugin = SampleReportPlugin()
# usage types
self.usage_type = models.UsageType(
name='UsageType1',
symbol='ut1',
by_warehouse=False,
by_cost=False,
type='BU',
)
self.usage_type.save()
self.usage_type_cost_wh = models.UsageType(
name='UsageType2',
symbol='ut2',
by_warehouse=True,
by_cost=True,
type='BU',
)
self.usage_type_cost_wh.save()
# warehouses
self.warehouse1 = models.Warehouse(
name='Warehouse1',
show_in_report=True,
)
self.warehouse1.save()
self.warehouse2 = models.Warehouse(
name='Warehouse2',
show_in_report=True,
)
self.warehouse2.save()
self.warehouses = models.Warehouse.objects.all()
# ventures
self.venture1 = models.Venture(
venture_id=1,
name='V1',
is_active=True,
)
self.venture1.save()
self.venture2 = models.Venture(
venture_id=2,
name='V2',
is_active=True,
)
self.venture2.save()
self.venture3 = models.Venture(venture_id=3, name='V3', is_active=True)
self.venture3.save()
self.venture4 = models.Venture(venture_id=4, name='V4', is_active=True)
self.venture4.save()
self.ventures = models.Venture.objects.all()
# daily usages of base type
# ut1:
# venture1: 10
# venture2: 20
# ut2:
# venture1: 20 (half in warehouse1, half in warehouse2)
# venture2: 40 (half in warehouse1, half in warehouse2)
start = datetime.date(2013, 10, 8)
end = datetime.date(2013, 10, 22)
base_usage_types = models.UsageType.objects.filter(type='BU')
for i, ut in enumerate(base_usage_types, start=1):
days = rrule.rrule(rrule.DAILY, dtstart=start, until=end)
for j, day in enumerate(days, start=1):
for k, venture in enumerate(self.ventures, start=1):
daily_usage = models.DailyUsage(
date=day,
pricing_venture=venture,
value=10 * i * k,
type=ut,
)
if ut.by_warehouse:
daily_usage.warehouse = (
self.warehouses[j % len(self.warehouses)]
)
daily_usage.save()
# usage prices
dates = [
(datetime.date(2013, 10, 5), datetime.date(2013, 10, 12)),
(datetime.date(2013, 10, 13), datetime.date(2013, 10, 17)),
(datetime.date(2013, 10, 18), datetime.date(2013, 10, 25)),
]
# (real cost/price, forecast cost/price)
ut_prices_costs = [
(self.usage_type, [(10, 50), (20, 60), (30, 70)]),
(self.usage_type_cost_wh, [
[(3600, 2400), (5400, 5400), (4800, 12000)], # warehouse1
[(3600, 5400), (3600, 1200), (7200, 3600)], # warehouse2
]),
]
def add_usage_price(usage_type, prices_costs, warehouse=None):
for daterange, price_cost in zip(dates, prices_costs):
start, end = daterange
usage_price = models.UsagePrice(
type=usage_type,
start=start,
end=end,
)
if warehouse is not None:
usage_price.warehouse = warehouse
if usage_type.by_cost:
usage_price.cost = price_cost[0]
usage_price.forecast_cost = price_cost[1]
else:
usage_price.price = price_cost[0]
usage_price.forecast_price = price_cost[1]
usage_price.save()
for ut, prices in ut_prices_costs:
if ut.by_warehouse:
for i, prices_wh in enumerate(prices):
warehouse = self.warehouses[i]
add_usage_price(ut, prices_wh, warehouse)
else:
add_usage_price(ut, prices)
def setUp(self):
# usage types
self.usage_type = models.UsageType(
name='UsageType1',
symbol='ut1',
by_warehouse=False,
by_cost=False,
type='BU',
divide_by=1,
)
self.usage_type.save()
self.usage_type_cost_wh = models.UsageType(
name='UsageType2',
symbol='ut2',
by_warehouse=True,
by_cost=True,
type='BU',
rounding=2,
)
self.usage_type_cost_wh.save()
self.usage_type_cost_sum = models.UsageType(
name='UsageType3',
symbol='ut3',
by_warehouse=False,
by_cost=True,
type='BU',
by_internet_provider=True,
rounding=1,
)
self.usage_type_cost_sum.save()
self.usage_type_average = models.UsageType(
name='UsageType4',
symbol='ut4',
by_warehouse=False,
by_cost=False,
type='BU',
average=True,
divide_by=2,
)
self.usage_type_average.save()
# warehouses
self.warehouse1 = models.Warehouse(
name='Warehouse1',
show_in_report=True,
)
self.warehouse1.save()
self.warehouse2 = models.Warehouse(
name='Warehouse2',
show_in_report=True,
)
self.warehouse2.save()
self.warehouses = models.Warehouse.objects.all()
# internet providers
self.net_provider1 = models.InternetProvider(
name='InternetProvider1', )
self.net_provider1.save()
self.net_provider2 = models.InternetProvider(
name='InternetProvider2', )
self.net_provider2.save()
self.net_providers = models.InternetProvider.objects.all()
# ventures
self.venture1 = models.Venture(venture_id=1, name='V1', is_active=True)
self.venture1.save()
self.venture2 = models.Venture(venture_id=2, name='V2', is_active=True)
self.venture2.save()
self.venture3 = models.Venture(venture_id=3, name='V3', is_active=True)
self.venture3.save()
self.venture4 = models.Venture(venture_id=4, name='V4', is_active=True)
self.venture4.save()
self.ventures_subset = [self.venture1, self.venture2]
self.ventures = models.Venture.objects.all()
# daily usages of base type
# ut1:
# venture1: 10
# venture2: 20
# ut2:
# venture1: 20 (half in warehouse1, half in warehouse2)
# venture2: 40 (half in warehouse1, half in warehouse2)
# ut3:
# venture1: 30
# venture2: 60
# ut4:
# venture1: 40
# venture2: 80
start = datetime.date(2013, 10, 8)
end = datetime.date(2013, 10, 22)
base_usage_types = models.UsageType.objects.filter(type='BU')
self.ventures_devices = {}
for k, venture in enumerate(self.ventures, start=1):
device = utils.get_or_create_device()
self.ventures_devices[venture] = device
for i, ut in enumerate(base_usage_types, start=1):
days = rrule.rrule(rrule.DAILY, dtstart=start, until=end)
for j, day in enumerate(days, start=1):
daily_usage = models.DailyUsage(
date=day,
pricing_venture=venture,
pricing_device=device,
value=10 * i * k,
type=ut,
)
if ut.by_warehouse:
daily_usage.warehouse = (
self.warehouses[j % len(self.warehouses)])
daily_usage.save()
# usage prices
dates = [
(datetime.date(2013, 10, 5), datetime.date(2013, 10, 12)),
(datetime.date(2013, 10, 13), datetime.date(2013, 10, 17)),
(datetime.date(2013, 10, 18), datetime.date(2013, 10, 25)),
]
# (real cost/price, forecast cost/price)
ut_prices_costs = [
(self.usage_type, [(10, 50), (20, 60), (30, 70)]),
(
self.usage_type_cost_wh,
[
[(3600, 2400), (5400, 5400), (4800, 12000)], # warehouse1
[(3600, 5400), (3600, 1200), (7200, 3600)], # warehouse2
]),
(
self.usage_type_cost_sum,
[
[(1000, 2000), (2000, 3000), (4000, 5000)], # provider 1
[(10000, 20000), (20000, 30000),
(40000, 50000)], # provider 2
]),
(self.usage_type_average, [(10, 20), (20, 30), (30, 40)]),
]
def add_usage_price(usage_type,
prices_costs,
net_provider=None,
warehouse=None):
for daterange, price_cost in zip(dates, prices_costs):
start, end = daterange
usage_price = models.UsagePrice(
type=usage_type,
start=start,
end=end,
)
if warehouse is not None:
usage_price.warehouse = warehouse
if net_provider is not None:
usage_price.internet_provider = net_provider
if usage_type.by_cost:
usage_price.cost = price_cost[0]
usage_price.forecast_cost = price_cost[1]
else:
usage_price.price = price_cost[0]
usage_price.forecast_price = price_cost[1]
usage_price.save()
for ut, prices in ut_prices_costs:
if ut.by_warehouse:
for i, prices_wh in enumerate(prices):
warehouse = self.warehouses[i]
add_usage_price(ut, prices_wh, warehouse=warehouse)
elif ut.by_internet_provider:
for i, prices_ip in enumerate(prices):
net_provider = self.net_providers[i]
add_usage_price(ut, prices_ip, net_provider=net_provider)
else:
add_usage_price(ut, prices)
def test_ventures_all(self):
day = datetime.date(2013, 4, 25)
venture = models.Venture(venture_id=3, name='a')
venture.save()
device = models.Device(
device_id=3,
asset_id=5,
)
device.save()
daily = models.DailyDevice(
pricing_device=device,
date=day,
name='ziew',
price='1337',
pricing_venture=venture,
)
daily.save()
subventure = models.Venture(venture_id=2, parent=venture, name='b')
subventure.save()
other_device = models.Device(
device_id=2,
asset_id=3,
)
other_device.save()
other_daily = models.DailyDevice(
pricing_device=other_device,
date=day,
name='ziew',
price='833833',
pricing_venture=subventure,
)
other_daily.save()
usage_type = models.UsageType(name='waciki')
usage_type.save()
daily_usage = models.DailyUsage(
type=usage_type,
value=32,
date=day,
pricing_venture=venture,
)
daily_usage.save()
extra_cost_type = models.ExtraCostType(name='waciki')
extra_cost_type.save()
extra_cost = models.ExtraCost(
pricing_venture=venture,
start=day,
end=day,
type=extra_cost_type,
price='65535',
)
extra_cost.save()
view = TopVentures()
for progress, data in view.get_data(day, day, show_in_ralph=True):
pass
self.assertEquals(
data,
[
[
3, # id
'a', # path
True, # show_in_ralph
'', # department
'', # business segment
'', # Profit center
2.0, # assets count
'835 170.00 PLN', # assets price
'0.00 PLN', # assets cost
32.0, # usage count
'0.00 PLN', # usage price
'65 535.00 PLN', # extra cost
],
],
)
view = AllVentures()
for progress, data in view.get_data(day, day, show_in_ralph=True):
pass
self.assertEquals(
data,
[
[
3,
'a',
True, # show_in_ralph
'',
'',
'',
1.0,
'1 337.00 PLN',
'0.00 PLN',
32.0,
'NO PRICE',
'65 535.00 PLN',
],
[
2,
'a/b',
True, # show_in_ralph
'',
'',
'',
1.0,
'833 833.00 PLN',
'0.00 PLN',
0,
'0.00 PLN',
'0.00 PLN',
],
],
)
def setUp(self):
# usage types
self.usage_type = models.UsageType(
name='UsageType1',
symbol='ut1',
by_warehouse=False,
by_cost=False,
type='BU',
)
self.usage_type.save()
self.usage_type_cost_wh = models.UsageType(
name='UsageType2',
symbol='ut2',
by_warehouse=True,
by_cost=True,
type='BU',
)
self.usage_type_cost_wh.save()
self.service_usage_type1 = models.UsageType(
name='ServiceUsageType1',
symbol='sut1',
type='SU',
divide_by=5,
rounding=2,
)
self.service_usage_type1.save()
self.service_usage_type2 = models.UsageType(
name='ServiceUsageType2',
# symbol='sut2',
type='SU',
)
self.service_usage_type2.save()
# warehouses
self.warehouse1 = models.Warehouse(
name='Warehouse1',
show_in_report=True,
)
self.warehouse1.save()
self.warehouse2 = models.Warehouse(
name='Warehouse2',
show_in_report=True,
)
self.warehouse2.save()
self.warehouses = models.Warehouse.objects.all()
# service
self.service = models.Service(name='Service1')
self.service.save()
self.service.base_usage_types.add(self.usage_type_cost_wh,
self.usage_type)
models.ServiceUsageTypes(
usage_type=self.service_usage_type1,
service=self.service,
start=datetime.date(2013, 10, 1),
end=datetime.date(2013, 10, 15),
percent=30,
).save()
models.ServiceUsageTypes(
usage_type=self.service_usage_type2,
service=self.service,
start=datetime.date(2013, 10, 1),
end=datetime.date(2013, 10, 15),
percent=70,
).save()
models.ServiceUsageTypes(
usage_type=self.service_usage_type1,
service=self.service,
start=datetime.date(2013, 10, 16),
end=datetime.date(2013, 10, 30),
percent=40,
).save()
models.ServiceUsageTypes(
usage_type=self.service_usage_type2,
service=self.service,
start=datetime.date(2013, 10, 16),
end=datetime.date(2013, 10, 30),
percent=60,
).save()
self.service.save()
# ventures
self.venture1 = models.Venture(
venture_id=1,
name='V1',
is_active=True,
service=self.service, # venture is providing service
)
self.venture1.save()
self.venture2 = models.Venture(
venture_id=2,
name='V2',
is_active=True,
service=self.service, # venture is providing service
)
self.venture2.save()
self.venture3 = models.Venture(venture_id=3, name='V3', is_active=True)
self.venture3.save()
self.venture4 = models.Venture(venture_id=4, name='V4', is_active=True)
self.venture4.save()
self.service_ventures = list(self.service.venture_set.all())
self.not_service_ventures = list(
models.Venture.objects.exclude(
id__in=[i.id for i in self.service_ventures]))
self.ventures = models.Venture.objects.all()
# daily usages of base type
# ut1:
# venture1: 10
# venture2: 20
# ut2:
# venture1: 20 (half in warehouse1, half in warehouse2)
# venture2: 40 (half in warehouse1, half in warehouse2)
start = datetime.date(2013, 10, 8)
end = datetime.date(2013, 10, 22)
base_usage_types = models.UsageType.objects.filter(type='BU')
for i, ut in enumerate(base_usage_types, start=1):
days = rrule.rrule(rrule.DAILY, dtstart=start, until=end)
for j, day in enumerate(days, start=1):
for k, venture in enumerate(self.ventures, start=1):
daily_usage = models.DailyUsage(
date=day,
pricing_venture=venture,
value=10 * i * k,
type=ut,
)
if ut.by_warehouse:
daily_usage.warehouse = (
self.warehouses[j % len(self.warehouses)])
daily_usage.save()
# usage prices
dates = [
(datetime.date(2013, 10, 5), datetime.date(2013, 10, 12)),
(datetime.date(2013, 10, 13), datetime.date(2013, 10, 17)),
(datetime.date(2013, 10, 18), datetime.date(2013, 10, 25)),
]
# (real cost/price, forecast cost/price)
ut_prices_costs = [
(self.usage_type, [(10, 50), (20, 60), (30, 70)]),
(
self.usage_type_cost_wh,
[
[(3600, 2400), (5400, 5400), (4800, 12000)], # warehouse1
[(3600, 5400), (3600, 1200), (7200, 3600)], # warehouse2
]),
]
def add_usage_price(usage_type, prices_costs, warehouse=None):
for daterange, price_cost in zip(dates, prices_costs):
start, end = daterange
usage_price = models.UsagePrice(
type=usage_type,
start=start,
end=end,
)
if warehouse is not None:
usage_price.warehouse = warehouse
if usage_type.by_cost:
usage_price.cost = price_cost[0]
usage_price.forecast_cost = price_cost[1]
else:
usage_price.price = price_cost[0]
usage_price.forecast_price = price_cost[1]
usage_price.save()
for ut, prices in ut_prices_costs:
if ut.by_warehouse:
for i, prices_wh in enumerate(prices):
warehouse = self.warehouses[i]
add_usage_price(ut, prices_wh, warehouse)
else:
add_usage_price(ut, prices)
# usage of service resources
start = datetime.date(2013, 10, 8)
end = datetime.date(2013, 10, 22)
service_usage_types = models.UsageType.objects.filter(type='SU')
for i, ut in enumerate(service_usage_types, start=1):
days = rrule.rrule(rrule.DAILY, dtstart=start, until=end)
for j, day in enumerate(days, start=1):
for k, venture in enumerate(self.not_service_ventures,
start=1):
daily_usage = models.DailyUsage(
date=day,
pricing_venture=venture,
value=10 * i * k,
type=ut,
)
daily_usage.save()
self.maxDiff = None
