def setup_package(): bonding_defaults, bonding_name2numeric = bonding_default_fpath() bonding_dump_patchers.append( mock.patch('vdsm.network.link.bond.sysfs_options.BONDING_DEFAULTS', bonding_defaults)) bonding_dump_patchers.append( mock.patch( 'vdsm.network.link.bond.sysfs_options_mapper.' 'BONDING_NAME2NUMERIC_PATH', bonding_name2numeric)) for patcher in bonding_dump_patchers: patcher.start()
def _assert_acquired_ifaces_post_switch_setup( self, _ovs_info, nets2add, expected_ifaces): ovsdb = driver.vsctl.create() with mock.patch('vdsm.network.ovs.driver.vsctl.Transaction.commit', return_value=None), \ mock.patch('vdsm.network.ovs.switch.link.get_link', return_value={'address': '01:23:45:67:89:ab'}): setup = switch.NetsAdditionSetup(ovsdb, _ovs_info) with setup.add(nets2add): pass self.assertEqual(setup.acquired_ifaces, expected_ifaces)
def _assert_acquired_ifaces_post_switch_setup(self, _ovs_info, nets2add, expected_ifaces): ovsdb = driver.vsctl.create() with mock.patch('vdsm.network.ovs.driver.vsctl.Transaction.commit', return_value=None), \ mock.patch('vdsm.network.ovs.switch.link.get_link', return_value={'address': '01:23:45:67:89:ab'}): setup = switch.NetsAdditionSetup(ovsdb, _ovs_info) with setup.add(nets2add): pass self.assertEqual(setup.acquired_ifaces, expected_ifaces)
def schema_from(yaml_str): pickled_yaml = pickle.dumps(yaml.load(yaml_str)) mocked_open = mock.mock_open(read_data=pickled_yaml) with mock.patch('{}.io.open'.format(vdsmapi.__name__), mocked_open, create=True): return vdsmapi.Schema.vdsm_api(strict_mode=False)
def _schema_from(yaml_str): pickled_yaml = pickle.dumps(yaml.load(yaml_str)) mocked_open = mock.mock_open(read_data=pickled_yaml) with mock.patch('{}.io.open'.format(vdsmapi.__name__), mocked_open, create=True): return vdsmapi.Schema.vdsm_api(strict_mode=False)
def _assert_acquired_ifaces_post_switch_setup(self, _ovs_info, nets2add, bonds2add, bonds2edit, expected_ifaces): ovsdb = driver.vsctl.create() with mock.patch('vdsm.network.ovs.driver.vsctl.Transaction.commit', return_value=None), \ mock.patch('vdsm.network.ovs.switch.link.get_link', return_value={'address': '01:23:45:67:89:ab'}): with switch.Setup(ovsdb, _ovs_info) as s: s.edit_bonds(bonds2edit) s.add_bonds(bonds2add) s.add_nets(nets2add) self.assertEqual(s.acquired_ifaces, expected_ifaces)
def test_host_not_in_shutdown(self): with mock.patch('dbus.SystemBus') as mock_sysbus: instance = mock_sysbus() obj = instance.get_object() obj.ListJobs.return_value = () in_shutdown = hostutils.host_in_shutdown() self.assertEqual(in_shutdown, False)
def test_supported(self): with namedTemporaryDir() as src: # A list of 3 file names, where the files are temporary. sizes = [os.path.basename(f.name) for f in [ tempfile.NamedTemporaryFile( dir=src, delete=False ) for _ in range(3) ]] with mock.patch('{}.open'.format(hugepages.__name__), mock.mock_open(read_data='0'), create=True): self.assertEqual(set(hugepages.supported(src)), set(sizes))
def test_state(self): with namedTemporaryDir() as src: # A list of 3 file names, where the files are temporary. sizes = [os.path.basename(f.name) for f in [ tempfile.NamedTemporaryFile( dir=src, delete=False ) for _ in range(3) ]] with mock.patch('{}.open'.format(hugepages.__name__), mock.mock_open(read_data='1234'), create=True): self.assertEqual(len(hugepages.state(src)), len(sizes)) for value in hugepages.state(src).values(): self.assertEqual(value, _STATE)
def test_dbus_exception(self): with mock.patch('dbus.SystemBus') as mock_sysbus: mock_sysbus.side_effect = dbus.DBusException() in_shutdown = hostutils.host_in_shutdown() self.assertEqual(in_shutdown, False)