def test_copy_conf_to_policyd( self, mock_policyd_dir_for, mock_os_walk, mock_os_path_exists, mock_write_file, mock_mkdir, ): mock_policyd_dir_for.return_value = '/etc' # Note '/some/dir' below has to match the patch on DASHBOAD_PKG_DIR # above. mock_os_walk.return_value = [('/some/dir/conf', ['a-dir'], ['file1']), ('/some/dir/conf/a-dir', [], ['file2'])] mock_os_path_exists.return_value = False with patch_open() as (_open, _file): _file.read.side_effect = ['content1', 'content2'] horizon_utils.copy_conf_to_policyd() mock_mkdir.assert_called_once_with('/etc/a-dir', owner='horizon', group='horizon', perms=0o775) _open.assert_has_calls([ call('/some/dir/conf/file1', 'r'), call('/some/dir/conf/a-dir/file2', 'r') ]) mock_write_file.assert_has_calls([ call('/etc/file1', 'content1', 'horizon', 'horizon'), call('/etc/a-dir/file2', 'content2', 'horizon', 'horizon') ])
def test_config_changed_upgrade_available(self): self.openstack_upgrade_available.return_value = True self.relations_of_type.return_value = False with patch_open() as (_open, _file): _file.read.return_value = "foo" hooks.config_changed() self.assertTrue(self.do_openstack_upgrade.called) self.assertTrue(self.CONFIGS.write_all.called)
def test_rsync_enable_rsync(self): with patch_open() as (_open, _file): ctxt = swift_context.RsyncContext() _file.read.return_value = 'RSYNC_ENABLE=false' ctxt.enable_rsyncd() _file.write.assert_called_with('RSYNC_ENABLE=true') _file.read.return_value = '#foo' ctxt.enable_rsyncd() _file.write.assert_called_with('RSYNC_ENABLE=true\n')
def test_config_changed_ipv6(self, mock_assert_charm_supports_ipv6): self.test_config.set('prefer-ipv6', True) self.openstack_upgrade_available.return_value = False self.relations_of_type.return_value = False with patch_open() as (_open, _file): _file.read.return_value = "foo" hooks.config_changed() self.assertTrue(self.CONFIGS.write_all.called) self.assertTrue(self.setup_rsync.called)
def test_config_changed_nrpe_master(self): self.openstack_upgrade_available.return_value = False self.relations_of_type.return_value = True with patch_open() as (_open, _file): _file.read.return_value = "foo" hooks.config_changed() self.assertTrue(self.CONFIGS.write_all.called) self.assertTrue(self.setup_rsync.called) self.assertTrue(self.update_nrpe_config.called)
def test_config_changed_with_openstack_upgrade_action(self): self.openstack_upgrade_available.return_value = True self.test_config.set('action-managed-upgrade', True) with patch_open() as (_open, _file): _file.read.return_value = "foo" hooks.config_changed() self.assertFalse(self.do_openstack_upgrade.called)
def test_find_block_devices(self): self.is_block_device.return_value = True self.is_device_mounted.side_effect = self._fake_is_device_mounted with patch_open() as (_open, _file): _file.read.return_value = PROC_PARTITIONS _file.readlines = MagicMock() _file.readlines.return_value = PROC_PARTITIONS.split('\n') result = swift_utils.find_block_devices() ex = ['/dev/sdb', '/dev/vdb', '/dev/cciss/c1d0'] self.assertEqual(ex, result)
def test_find_block_devices_real_world(self): self.is_block_device.return_value = True side_effect = lambda x: x in ["/dev/sdb", "/dev/sdb1"] # flake8: noqa self.is_device_mounted.side_effect = side_effect with patch_open() as (_open, _file): _file.read.return_value = REAL_WORLD_PARTITIONS _file.readlines = MagicMock() _file.readlines.return_value = REAL_WORLD_PARTITIONS.split('\n') result = swift_utils.find_block_devices() expected = ["/dev/sda"] self.assertEqual(expected, result)