class MockVirConnectTests(TestCaseBase): def setUp(self): self._vms = [MockVirDomain(*spec) for spec in VM_SPECS] self._mock = MockVirConnect(vms=self._vms) def test_list_all_domains(self): vms = self._mock.listAllDomains() assert len(vms) == len(self._vms) def test_list_defined_domains(self): vms = self._mock.listDefinedDomains() assert sorted(vms) == \ sorted(spec.name for spec in VM_SPECS if not spec.active) def test_list_domains_id(self): vms = self._mock.listDomainsID() assert len(vms) == 2 def test_lookup_by_name(self): vm = self._mock.lookupByName('RHEL_0') assert 'RHEL_0' == vm.name() def test_lookup_by_name_failed(self): with pytest.raises(libvirt.libvirtError): self._mock.lookupByName('fakename') def test_lookup_by_id(self): vm = self._mock.lookupByID(0) assert 0 == vm.ID() def test_lookup_by_id_failed(self): with pytest.raises(libvirt.libvirtError): self._mock.lookupByID(99)
class MockVirConnectTests(TestCaseBase): def setUp(self): self._vms = [MockVirDomain(*spec) for spec in VM_SPECS] self._mock = MockVirConnect(vms=self._vms) def test_list_all_domains(self): vms = self._mock.listAllDomains() self.assertEqual(len(vms), len(self._vms)) def test_list_defined_domains(self): vms = self._mock.listDefinedDomains() self.assertEqual( sorted(vms), sorted(spec.name for spec in VM_SPECS if not spec.active)) def test_list_domains_id(self): vms = self._mock.listDomainsID() self.assertEqual(len(vms), 2) def test_lookup_by_name(self): vm = self._mock.lookupByName('RHEL_0') self.assertEqual('RHEL_0', vm.name()) def test_lookup_by_name_failed(self): self.assertRaises(libvirt.libvirtError, self._mock.lookupByName, 'fakename') def test_lookup_by_id(self): vm = self._mock.lookupByID(0) self.assertEqual(0, vm.ID()) def test_lookup_by_id_failed(self): self.assertRaises(libvirt.libvirtError, self._mock.lookupByID, 99)
def _connect(uri, username, passwd): mock = MockVirConnect(vms=self._vms) mock.listAllDomains = legacylistAllDomains setattr(mock, methodname, fakemethod) return mock
def _connect(uri, username, passwd): mock = MockVirConnect(vms=self._vms) mock.listAllDomains = legacylistAllDomainsWrongRaise return mock