8110c1b0c8
Supports iso9660 ConfigDriveV2 drives, which can be attached to the instance on boot as VHD or ISO. Admin password injection can be selectively enabled for security concerns. Updates also the hyperv tests. Change-Id: I22d1ab57bd2f75ddf8758b2ea1a958d7bbe35731
99 lines
3.2 KiB
Python
99 lines
3.2 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2012 Cloudbase Solutions Srl
|
|
#
|
|
# 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.
|
|
|
|
"""
|
|
TestCase for MockProxy based tests and related classes.
|
|
"""
|
|
|
|
import gzip
|
|
import os
|
|
import pickle
|
|
import sys
|
|
|
|
from nova import test
|
|
from nova.tests.hyperv import mockproxy
|
|
|
|
gen_test_mocks_key = 'NOVA_GENERATE_TEST_MOCKS'
|
|
|
|
|
|
class BaseTestCase(test.TestCase):
|
|
"""TestCase for MockProxy based tests."""
|
|
|
|
def run(self, result=None):
|
|
self._currentResult = result
|
|
super(BaseTestCase, self).run(result)
|
|
|
|
def setUp(self):
|
|
super(BaseTestCase, self).setUp()
|
|
self._mps = {}
|
|
|
|
def tearDown(self):
|
|
super(BaseTestCase, self).tearDown()
|
|
|
|
has_errors = len([test for (test, msgs) in self._currentResult.errors
|
|
if test.id() == self.id()]) > 0
|
|
failed = len([test for (test, msgs) in self._currentResult.failures
|
|
if test.id() == self.id()]) > 0
|
|
|
|
if not has_errors and not failed:
|
|
self._save_mock_proxies()
|
|
|
|
def _save_mock(self, name, mock):
|
|
path = self._get_stub_file_path(self.id(), name)
|
|
pickle.dump(mock, gzip.open(path, 'wb'))
|
|
|
|
def _get_stub_file_path(self, test_name, mock_name):
|
|
# test naming differs between platforms
|
|
prefix = 'nova.tests.'
|
|
if test_name.startswith(prefix):
|
|
test_name = test_name[len(prefix):]
|
|
file_name = '{0}_{1}.p.gz'.format(test_name, mock_name)
|
|
return os.path.join(os.path.dirname(mockproxy.__file__),
|
|
"stubs", file_name)
|
|
|
|
def _load_mock(self, name):
|
|
path = self._get_stub_file_path(self.id(), name)
|
|
if os.path.exists(path):
|
|
return pickle.load(gzip.open(path, 'rb'))
|
|
else:
|
|
return None
|
|
|
|
def _load_mock_or_create_proxy(self, module_name):
|
|
m = None
|
|
if not gen_test_mocks_key in os.environ or \
|
|
os.environ[gen_test_mocks_key].lower() \
|
|
not in ['true', 'yes', '1']:
|
|
m = self._load_mock(module_name)
|
|
else:
|
|
__import__(module_name)
|
|
module = sys.modules[module_name]
|
|
m = mockproxy.MockProxy(module)
|
|
self._mps[module_name] = m
|
|
return m
|
|
|
|
def _inject_mocks_in_modules(self, objects_to_mock, modules_to_test):
|
|
for module_name in objects_to_mock:
|
|
mp = self._load_mock_or_create_proxy(module_name)
|
|
for mt in modules_to_test:
|
|
module_local_name = module_name.split('.')[-1]
|
|
setattr(mt, module_local_name, mp)
|
|
|
|
def _save_mock_proxies(self):
|
|
for name, mp in self._mps.items():
|
|
m = mp.get_mock()
|
|
if m.has_values():
|
|
self._save_mock(name, m)
|