def test_schema_with_warehouse(self):
result = UsagePlugin(usage_type=self.usage_type_cost_wh, type='schema')
self.assertEquals(
result,
OrderedDict([
('ut_2_count_wh_1', {
'name': _('UsageType2 count (Warehouse1)'),
'rounding': 2,
'divide_by': 0,
}),
('ut_2_cost_wh_1', {
'name': _('UsageType2 cost (Warehouse1)'),
'currency': True,
}),
('ut_2_count_wh_2', {
'name': _('UsageType2 count (Warehouse2)'),
'rounding': 2,
'divide_by': 0,
}),
('ut_2_cost_wh_2', {
'name': _('UsageType2 cost (Warehouse2)'),
'currency': True,
}),
('ut_2_total_cost', {
'name': _('UsageType2 total cost'),
'currency': True,
'total_cost': True,
}),
]))
def test_schema(self):
result = UsagePlugin(usage_type=self.usage_type, type='schema')
self.assertEquals(
result,
OrderedDict([
('ut_1_count', {
'name': _('UsageType1 count')
}),
('ut_1_cost', {
'name': _('UsageType1 cost'),
'currency': True,
'total_cost': True,
}),
]))
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 test_usages_schema(self):
result = UsagePlugin(usage_type=self.usage_type_cost_wh,
type='dailyusages_header')
self.assertEquals(result, self.usage_type_cost_wh.name)
