4836eecd60
This change rewrites FakePopen to permit registering commands and synthetic output such that client code can call subprocess.Popen() and receive required output without actually executing commands. With this change, it is possible to mock most of the output require during a packstack run. Change-Id: If81f71cfe89a61e25fb62c97ed5b3759039a0bfb
95 lines
3.7 KiB
Python
95 lines
3.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2013, Red Hat, 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 os
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
from unittest import TestCase
|
|
|
|
from packstack.modules import ospluginutils, puppet
|
|
from packstack.installer import run_setup, basedefs
|
|
from packstack.installer.utils import shell
|
|
|
|
from ..test_base import PackstackTestCaseMixin, FakePopen
|
|
|
|
|
|
class CommandLineTestCase(PackstackTestCaseMixin, TestCase):
|
|
def test_running_install_hosts(self):
|
|
"""
|
|
Test packstack.installer.run_setup.main
|
|
|
|
This test effectivly runs all of the python code ran by
|
|
packstack --install-hosts=127.0.0.1 --os-swift-install=y \
|
|
--nagios-install=y
|
|
|
|
It is a fairly wide net but boost code coverage of the packstack
|
|
python code to about 85%, more finer grained tests should also be
|
|
Added to target speficic test cases.
|
|
|
|
Popen is replaced in PackstackTestCaseMixin so no actual commands get
|
|
run on the host running the unit tests
|
|
"""
|
|
# we need following to pass manage_epel(enabled=1) and
|
|
# manage_rdo(havana-6.noarch\nenabled=0) functions
|
|
subprocess.Popen = FakePopen
|
|
FakePopen.register('cat /etc/resolv.conf | grep nameserver',
|
|
stdout='nameserver 127.0.0.1')
|
|
FakePopen.register("rpm -q rdo-release "
|
|
"--qf='%{version}-%{release}.%{arch}\n'",
|
|
stdout='icehouse-2.noarch\n')
|
|
FakePopen.register_as_script('yum-config-manager --enable '
|
|
'openstack-icehouse',
|
|
stdout='[openstack-icehouse]\nenabled=1')
|
|
|
|
# create a dummy public key
|
|
dummy_public_key = os.path.join(self.tempdir, 'id_rsa.pub')
|
|
with open(dummy_public_key, 'w') as dummy:
|
|
dummy.write('ssh-rsa AAAAblablabla')
|
|
|
|
# Save sys.argv and replace it with the args we want optparse to use
|
|
orig_argv = sys.argv
|
|
sys.argv = ['packstack', '--debug',
|
|
'--ssh-public-key=%s' % dummy_public_key,
|
|
'--install-hosts=127.0.0.1', '--os-swift-install=y',
|
|
'--nagios-install=y', '--use-epel=y']
|
|
|
|
# There is no puppet logfile to validate, so replace
|
|
# ospluginutils.validate_puppet_logfile with a mock function
|
|
orig_validate_logfile = puppet.validate_logfile
|
|
puppet.validate_logfile = lambda a: None
|
|
puppet.scan_logfile = lambda a: []
|
|
|
|
# If there is a error in a plugin sys.exit() gets called, this masks
|
|
# the actual error that should be reported, so we replace it to
|
|
# raise Exception, packstack logging gives a more infomrative error
|
|
def raise_(ex):
|
|
raise ex
|
|
orig_sys_exit = sys.exit
|
|
sys.exit = lambda a: raise_(Exception('Error during install-hosts'))
|
|
|
|
try:
|
|
run_setup.main()
|
|
finally:
|
|
sys.argv = orig_argv
|
|
ospluginutils.validate_puppet_logfile = orig_validate_logfile
|
|
sys.exit = orig_sys_exit
|
|
try:
|
|
shutil.rmtree(basedefs.VAR_DIR)
|
|
except:
|
|
pass
|