1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
# Copyright (c) 2017-2018 VMware, 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.
import mock
import unittest
from rest_framework import status
from vio.pub.msapi import extsys
from vio.pub.vim.vimapi.nova import OperateHypervisor
from vio.pub.vim.vimapi.nova import OperateLimits
from vio.pub.vim.vimapi.nova import OperateNova
from vio.swagger.views.capacity.views import CapacityCheck
from cinderclient import client
VIM_INFO = {'vimId': 1, 'name': 'openstack_regionone', 'userName': 'user1',
'password': '1234', 'url': 'abc', 'tenant': 't1'}
class CapacityCheckTest(unittest.TestCase):
def setUp(self):
self.view = CapacityCheck()
@mock.patch.object(OperateNova, "OperateAZ")
@mock.patch.object(OperateHypervisor, "OperateHypervisor")
@mock.patch.object(OperateLimits, "OperateLimits")
@mock.patch.object(client, "Client")
@mock.patch.object(extsys, "get_vim_by_id")
def test_check_capacity_success(self, mock_get_vim, mock_cinder,
mock_limit, mock_hypervisor, mock_az):
mock_get_vim.return_value = VIM_INFO
req = mock.Mock()
req.body = """{
"vCPU": 1,
"Memory": 1,
"Storage": 500
}"""
oplimits = mock.Mock()
absolute = mock.Mock(
total_cores=20, total_cores_used=1,
total_ram=128*1024, total_ram_used=4*1024)
oplimits.get_limits.return_value = mock.Mock(absolute=absolute)
mock_limit.return_value = oplimits
climits = mock.Mock()
climits.to_dict.return_value = {
"absolute": {
"maxTotalVolumeGigabytes": 5000,
"totalGigabytesUsed": 100
}
}
cclient = mock.Mock()
cclient.limits.get.return_value = climits
mock_cinder.return_value = cclient
nazs = [mock.Mock(name="nova", hosts={"compute01": {}})]
nclient = mock.Mock()
nclient.list_availability_zones.return_value = nazs
mock_az.return_value = nclient
ophypervisor = mock.Mock()
ophypervisor.list_hypervisors.return_value = [
mock.Mock(id="compute01", status="enabled")]
hyper = mock.Mock()
hyper.to_dict.return_value = {
"service_details": {"host": "compute01"},
"vcpus": 20,
"vcpus_used": 1,
"memory_size": 128*1024,
"memory_used": 4*1024,
"local_disk_size": 5000,
"local_disk_used": 100
}
ophypervisor.get_hypervisor.return_value = hyper
mock_hypervisor.return_value = ophypervisor
resp = self.view.post(req, "openstack_regionone")
self.assertEqual(status.HTTP_200_OK, resp.status_code)
self.assertEqual({"result": True}, resp.data)
@mock.patch.object(OperateLimits, "OperateLimits")
@mock.patch.object(extsys, "get_vim_by_id")
def test_check_capacity_nova_limits_failed(
self, mock_get_vim, mock_limit):
mock_get_vim.return_value = VIM_INFO
req = mock.Mock()
req.body = """{
"vCPU": 1,
"Memory": 1,
"Storage": 500
}"""
oplimits = mock.Mock()
absolute = mock.Mock(
total_cores=1, total_cores_used=1,
total_ram=128*1024, total_ram_used=4*1024)
oplimits.get_limits.return_value = mock.Mock(absolute=absolute)
mock_limit.return_value = oplimits
resp = self.view.post(req, "openstack_regionone")
self.assertEqual(status.HTTP_200_OK, resp.status_code)
self.assertEqual({"result": False}, resp.data)
def test_check_capacity_invalid_input(self):
req = mock.Mock()
req.body = "hello world"
resp = self.view.post(req, "openstack_regionone")
self.assertEqual(status.HTTP_400_BAD_REQUEST, resp.status_code)
|
