def test_read_domain_only_once(self): """Confirm that the domain is read from a file only once, even if the object is instantiated multiple times""" self.mox.StubOutWithMock(dom, 'read_domains') self.mox.StubOutWithMock(dom, 'write_domains') dom.read_domains('/tftpboot/test_fake_dom_file').AndReturn([]) dom.write_domains('/tftpboot/test_fake_dom_file', []) self.mox.ReplayAll() # Instantiate multiple instances x = dom.BareMetalDom() x = dom.BareMetalDom() x = dom.BareMetalDom()
def test_find_domain(self): domain = { 'status': 1, 'name': 'instance-00000001', 'memory_kb': 16777216, 'kernel_id': '1896115634', 'ramdisk_id': '', 'image_id': '1552326678', 'vcpus': 1, 'node_id': 6, 'mac_address': '02:16:3e:01:4e:c9', 'ip_address': '10.5.1.2' } # Create the mock objects self.mox.StubOutWithMock(dom, 'read_domains') self.mox.StubOutWithMock(dom, 'write_domains') # Expected calls dom.read_domains('/tftpboot/' 'test_fake_dom_file').AndReturn(fake_domains) dom.write_domains('/tftpboot/test_fake_dom_file', fake_domains) self.mox.ReplayAll() # Code under test bmdom = dom.BareMetalDom() # Expected values self.assertEquals(bmdom.find_domain('instance-00000001'), domain)
def test_init_remove_non_running_domain(self): """Check to see that all entries in the domain list are removed except for the one that is in the running state""" fake_file = StringIO.StringIO() domains = [ dict(node_id=1, name='i-00000001', status=power_state.NOSTATE), dict(node_id=2, name='i-00000002', status=power_state.RUNNING), dict(node_id=3, name='i-00000003', status=power_state.BLOCKED), dict(node_id=4, name='i-00000004', status=power_state.PAUSED), dict(node_id=5, name='i-00000005', status=power_state.SHUTDOWN), dict(node_id=6, name='i-00000006', status=power_state.SHUTOFF), dict(node_id=7, name='i-00000007', status=power_state.CRASHED), dict(node_id=8, name='i-00000008', status=power_state.SUSPENDED), dict(node_id=9, name='i-00000009', status=power_state.FAILED) ] # Create the mock objects self.mox.StubOutWithMock(dom, 'read_domains') self.mox.StubOutWithMock(dom, 'write_domains') dom.read_domains('/tftpboot/test_fake_dom_file').AndReturn(domains) dom.write_domains('/tftpboot/test_fake_dom_file', domains) self.mox.ReplayAll() # Code under test bmdom = dom.BareMetalDom() self.assertEqual(bmdom.domains, [{ 'node_id': 2, 'name': 'i-00000002', 'status': power_state.RUNNING }]) self.assertEqual(bmdom.fake_dom_nums, 1)
def test_init_no_domains(self): # Create the mock objects self.mox.StubOutWithMock(dom, 'read_domains') self.mox.StubOutWithMock(dom, 'write_domains') dom.read_domains('/tftpboot/test_fake_dom_file').AndReturn([]) dom.write_domains('/tftpboot/test_fake_dom_file', []) self.mox.ReplayAll() # Code under test bmdom = dom.BareMetalDom() # Expectd values self.assertEqual(bmdom.fake_dom_nums, 0)
def _get_connection(self): self._wrapped_conn = dom.BareMetalDom() return self._wrapped_conn