def test_query_usage_report(self):
timestamp_from = '2012-03-24T16:44:21'
timestamp_to = '2012-03-24T16:46:21'
datetime_from = iso8601.parse_date(timestamp_from)
datetime_to = iso8601.parse_date(timestamp_to)
project_subscriptions = [
{
'id': self.subscription_id,
'resource_uuid': self.resource_uuid,
'resource_name': self.resource_name,
'product': {
'region': {
'name': self.region_name,
},
'item': {
'name': self.item_name,
},
'item_type': {
'name': self.item_type_name,
},
'order_unit': 'hours',
'order_size': 1,
'price': self.product_price,
'currency': 'CNY',
}
},
{
'id': 60,
'resource_uuid': 'a-fake-uuid-1',
'resource_name': 'a_fake_name_1',
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'instance',
},
'item_type': {
'name': 'm1.large',
},
'order_unit': 'months',
'order_size': 1,
'price': Decimal(2100.00).quantize(Decimal('0.01')),
'currency': 'CNY',
}
},
{
'id': 61,
'resource_uuid': 'a-fake-uuid-2',
'resource_name': '10.211.23.45',
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'floating_ip',
},
'item_type': {
'name': 'default',
},
'order_unit': 'days',
'order_size': 1,
'price': Decimal(1.10).quantize(Decimal('0.01')),
'currency': 'CNY',
}
},
{
'id': 62,
'resource_uuid': 'a-fake-uuid-3',
'resource_name': '170.1.223.5',
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'floating_ip',
},
'item_type': {
'name': 'default',
},
'order_unit': 'days',
'order_size': 1,
'price': Decimal(1.10).quantize(Decimal('0.01')),
'currency': 'CNY',
}
},
{
'id': 63,
'resource_uuid': 'a-fake-uuid-4',
'resource_name': 'a_fake_name_4',
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'load_balancer',
},
'item_type': {
'name': 'default',
},
'order_unit': 'days',
'order_size': 1,
'price': Decimal(2.70).quantize(Decimal('0.01')),
'currency': 'CNY',
}
},
{
'id': 64,
'resource_uuid': 'a-fake-uuid-5',
'resource_name': 'a_fake_name_5',
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'load_balancer',
},
'item_type': {
'name': 'default',
},
'order_unit': 'days',
'order_size': 1,
'price': Decimal(2.70).quantize(Decimal('0.01')),
'currency': 'CNY',
}
},
{
'id': 65,
'resource_uuid': self.resource_uuid,
'resource_name': '192.168.0.2',
'product': {
'region': {
'name': self.region_name,
},
'item': {
'name': 'network-out',
},
'item_type': {
'name': 'default',
},
'order_unit': 'KBytes',
'order_size': 1,
'price': Decimal(0.70).quantize(Decimal('0.01')),
'currency': 'CNY',
}
},
{
'id': 66,
'resource_uuid': 'a-fake-uuid-1',
'resource_name': '192.168.0.3',
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'network-out',
},
'item_type': {
'name': 'default',
},
'order_unit': 'KBytes',
'order_size': 1,
'price': Decimal(0.70).quantize(Decimal('0.01')),
'currency': 'CNY',
}
},
]
purchases1 = [
{
'quantity': 6,
'line_total': Decimal(14.40).quantize(Decimal('0.01')),
},
{
'quantity': 8,
'line_total': Decimal(19.20).quantize(Decimal('0.01')),
},
{
'quantity': 2,
'line_total': Decimal(4.80).quantize(Decimal('0.01')),
},
]
purchases2 = [
{
'quantity': 1,
'line_total': Decimal(2100.00).quantize(Decimal('0.01')),
},
]
purchases3 = [
{
'quantity': 6,
'line_total': Decimal(6.60).quantize(Decimal('0.01')),
},
{
'quantity': 8,
'line_total': Decimal(8.80).quantize(Decimal('0.01')),
},
{
'quantity': 5,
'line_total': Decimal(5.50).quantize(Decimal('0.01')),
},
]
purchases4 = [
{
'quantity': 6,
'line_total': Decimal(6.60).quantize(Decimal('0.01')),
},
{
'quantity': 3,
'line_total': Decimal(3.30).quantize(Decimal('0.01')),
},
{
'quantity': 2,
'line_total': Decimal(2.20).quantize(Decimal('0.01')),
},
]
purchases5 = [
{
'quantity': 6,
'line_total': Decimal(16.20).quantize(Decimal('0.01')),
},
{
'quantity': 3,
'line_total': Decimal(8.10).quantize(Decimal('0.01')),
},
{
'quantity': 4,
'line_total': Decimal(10.80).quantize(Decimal('0.01')),
},
]
purchases6 = [
{
'quantity': 6,
'line_total': Decimal(16.20).quantize(Decimal('0.01')),
},
{
'quantity': 8,
'line_total': Decimal(21.60).quantize(Decimal('0.01')),
},
{
'quantity': 13,
'line_total': Decimal(35.10).quantize(Decimal('0.01')),
},
]
purchases7 = [
{
'quantity': 1000,
'line_total': Decimal(700.00).quantize(Decimal('0.01')),
},
{
'quantity': 800,
'line_total': Decimal(560.00).quantize(Decimal('0.01')),
},
{
'quantity': 52,
'line_total': Decimal(36.40).quantize(Decimal('0.01')),
},
]
purchases8 = [
{
'quantity': 9000,
'line_total': Decimal(6300.00).quantize(Decimal('0.01')),
},
{
'quantity': 800,
'line_total': Decimal(560.00).quantize(Decimal('0.01')),
},
{
'quantity': 53,
'line_total': Decimal(37.10).quantize(Decimal('0.01')),
},
]
usage_report = {
'load_balancer': {
'default': [
('a-fake-uuid-4', 'a_fake_name_4', 'default', 'days',
2.70, 'CNY', 13, 35.10),
('a-fake-uuid-5', 'a_fake_name_5', 'default', 'days',
2.70, 'CNY', 27, 72.90),
]
},
'instance': {
'default': [
(self.resource_uuid, self.resource_name,
self.item_type_name, 'hours',
float(self.product_price), 'CNY', 16, 38.40),
('a-fake-uuid-1', 'a_fake_name_1', 'm1.large', 'months',
2100.00, 'CNY', 1, 2100.00),
],
},
'floating_ip': {
'default': [
('a-fake-uuid-2', '10.211.23.45', 'default', 'days',
1.10, 'CNY', 19, 20.90),
('a-fake-uuid-3', '170.1.223.5', 'default', 'days',
1.10, 'CNY', 11, 12.10),
],
},
'network-out': {
'default': [
(self.resource_uuid, '192.168.0.2', 'default', 'KBytes',
0.70, 'CNY', 1852, 1296.40),
('a-fake-uuid-1', '192.168.0.3', 'default', 'KBytes',
0.70, 'CNY', 9853, 6897.10),
]
},
}
self.mox.StubOutWithMock(db, 'subscription_get_all_by_project')
_purchase_func = 'purchase_get_all_by_subscription_and_timeframe'
self.mox.StubOutWithMock(db, _purchase_func)
db.subscription_get_all_by_project(self.context, self.tenant_id).\
AndReturn(project_subscriptions)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
self.subscription_id, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases1)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
60, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases2)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
61, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases3)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
62, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases4)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
63, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases5)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
64, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases6)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
65, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases7)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
66, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases8)
self.mox.ReplayAll()
result = api.query_usage_report(self.context,
timestamp_from,
timestamp_to)
self.mox.VerifyAll()
expect_keys = usage_report.keys().sort()
actual_keys = result['data'].keys().sort()
self.assertEqual(expect_keys, actual_keys)
for k, v in result['data'].iteritems():
self.assertEqual(v, usage_report[k])
def test_query_usage_report(self):
timestamp_from = '2012-03-24T16:44:21+00:00'
timestamp_to = '2012-03-24T16:46:21+00:00'
datetime_from = iso8601.parse_date(timestamp_from)
datetime_to = iso8601.parse_date(timestamp_to)
project_subscriptions = [
{
'id': self.subscription_id,
'resource_uuid': self.resource_uuid,
'resource_name': self.resource_name,
'created_at': datetime_from,
'expires_at': datetime_to,
'product': {
'region': {
'name': self.region_name,
},
'item': {
'name': self.item_name,
},
'item_type': {
'name': self.item_type_name,
},
'order_unit': 'hours',
'order_size': 1,
'price': self.product_price,
'currency': 'CNY',
}
},
{
'id': 60,
'resource_uuid': 'a-fake-uuid-1',
'resource_name': 'a_fake_name_1',
'created_at': datetime_from,
'expires_at': datetime_to,
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'instance',
},
'item_type': {
'name': 'm1.large',
},
'order_unit': 'months',
'order_size': 1,
'price': 2100.00,
'currency': 'CNY',
}
},
{
'id': 61,
'resource_uuid': 'a-fake-uuid-2',
'resource_name': '10.211.23.45',
'created_at': datetime_from,
'expires_at': datetime_to,
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'floating_ip',
},
'item_type': {
'name': 'default',
},
'order_unit': 'days',
'order_size': 1,
'price': 1.10,
'currency': 'CNY',
}
},
{
'id': 62,
'resource_uuid': 'a-fake-uuid-3',
'resource_name': '170.1.223.5',
'created_at': datetime_from,
'expires_at': datetime_to,
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'floating_ip',
},
'item_type': {
'name': 'default',
},
'order_unit': 'days',
'order_size': 1,
'price': 1.10,
'currency': 'CNY',
}
},
{
'id': 63,
'resource_uuid': 'a-fake-uuid-4',
'resource_name': 'a_fake_name_4',
'created_at': datetime_from,
'expires_at': datetime_to,
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'load_balancer',
},
'item_type': {
'name': 'default',
},
'order_unit': 'days',
'order_size': 1,
'price': 2.70,
'currency': 'CNY',
}
},
{
'id': 64,
'resource_uuid': 'a-fake-uuid-5',
'resource_name': 'a_fake_name_5',
'created_at': datetime_from,
'expires_at': datetime_to,
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'load_balancer',
},
'item_type': {
'name': 'default',
},
'order_unit': 'days',
'order_size': 1,
'price': 2.70,
'currency': 'CNY',
}
},
{
'id': 65,
'resource_uuid': self.resource_uuid,
'resource_name': '192.168.0.2',
'created_at': datetime_from,
'expires_at': datetime_to,
'product': {
'region': {
'name': self.region_name,
},
'item': {
'name': 'network',
},
'item_type': {
'name': 'default',
},
'order_unit': 'KBytes',
'order_size': 1,
'price': 0.70,
'currency': 'CNY',
}
},
{
'id': 66,
'resource_uuid': 'a-fake-uuid-1',
'resource_name': '192.168.0.3',
'created_at': datetime_from,
'expires_at': datetime_to,
'product': {
'region': {
'name': 'default',
},
'item': {
'name': 'network',
},
'item_type': {
'name': 'default',
},
'order_unit': 'KBytes',
'order_size': 1,
'price': 0.70,
'currency': 'CNY',
}
},
]
purchases1 = [
{
'quantity': 6,
'line_total': 14.40,
},
{
'quantity': 8,
'line_total': 19.20,
},
{
'quantity': 2,
'line_total': 4.80,
},
]
purchases2 = [
{
'quantity': 1,
'line_total': 2100.00,
},
]
purchases3 = [
{
'quantity': 6,
'line_total': 6.60,
},
{
'quantity': 8,
'line_total': 8.80,
},
{
'quantity': 5,
'line_total': 5.50,
},
]
purchases4 = [
{
'quantity': 6,
'line_total': 6.60,
},
{
'quantity': 3,
'line_total': 3.30,
},
{
'quantity': 2,
'line_total': 2.20,
},
]
purchases5 = [
{
'quantity': 6,
'line_total': 16.20,
},
{
'quantity': 3,
'line_total': 8.10,
},
{
'quantity': 4,
'line_total': 10.80,
},
]
purchases6 = [
{
'quantity': 6,
'line_total': 16.20,
},
{
'quantity': 8,
'line_total': 21.60,
},
{
'quantity': 13,
'line_total': 35.10,
},
]
purchases7 = [
{
'quantity': 1000,
'line_total': 700.00,
},
{
'quantity': 800,
'line_total': 560.00,
},
{
'quantity': 52,
'line_total': 36.40,
},
]
purchases8 = [
{
'quantity': 9000,
'line_total': 6300.00,
},
{
'quantity': 800,
'line_total': 560.00,
},
{
'quantity': 53,
'line_total': 37.10,
},
]
usage_report = {
'default': {
'load_balancer': [
('a-fake-uuid-4', 'a_fake_name_4', 'default', 'days',
2.70, 'CNY', 13, 35.10, timestamp_from, timestamp_to),
('a-fake-uuid-5', 'a_fake_name_5', 'default', 'days',
2.70, 'CNY', 27, 72.90, timestamp_from, timestamp_to),
],
'instance': [
(self.resource_uuid, self.resource_name,
self.item_type_name, 'hours',
self.product_price, 'CNY', 16, 38.40,
timestamp_from, timestamp_to),
('a-fake-uuid-1', 'a_fake_name_1', 'm1.large', 'months',
2100.00, 'CNY', 1, 2100.00, timestamp_from, timestamp_to),
],
'floating_ip': [
('a-fake-uuid-2', '10.211.23.45', 'default', 'days',
1.10, 'CNY', 19, 20.90, timestamp_from, timestamp_to),
('a-fake-uuid-3', '170.1.223.5', 'default', 'days',
1.10, 'CNY', 11, 12.10, timestamp_from, timestamp_to),
],
'network': [
(self.resource_uuid, '192.168.0.2', 'default', 'KBytes',
0.70, 'CNY', 1852, 1296.40, timestamp_from, timestamp_to),
('a-fake-uuid-1', '192.168.0.3', 'default', 'KBytes',
0.70, 'CNY', 9853, 6897.10, timestamp_from, timestamp_to),
],
},
}
self.mox.StubOutWithMock(db, 'subscription_get_all_by_project')
_purchase_func = 'purchase_get_all_by_subscription_and_timeframe'
self.mox.StubOutWithMock(db, _purchase_func)
db.subscription_get_all_by_project(self.context, self.tenant_id).\
AndReturn(project_subscriptions)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
self.subscription_id, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases1)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
60, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases2)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
61, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases3)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
62, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases4)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
63, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases5)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
64, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases6)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
65, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases7)
db.purchase_get_all_by_subscription_and_timeframe(self.context,
66, datetime_from, datetime_to).\
InAnyOrder().AndReturn(purchases8)
self.mox.ReplayAll()
result = api.query_usage_report(self.context,
timestamp_from,
timestamp_to)
self.mox.VerifyAll()
expect_keys = usage_report.keys().sort()
actual_keys = result['data'].keys().sort()
self.assertEqual(expect_keys, actual_keys)
for k, v in result['data'].iteritems():
new_report = dict()
for region_name, usage_data in v.iteritems():
new_data = new_report.setdefault(region_name, list())
for a, b, c, d, e, f, g, h, i, j in usage_data:
datum = (a, b, c, d, e, f, g,
float(Decimal(h).quantize(Decimal('0.01'))), i, j)
new_data.append(datum)
self.assertEqual(new_report, usage_report[k])