def test_get_mors_with_properties(self): vmware.get_dvss(self.mock_dc_ref) self.mock_get_mors.assert_called_once_with( self.mock_si, vim.DistributedVirtualSwitch, container_ref=self.mock_dc_ref, property_list=['name'], traversal_spec=self.mock_traversal_spec)
def test_traversal_spec(self): mock_traversal_spec = MagicMock(return_value='traversal_spec') with patch( 'salt.utils.vmware.vmodl.query.PropertyCollector.TraversalSpec', mock_traversal_spec): vmware.get_dvss(self.mock_dc_ref) mock_traversal_spec.assert_has_calls([ call(path='childEntity', skip=False, type=vim.Folder), call(path='networkFolder', skip=True, type=vim.Datacenter, selectSet=['traversal_spec']) ])
def test_get_managed_object_name_call(self): mock_get_managed_object_name = MagicMock() with patch('salt.utils.vmware.get_managed_object_name', mock_get_managed_object_name): vmware.get_dvss(self.mock_dc_ref) mock_get_managed_object_name.assert_called_once_with(self.mock_dc_ref)
def test_filtered_pgs(self): ret = vmware.get_dvss(self.mock_dc_ref, dvs_names=['fake_pg1', 'fake_pg3', 'no_pg']) self.assertEqual( ret, [self.mock_items[0]['object'], self.mock_items[2]['object']])
def test_get_all_dvss(self): ret = vmware.get_dvss(self.mock_dc_ref, get_all_dvss=True) self.assertEqual(ret, [i['object'] for i in self.mock_items])
def test_get_no_dvss(self): ret = vmware.get_dvss(self.mock_dc_ref) self.assertEqual(ret, [])