0d81e1b86a
Change-Id: Ie5d20b2f44e099fe46a1c59a7e823a3b115aab34 Partially Implements: blueprint climate-dashboard Depends-On: I8065118fca5588e89888ef11888696886573d379
286 lines
10 KiB
Python
286 lines
10 KiB
Python
# 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.
|
|
|
|
from datetime import datetime
|
|
|
|
from django.core.urlresolvers import reverse
|
|
from django import http
|
|
from mox3.mox import IsA
|
|
import pytz
|
|
|
|
from blazar_dashboard import api
|
|
from blazar_dashboard.test import helpers as test
|
|
|
|
import logging
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
INDEX_TEMPLATE = 'project/leases/index.html'
|
|
INDEX_URL = reverse('horizon:project:leases:index')
|
|
DETAIL_TEMPLATE = 'project/leases/detail.html'
|
|
DETAIL_URL_BASE = 'horizon:project:leases:detail'
|
|
CREATE_URL = reverse('horizon:project:leases:create')
|
|
CREATE_TEMPLATE = 'project/leases/create.html'
|
|
UPDATE_URL_BASE = 'horizon:project:leases:update'
|
|
UPDATE_TEMPLATE = 'project/leases/update.html'
|
|
|
|
|
|
class LeasesTests(test.TestCase):
|
|
@test.create_stubs({api.client: ('lease_list',)})
|
|
def test_index(self):
|
|
leases = self.leases.list()
|
|
api.client.lease_list(IsA(http.HttpRequest)).AndReturn(leases)
|
|
self.mox.ReplayAll()
|
|
|
|
res = self.client.get(INDEX_URL)
|
|
self.assertTemplateUsed(res, INDEX_TEMPLATE)
|
|
self.assertNoMessages(res)
|
|
self.assertContains(res, 'lease-2')
|
|
self.assertContains(res, 'lease-1')
|
|
|
|
@test.create_stubs({api.client: ('lease_list',)})
|
|
def test_index_no_leases(self):
|
|
api.client.lease_list(IsA(http.HttpRequest)).AndReturn(())
|
|
self.mox.ReplayAll()
|
|
|
|
res = self.client.get(INDEX_URL)
|
|
self.assertTemplateUsed(res, INDEX_TEMPLATE)
|
|
self.assertNoMessages(res)
|
|
self.assertContains(res, 'No items to display')
|
|
|
|
@test.create_stubs({api.client: ('lease_list',)})
|
|
def test_index_error(self):
|
|
api.client.lease_list(
|
|
IsA(http.HttpRequest)
|
|
).AndRaise(self.exceptions.blazar)
|
|
self.mox.ReplayAll()
|
|
|
|
res = self.client.get(INDEX_URL)
|
|
self.assertTemplateUsed(res, INDEX_TEMPLATE)
|
|
self.assertMessageCount(res, error=1)
|
|
|
|
def test_lease_actions(self):
|
|
pass
|
|
|
|
@test.create_stubs({api.client: ('lease_get',)})
|
|
def test_lease_detail(self):
|
|
lease = self.leases.get(name='lease-1')
|
|
api.client.lease_get(IsA(http.HttpRequest),
|
|
lease['id']).AndReturn(lease)
|
|
self.mox.ReplayAll()
|
|
|
|
res = self.client.get(reverse(DETAIL_URL_BASE, args=[lease['id']]))
|
|
self.assertTemplateUsed(res, DETAIL_TEMPLATE)
|
|
self.assertContains(res, 'lease-1')
|
|
|
|
@test.create_stubs({api.client: ('lease_get',)})
|
|
def test_lease_detail_error(self):
|
|
api.client.lease_get(IsA(http.HttpRequest),
|
|
'invalid').AndRaise(self.exceptions.blazar)
|
|
self.mox.ReplayAll()
|
|
|
|
res = self.client.get(reverse(DETAIL_URL_BASE, args=['invalid']))
|
|
self.assertTemplateNotUsed(res, DETAIL_TEMPLATE)
|
|
self.assertMessageCount(error=1)
|
|
self.assertRedirectsNoFollow(res, INDEX_URL)
|
|
|
|
@test.create_stubs({api.client: ('lease_create', )})
|
|
def test_create_lease_host_reservation(self):
|
|
start_date = datetime(2030, 6, 27, 18, 0, tzinfo=pytz.utc)
|
|
end_date = datetime(2030, 6, 30, 18, 0, tzinfo=pytz.utc)
|
|
new_lease = self.leases.get(name='lease-1')
|
|
api.client.lease_create(
|
|
IsA(http.HttpRequest),
|
|
'lease-1',
|
|
start_date.strftime('%Y-%m-%d %H:%M'),
|
|
end_date.strftime('%Y-%m-%d %H:%M'),
|
|
[
|
|
{
|
|
'min': 1,
|
|
'max': 1,
|
|
'hypervisor_properties': '[">=", "$vcpus", "2"]',
|
|
'resource_properties': '',
|
|
'resource_type': 'physical:host',
|
|
}
|
|
],
|
|
[]
|
|
).AndReturn(new_lease)
|
|
self.mox.ReplayAll()
|
|
form_data = {
|
|
'name': 'lease-1',
|
|
'start_date': start_date.strftime('%Y-%m-%d %H:%M'),
|
|
'end_date': end_date.strftime('%Y-%m-%d %H:%M'),
|
|
'resource_type': 'host',
|
|
'min_hosts': 1,
|
|
'max_hosts': 1,
|
|
'hypervisor_properties': '[">=", "$vcpus", "2"]'
|
|
}
|
|
|
|
res = self.client.post(CREATE_URL, form_data)
|
|
self.assertNoFormErrors(res)
|
|
self.assertMessageCount(success=1)
|
|
self.assertRedirectsNoFollow(res, INDEX_URL)
|
|
|
|
@test.create_stubs({api.client: ('lease_create', )})
|
|
def test_create_lease_instance_reservation(self):
|
|
start_date = datetime(2030, 6, 27, 18, 0, tzinfo=pytz.utc)
|
|
end_date = datetime(2030, 6, 30, 18, 0, tzinfo=pytz.utc)
|
|
dummy_lease = {}
|
|
api.client.lease_create(
|
|
IsA(http.HttpRequest),
|
|
'lease-1',
|
|
start_date.strftime('%Y-%m-%d %H:%M'),
|
|
end_date.strftime('%Y-%m-%d %H:%M'),
|
|
[
|
|
{
|
|
'resource_type': 'virtual:instance',
|
|
'amount': 3,
|
|
'vcpus': 2,
|
|
'memory_mb': 4096,
|
|
'disk_gb': 128,
|
|
'affinity': False
|
|
}
|
|
],
|
|
[]
|
|
).AndReturn(dummy_lease)
|
|
self.mox.ReplayAll()
|
|
form_data = {
|
|
'name': 'lease-1',
|
|
'start_date': start_date.strftime('%Y-%m-%d %H:%M'),
|
|
'end_date': end_date.strftime('%Y-%m-%d %H:%M'),
|
|
'resource_type': 'instance',
|
|
'amount': 3,
|
|
'vcpus': 2,
|
|
'memory_mb': 4096,
|
|
'disk_gb': 128
|
|
}
|
|
|
|
res = self.client.post(CREATE_URL, form_data)
|
|
self.assertNoFormErrors(res)
|
|
self.assertMessageCount(success=1)
|
|
self.assertRedirectsNoFollow(res, INDEX_URL)
|
|
|
|
@test.create_stubs({api.client: ('lease_create', )})
|
|
def test_create_lease_client_error(self):
|
|
start_date = datetime(2030, 6, 27, 18, 0, tzinfo=pytz.utc)
|
|
end_date = datetime(2030, 6, 30, 18, 0, tzinfo=pytz.utc)
|
|
api.client.lease_create(
|
|
IsA(http.HttpRequest),
|
|
'lease-1',
|
|
start_date.strftime('%Y-%m-%d %H:%M'),
|
|
end_date.strftime('%Y-%m-%d %H:%M'),
|
|
[
|
|
{
|
|
'min': 1,
|
|
'max': 1,
|
|
'hypervisor_properties': '',
|
|
'resource_properties': '',
|
|
'resource_type': 'physical:host',
|
|
}
|
|
],
|
|
[]
|
|
).AndRaise(self.exceptions.blazar)
|
|
self.mox.ReplayAll()
|
|
form_data = {
|
|
'name': 'lease-1',
|
|
'start_date': start_date.strftime('%Y-%m-%d %H:%M'),
|
|
'end_date': end_date.strftime('%Y-%m-%d %H:%M'),
|
|
'resource_type': 'host',
|
|
'min_hosts': 1,
|
|
'max_hosts': 1,
|
|
}
|
|
|
|
res = self.client.post(CREATE_URL, form_data)
|
|
self.assertTemplateUsed(res, CREATE_TEMPLATE)
|
|
self.assertNoFormErrors(res)
|
|
self.assertContains(res, 'An error occurred while creating')
|
|
|
|
@test.create_stubs({api.client: ('lease_get', 'lease_update')})
|
|
def test_update_lease(self):
|
|
lease = self.leases.get(name='lease-1')
|
|
api.client.lease_get(
|
|
IsA(http.HttpRequest),
|
|
lease['id']
|
|
).AndReturn(lease)
|
|
api.client.lease_update(
|
|
IsA(http.HttpRequest),
|
|
lease_id=lease['id'],
|
|
name='newname',
|
|
prolong_for='1h'
|
|
)
|
|
form_data = {
|
|
'lease_id': lease['id'],
|
|
'lease_name': 'newname',
|
|
'end_time': '+1h'
|
|
}
|
|
self.mox.ReplayAll()
|
|
|
|
res = self.client.post(reverse(UPDATE_URL_BASE, args=[lease['id']]),
|
|
form_data)
|
|
self.assertNoFormErrors(res)
|
|
self.assertMessageCount(success=1)
|
|
self.assertRedirectsNoFollow(res, INDEX_URL)
|
|
|
|
@test.create_stubs({api.client: ('lease_get', 'lease_update')})
|
|
def test_update_lease_error(self):
|
|
lease = self.leases.get(name='lease-1')
|
|
api.client.lease_get(
|
|
IsA(http.HttpRequest),
|
|
lease['id']
|
|
).AndReturn(lease)
|
|
api.client.lease_update(
|
|
IsA(http.HttpRequest),
|
|
lease_id=lease['id'],
|
|
name='newname',
|
|
prolong_for='1h'
|
|
).AndRaise(self.exceptions.blazar)
|
|
form_data = {
|
|
'lease_id': lease['id'],
|
|
'lease_name': 'newname',
|
|
'end_time': '+1h'
|
|
}
|
|
self.mox.ReplayAll()
|
|
|
|
res = self.client.post(reverse(UPDATE_URL_BASE, args=[lease['id']]),
|
|
form_data)
|
|
self.assertTemplateUsed(UPDATE_TEMPLATE)
|
|
self.assertNoFormErrors(res)
|
|
self.assertContains(res, 'An error occurred while updating')
|
|
|
|
@test.create_stubs({api.client: ('lease_list', 'lease_delete')})
|
|
def test_delete_lease(self):
|
|
leases = self.leases.list()
|
|
lease = self.leases.get(name='lease-1')
|
|
api.client.lease_list(IsA(http.HttpRequest)).AndReturn(leases)
|
|
api.client.lease_delete(IsA(http.HttpRequest), lease['id'])
|
|
self.mox.ReplayAll()
|
|
|
|
action = 'leases__delete__%s' % lease['id']
|
|
form_data = {'action': action}
|
|
res = self.client.post(INDEX_URL, form_data)
|
|
self.assertMessageCount(success=1)
|
|
self.assertRedirectsNoFollow(res, INDEX_URL)
|
|
|
|
@test.create_stubs({api.client: ('lease_list', 'lease_delete')})
|
|
def test_delete_lease_error(self):
|
|
leases = self.leases.list()
|
|
lease = self.leases.get(name='lease-1')
|
|
api.client.lease_list(IsA(http.HttpRequest)).AndReturn(leases)
|
|
api.client.lease_delete(IsA(http.HttpRequest),
|
|
lease['id']).AndRaise(self.exceptions.blazar)
|
|
self.mox.ReplayAll()
|
|
|
|
action = 'leases__delete__%s' % lease['id']
|
|
form_data = {'action': action}
|
|
res = self.client.post(INDEX_URL, form_data)
|
|
self.assertMessageCount(error=1)
|
|
self.assertRedirectsNoFollow(res, INDEX_URL)
|