788388c486
Previously limit information was always retrieved but it is not used in the admin overview. This commit moves the limit related logic into a separate class ProjectUsageView. testing of usage.views is covered by individual panel tests, so UTs for panels which uses UsageView rather than the new ProjectUsageView need to be updated in this commit. blueprint make-quotas-great-again Change-Id: Id93b66d11e3b0c98d1fbb0454bb43984c2999d92
197 lines
8.4 KiB
Python
197 lines
8.4 KiB
Python
# Copyright 2012 United States Government as represented by the
|
|
# Administrator of the National Aeronautics and Space Administration.
|
|
# All Rights Reserved.
|
|
#
|
|
# Copyright 2012 Nebula, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import datetime
|
|
|
|
from django.core.urlresolvers import reverse
|
|
from django import http
|
|
from django.test.utils import override_settings
|
|
from django.utils import encoding
|
|
from django.utils import timezone
|
|
|
|
from mox3.mox import IsA
|
|
|
|
from horizon.templatetags import sizeformat
|
|
|
|
from openstack_dashboard import api
|
|
from openstack_dashboard.test import helpers as test
|
|
from openstack_dashboard import usage
|
|
|
|
|
|
INDEX_URL = reverse('horizon:project:overview:index')
|
|
|
|
|
|
class UsageViewTests(test.BaseAdminViewTests):
|
|
|
|
def _stub_api_calls(self, nova_stu_enabled):
|
|
self.mox.StubOutWithMock(api.nova, 'usage_list')
|
|
self.mox.StubOutWithMock(api.nova, 'extension_supported')
|
|
self.mox.StubOutWithMock(api.keystone, 'tenant_list')
|
|
|
|
api.nova.extension_supported(
|
|
'SimpleTenantUsage', IsA(http.HttpRequest)) \
|
|
.AndReturn(nova_stu_enabled)
|
|
|
|
@override_settings(OVERVIEW_DAYS_RANGE=None)
|
|
def test_usage(self):
|
|
self._test_usage(nova_stu_enabled=True, overview_days_range=None)
|
|
|
|
def test_usage_1_day(self):
|
|
self._test_usage(nova_stu_enabled=True)
|
|
|
|
@override_settings(OVERVIEW_DAYS_RANGE=None)
|
|
def test_usage_disabled(self):
|
|
self._test_usage(nova_stu_enabled=False, overview_days_range=None)
|
|
|
|
def test_usage_with_deleted_tenant(self):
|
|
self._test_usage(tenant_deleted=True)
|
|
|
|
def _get_start_end_range(self, overview_days_range):
|
|
now = timezone.now()
|
|
if overview_days_range:
|
|
start_day = now - datetime.timedelta(days=overview_days_range)
|
|
else:
|
|
start_day = datetime.date(now.year, now.month, 1)
|
|
return start_day, now
|
|
|
|
def _test_usage(self, nova_stu_enabled=True, tenant_deleted=False,
|
|
overview_days_range=1):
|
|
self._stub_api_calls(nova_stu_enabled)
|
|
api.nova.extension_supported(
|
|
'SimpleTenantUsage', IsA(http.HttpRequest)) \
|
|
.AndReturn(nova_stu_enabled)
|
|
usage_list = [api.nova.NovaUsage(u) for u in self.usages.list()]
|
|
if tenant_deleted:
|
|
api.keystone.tenant_list(IsA(http.HttpRequest)) \
|
|
.AndReturn([[self.tenants.first()], False])
|
|
else:
|
|
api.keystone.tenant_list(IsA(http.HttpRequest)) \
|
|
.AndReturn([self.tenants.list(), False])
|
|
|
|
if nova_stu_enabled:
|
|
start_day, now = self._get_start_end_range(overview_days_range)
|
|
api.nova.usage_list(IsA(http.HttpRequest),
|
|
datetime.datetime(start_day.year,
|
|
start_day.month,
|
|
start_day.day, 0, 0, 0, 0),
|
|
datetime.datetime(now.year,
|
|
now.month,
|
|
now.day, 23, 59, 59, 0)) \
|
|
.AndReturn(usage_list)
|
|
|
|
self.mox.ReplayAll()
|
|
res = self.client.get(reverse('horizon:admin:overview:index'))
|
|
self.assertTemplateUsed(res, 'admin/overview/usage.html')
|
|
self.assertIsInstance(res.context['usage'], usage.GlobalUsage)
|
|
self.assertEqual(nova_stu_enabled,
|
|
res.context['simple_tenant_usage_enabled'])
|
|
|
|
usage_table = encoding.smart_str(u'''
|
|
<tr class="" data-object-id="1" id="global_usage__row__1">
|
|
<td class="sortable normal_column">test_tenant</td>
|
|
<td class="sortable normal_column">%s</td>
|
|
<td class="sortable normal_column">%s</td>
|
|
<td class="sortable normal_column">%s</td>
|
|
<td class="sortable normal_column">%.2f</td>
|
|
<td class="sortable normal_column">%.2f</td>
|
|
<td class="sortable normal_column">%.2f</td>
|
|
</tr>
|
|
''' % (usage_list[0].vcpus,
|
|
sizeformat.diskgbformat(usage_list[0].local_gb),
|
|
sizeformat.mb_float_format(usage_list[0].memory_mb),
|
|
usage_list[0].vcpu_hours,
|
|
usage_list[0].disk_gb_hours,
|
|
usage_list[0].memory_mb_hours)
|
|
)
|
|
|
|
# test for deleted project
|
|
usage_table_deleted = encoding.smart_str(u'''
|
|
<tr class="" data-object-id="3" id="global_usage__row__3">
|
|
<td class="sortable normal_column">3 (Deleted)</td>
|
|
<td class="sortable normal_column">%s</td>
|
|
<td class="sortable normal_column">%s</td>
|
|
<td class="sortable normal_column">%s</td>
|
|
<td class="sortable normal_column">%.2f</td>
|
|
<td class="sortable normal_column">%.2f</td>
|
|
<td class="sortable normal_column">%.2f</td>
|
|
</tr>
|
|
''' % (usage_list[1].vcpus,
|
|
sizeformat.diskgbformat(usage_list[1].local_gb),
|
|
sizeformat.mb_float_format(usage_list[1].memory_mb),
|
|
usage_list[1].vcpu_hours,
|
|
usage_list[1].disk_gb_hours,
|
|
usage_list[1].memory_mb_hours)
|
|
)
|
|
|
|
if nova_stu_enabled:
|
|
self.assertContains(res, usage_table, html=True)
|
|
if tenant_deleted:
|
|
self.assertContains(res, usage_table_deleted, html=True)
|
|
else:
|
|
self.assertNotContains(res, usage_table_deleted, html=True)
|
|
else:
|
|
self.assertNotContains(res, usage_table, html=True)
|
|
|
|
@override_settings(OVERVIEW_DAYS_RANGE=None)
|
|
def test_usage_csv(self):
|
|
self._test_usage_csv(nova_stu_enabled=True, overview_days_range=None)
|
|
|
|
def test_usage_csv_1_day(self):
|
|
self._test_usage_csv(nova_stu_enabled=True)
|
|
|
|
@override_settings(OVERVIEW_DAYS_RANGE=None)
|
|
def test_usage_csv_disabled(self):
|
|
self._test_usage_csv(nova_stu_enabled=False, overview_days_range=None)
|
|
|
|
def _test_usage_csv(self, nova_stu_enabled=True, overview_days_range=1):
|
|
self._stub_api_calls(nova_stu_enabled)
|
|
api.nova.extension_supported(
|
|
'SimpleTenantUsage', IsA(http.HttpRequest)) \
|
|
.AndReturn(nova_stu_enabled)
|
|
usage_obj = [api.nova.NovaUsage(u) for u in self.usages.list()]
|
|
api.keystone.tenant_list(IsA(http.HttpRequest)) \
|
|
.AndReturn([self.tenants.list(), False])
|
|
if nova_stu_enabled:
|
|
start_day, now = self._get_start_end_range(overview_days_range)
|
|
api.nova.usage_list(IsA(http.HttpRequest),
|
|
datetime.datetime(start_day.year,
|
|
start_day.month,
|
|
start_day.day,
|
|
0, 0, 0, 0),
|
|
datetime.datetime(now.year,
|
|
now.month,
|
|
now.day, 23, 59, 59, 0)) \
|
|
.AndReturn(usage_obj)
|
|
self.mox.ReplayAll()
|
|
|
|
csv_url = reverse('horizon:admin:overview:index') + "?format=csv"
|
|
res = self.client.get(csv_url)
|
|
self.assertTemplateUsed(res, 'admin/overview/usage.csv')
|
|
self.assertIsInstance(res.context['usage'], usage.GlobalUsage)
|
|
hdr = 'Project Name,VCPUs,RAM (MB),Disk (GB),Usage (Hours)'
|
|
self.assertContains(res, '%s\r\n' % hdr)
|
|
|
|
if nova_stu_enabled:
|
|
for obj in usage_obj:
|
|
row = u'{0},{1},{2},{3},{4:.2f}\r\n'.format(obj.project_name,
|
|
obj.vcpus,
|
|
obj.memory_mb,
|
|
obj.disk_gb_hours,
|
|
obj.vcpu_hours)
|
|
self.assertContains(res, row)
|