358 lines
14 KiB
Python
358 lines
14 KiB
Python
from mock import (
|
|
call,
|
|
patch,
|
|
MagicMock
|
|
)
|
|
|
|
from test_utils import (
|
|
CharmTestCase,
|
|
patch_open
|
|
)
|
|
from charmhelpers.contrib.openstack.ip import PUBLIC
|
|
|
|
dnsmock = MagicMock()
|
|
modules = {
|
|
'dns': dnsmock,
|
|
'dns.resolver': dnsmock,
|
|
}
|
|
module_patcher = patch.dict('sys.modules', modules)
|
|
module_patcher.start()
|
|
|
|
with patch('charmhelpers.fetch.apt_install'):
|
|
with patch('utils.register_configs'):
|
|
import hooks as ceph_hooks
|
|
|
|
TO_PATCH = [
|
|
'CONFIGS',
|
|
'add_source',
|
|
'apt_update',
|
|
'apt_install',
|
|
'apt_purge',
|
|
'config',
|
|
'cmp_pkgrevno',
|
|
'execd_preinstall',
|
|
'enable_pocket',
|
|
'get_iface_for_address',
|
|
'get_netmask_for_address',
|
|
'glob',
|
|
'is_apache_24',
|
|
'log',
|
|
'lsb_release',
|
|
'open_port',
|
|
'os',
|
|
'relation_ids',
|
|
'relation_set',
|
|
'relation_get',
|
|
'related_units',
|
|
'render_template',
|
|
'shutil',
|
|
'status_set',
|
|
'subprocess',
|
|
'sys',
|
|
'unit_get',
|
|
]
|
|
|
|
|
|
class CephRadosGWTests(CharmTestCase):
|
|
|
|
def setUp(self):
|
|
super(CephRadosGWTests, self).setUp(ceph_hooks, TO_PATCH)
|
|
self.config.side_effect = self.test_config.get
|
|
self.test_config.set('source', 'distro')
|
|
self.test_config.set('key', 'secretkey')
|
|
self.test_config.set('use-syslog', False)
|
|
|
|
def test_install_www_scripts(self):
|
|
self.glob.glob.return_value = ['files/www/bob']
|
|
ceph_hooks.install_www_scripts()
|
|
self.shutil.copy.assert_called_with('files/www/bob', '/var/www/')
|
|
|
|
def test_install_ceph_optimised_packages(self):
|
|
self.lsb_release.return_value = {'DISTRIB_CODENAME': 'vivid'}
|
|
fastcgi_source = (
|
|
'http://gitbuilder.ceph.com/'
|
|
'libapache-mod-fastcgi-deb-vivid-x86_64-basic/ref/master')
|
|
apache_source = (
|
|
'http://gitbuilder.ceph.com/'
|
|
'apache2-deb-vivid-x86_64-basic/ref/master')
|
|
calls = [
|
|
call(fastcgi_source, key='6EAEAE2203C3951A'),
|
|
call(apache_source, key='6EAEAE2203C3951A'),
|
|
]
|
|
ceph_hooks.install_ceph_optimised_packages()
|
|
self.add_source.assert_has_calls(calls)
|
|
|
|
def test_install_packages(self):
|
|
self.test_config.set('use-ceph-optimised-packages', '')
|
|
ceph_hooks.install_packages()
|
|
self.add_source.assert_called_with('distro', 'secretkey')
|
|
self.assertTrue(self.apt_update.called)
|
|
self.apt_install.assert_called_with(['libapache2-mod-fastcgi',
|
|
'apache2'], fatal=True)
|
|
|
|
def test_install_optimised_packages_no_embedded(self):
|
|
self.test_config.set('use-ceph-optimised-packages', True)
|
|
self.test_config.set('use-embedded-webserver', False)
|
|
_install_packages = self.patch('install_ceph_optimised_packages')
|
|
ceph_hooks.install_packages()
|
|
self.add_source.assert_called_with('distro', 'secretkey')
|
|
self.assertTrue(self.apt_update.called)
|
|
self.assertTrue(_install_packages.called)
|
|
self.apt_install.assert_called_with(['libapache2-mod-fastcgi',
|
|
'apache2'], fatal=True)
|
|
|
|
def test_install_optimised_packages_embedded(self):
|
|
self.test_config.set('use-ceph-optimised-packages', True)
|
|
self.test_config.set('use-embedded-webserver', True)
|
|
_install_packages = self.patch('install_ceph_optimised_packages')
|
|
ceph_hooks.install_packages()
|
|
self.add_source.assert_called_with('distro', 'secretkey')
|
|
self.assertTrue(self.apt_update.called)
|
|
self.assertFalse(_install_packages.called)
|
|
self.apt_install.assert_called_with(ceph_hooks.PACKAGES,
|
|
fatal=True)
|
|
self.apt_purge.assert_called_with(['libapache2-mod-fastcgi',
|
|
'apache2'])
|
|
|
|
def test_install(self):
|
|
_install_packages = self.patch('install_packages')
|
|
ceph_hooks.install()
|
|
self.assertTrue(self.execd_preinstall.called)
|
|
self.assertTrue(_install_packages.called)
|
|
self.enable_pocket.assert_called_with('multiverse')
|
|
self.os.makedirs.called_with('/var/lib/ceph/nss')
|
|
|
|
def test_emit_apacheconf(self):
|
|
self.is_apache_24.return_value = True
|
|
self.unit_get.return_value = '10.0.0.1'
|
|
apachecontext = {
|
|
"hostname": '10.0.0.1',
|
|
"port": 70,
|
|
}
|
|
vhost_file = '/etc/apache2/sites-available/rgw.conf'
|
|
with patch_open() as (_open, _file):
|
|
ceph_hooks.emit_apacheconf()
|
|
_open.assert_called_with(vhost_file, 'w')
|
|
self.render_template.assert_called_with('rgw', apachecontext)
|
|
|
|
def test_apache_sites24(self):
|
|
self.is_apache_24.return_value = True
|
|
ceph_hooks.apache_sites()
|
|
calls = [
|
|
call(['a2dissite', '000-default']),
|
|
call(['a2ensite', 'rgw']),
|
|
]
|
|
self.subprocess.check_call.assert_has_calls(calls)
|
|
|
|
def test_apache_sites22(self):
|
|
self.is_apache_24.return_value = False
|
|
ceph_hooks.apache_sites()
|
|
calls = [
|
|
call(['a2dissite', 'default']),
|
|
call(['a2ensite', 'rgw']),
|
|
]
|
|
self.subprocess.check_call.assert_has_calls(calls)
|
|
|
|
def test_apache_modules(self):
|
|
ceph_hooks.apache_modules()
|
|
calls = [
|
|
call(['a2enmod', 'fastcgi']),
|
|
call(['a2enmod', 'rewrite']),
|
|
]
|
|
self.subprocess.check_call.assert_has_calls(calls)
|
|
|
|
def test_apache_reload(self):
|
|
ceph_hooks.apache_reload()
|
|
calls = [
|
|
call(['service', 'apache2', 'reload']),
|
|
]
|
|
self.subprocess.call.assert_has_calls(calls)
|
|
|
|
@patch.object(ceph_hooks, 'apache_ports', lambda *args: True)
|
|
@patch.object(ceph_hooks, 'mkdir', lambda *args: None)
|
|
def test_config_changed(self):
|
|
_install_packages = self.patch('install_packages')
|
|
_emit_apacheconf = self.patch('emit_apacheconf')
|
|
_install_www_scripts = self.patch('install_www_scripts')
|
|
_apache_sites = self.patch('apache_sites')
|
|
_apache_modules = self.patch('apache_modules')
|
|
_apache_reload = self.patch('apache_reload')
|
|
ceph_hooks.config_changed()
|
|
self.assertTrue(_install_packages.called)
|
|
self.CONFIGS.write_all.assert_called_with()
|
|
self.assertTrue(_emit_apacheconf.called)
|
|
self.assertTrue(_install_www_scripts.called)
|
|
self.assertTrue(_apache_sites.called)
|
|
self.assertTrue(_apache_modules.called)
|
|
self.assertTrue(_apache_reload.called)
|
|
|
|
@patch.object(ceph_hooks, 'is_request_complete',
|
|
lambda *args, **kwargs: True)
|
|
def test_mon_relation(self):
|
|
_ceph = self.patch('ceph')
|
|
_restart = self.patch('restart')
|
|
self.relation_get.return_value = 'seckey'
|
|
ceph_hooks.mon_relation()
|
|
self.assertTrue(_restart.called)
|
|
_ceph.import_radosgw_key.assert_called_with('seckey')
|
|
self.CONFIGS.write_all.assert_called_with()
|
|
|
|
@patch.object(ceph_hooks, 'is_request_complete',
|
|
lambda *args, **kwargs: True)
|
|
def test_mon_relation_nokey(self):
|
|
_ceph = self.patch('ceph')
|
|
_restart = self.patch('restart')
|
|
self.relation_get.return_value = None
|
|
ceph_hooks.mon_relation()
|
|
self.assertFalse(_ceph.import_radosgw_key.called)
|
|
self.assertFalse(_restart.called)
|
|
self.CONFIGS.write_all.assert_called_with()
|
|
|
|
@patch.object(ceph_hooks, 'send_request_if_needed')
|
|
@patch.object(ceph_hooks, 'is_request_complete',
|
|
lambda *args, **kwargs: False)
|
|
def test_mon_relation_send_broker_request(self,
|
|
mock_send_request_if_needed):
|
|
_ceph = self.patch('ceph')
|
|
_restart = self.patch('restart')
|
|
self.relation_get.return_value = 'seckey'
|
|
ceph_hooks.mon_relation()
|
|
self.assertFalse(_restart.called)
|
|
self.assertFalse(_ceph.import_radosgw_key.called)
|
|
self.assertFalse(self.CONFIGS.called)
|
|
self.assertTrue(mock_send_request_if_needed.called)
|
|
|
|
def test_gateway_relation(self):
|
|
self.unit_get.return_value = 'myserver'
|
|
ceph_hooks.gateway_relation()
|
|
self.relation_set.assert_called_with(hostname='myserver', port=80)
|
|
|
|
def test_start(self):
|
|
ceph_hooks.start()
|
|
cmd = ['service', 'radosgw', 'start']
|
|
self.subprocess.call.assert_called_with(cmd)
|
|
|
|
def test_stop(self):
|
|
ceph_hooks.stop()
|
|
cmd = ['service', 'radosgw', 'stop']
|
|
self.subprocess.call.assert_called_with(cmd)
|
|
|
|
def test_restart(self):
|
|
ceph_hooks.restart()
|
|
cmd = ['service', 'radosgw', 'restart']
|
|
self.subprocess.call.assert_called_with(cmd)
|
|
|
|
@patch.object(ceph_hooks, 'setup_keystone_certs')
|
|
@patch('charmhelpers.contrib.openstack.ip.service_name',
|
|
lambda *args: 'ceph-radosgw')
|
|
@patch('charmhelpers.contrib.openstack.ip.config')
|
|
def test_identity_joined_early_version(self, _config,
|
|
mock_setup_keystone_certs):
|
|
self.cmp_pkgrevno.return_value = -1
|
|
ceph_hooks.identity_joined()
|
|
self.assertTrue(mock_setup_keystone_certs.called)
|
|
self.sys.exit.assert_called_with(1)
|
|
|
|
@patch('charmhelpers.contrib.openstack.ip.service_name',
|
|
lambda *args: 'ceph-radosgw')
|
|
@patch('charmhelpers.contrib.openstack.ip.resolve_address')
|
|
@patch('charmhelpers.contrib.openstack.ip.config')
|
|
def test_identity_joined(self, _config, _resolve_address):
|
|
self.related_units = ['unit/0']
|
|
self.cmp_pkgrevno.return_value = 1
|
|
_resolve_address.return_value = 'myserv'
|
|
_config.side_effect = self.test_config.get
|
|
self.test_config.set('region', 'region1')
|
|
self.test_config.set('operator-roles', 'admin')
|
|
self.unit_get.return_value = 'myserv'
|
|
ceph_hooks.identity_joined(relid='rid')
|
|
self.relation_set.assert_called_with(
|
|
service='swift',
|
|
region='region1',
|
|
public_url='http://myserv:80/swift/v1',
|
|
internal_url='http://myserv:80/swift/v1',
|
|
requested_roles='admin',
|
|
relation_id='rid',
|
|
admin_url='http://myserv:80/swift')
|
|
|
|
@patch('charmhelpers.contrib.openstack.ip.service_name',
|
|
lambda *args: 'ceph-radosgw')
|
|
@patch('charmhelpers.contrib.openstack.ip.is_clustered')
|
|
@patch('charmhelpers.contrib.openstack.ip.unit_get')
|
|
@patch('charmhelpers.contrib.openstack.ip.config')
|
|
def test_identity_joined_public_name(self, _config, _unit_get,
|
|
_is_clustered):
|
|
self.related_units = ['unit/0']
|
|
_config.side_effect = self.test_config.get
|
|
self.test_config.set('os-public-hostname', 'files.example.com')
|
|
_unit_get.return_value = 'myserv'
|
|
_is_clustered.return_value = False
|
|
ceph_hooks.identity_joined(relid='rid')
|
|
self.relation_set.assert_called_with(
|
|
service='swift',
|
|
region='RegionOne',
|
|
public_url='http://files.example.com:80/swift/v1',
|
|
internal_url='http://myserv:80/swift/v1',
|
|
requested_roles='Member,Admin',
|
|
relation_id='rid',
|
|
admin_url='http://myserv:80/swift')
|
|
|
|
@patch.object(ceph_hooks, 'identity_joined')
|
|
def test_identity_changed(self, mock_identity_joined):
|
|
_restart = self.patch('restart')
|
|
ceph_hooks.identity_changed()
|
|
self.CONFIGS.write_all.assert_called_with()
|
|
self.assertTrue(_restart.called)
|
|
self.assertTrue(mock_identity_joined.called)
|
|
|
|
@patch('charmhelpers.contrib.openstack.ip.is_clustered')
|
|
@patch('charmhelpers.contrib.openstack.ip.unit_get')
|
|
@patch('charmhelpers.contrib.openstack.ip.config')
|
|
def test_canonical_url_ipv6(self, _config, _unit_get, _is_clustered):
|
|
ipv6_addr = '2001:db8:85a3:8d3:1319:8a2e:370:7348'
|
|
_config.side_effect = self.test_config.get
|
|
_unit_get.return_value = ipv6_addr
|
|
_is_clustered.return_value = False
|
|
self.assertEquals(ceph_hooks.canonical_url({}, PUBLIC),
|
|
'http://[%s]' % ipv6_addr)
|
|
|
|
def test_cluster_changed(self):
|
|
_id_joined = self.patch('identity_joined')
|
|
self.relation_ids.return_value = ['rid']
|
|
ceph_hooks.cluster_changed()
|
|
self.CONFIGS.write_all.assert_called_with()
|
|
_id_joined.assert_called_with(relid='rid')
|
|
|
|
def test_ha_relation_joined_no_vip(self):
|
|
self.test_config.set('vip', '')
|
|
ceph_hooks.ha_relation_joined()
|
|
self.sys.exit.assert_called_with(1)
|
|
|
|
def test_ha_relation_joined_vip(self):
|
|
self.test_config.set('ha-bindiface', 'eth8')
|
|
self.test_config.set('ha-mcastport', '5000')
|
|
self.test_config.set('vip', '10.0.0.10')
|
|
self.get_iface_for_address.return_value = 'eth7'
|
|
self.get_netmask_for_address.return_value = '255.255.0.0'
|
|
ceph_hooks.ha_relation_joined()
|
|
eth_params = ('params ip="10.0.0.10" cidr_netmask="255.255.0.0" '
|
|
'nic="eth7"')
|
|
resources = {'res_cephrg_haproxy': 'lsb:haproxy',
|
|
'res_cephrg_eth7_vip': 'ocf:heartbeat:IPaddr2'}
|
|
resource_params = {'res_cephrg_haproxy': 'op monitor interval="5s"',
|
|
'res_cephrg_eth7_vip': eth_params}
|
|
self.relation_set.assert_called_with(
|
|
init_services={'res_cephrg_haproxy': 'haproxy'},
|
|
corosync_bindiface='eth8',
|
|
corosync_mcastport='5000',
|
|
resource_params=resource_params,
|
|
resources=resources,
|
|
clones={'cl_cephrg_haproxy': 'res_cephrg_haproxy'})
|
|
|
|
def test_ha_relation_changed(self):
|
|
_id_joined = self.patch('identity_joined')
|
|
self.relation_get.return_value = True
|
|
self.relation_ids.return_value = ['rid']
|
|
ceph_hooks.ha_relation_changed()
|
|
_id_joined.assert_called_with(relid='rid')
|