def test_full_dryrun(self): self.ch = CloudHandler(self.cfg) with util.global_dry_run(): nid = self.ch.create_node(cfg.node_defs['node1']) self.sc = sc.ServiceComposer.instantiate(protocol='dummy') self.uds = UDS.instantiate(protocol='dict') mib = ib.real_main_info_broker = \ ib.InfoRouter(sub_providers=[ self.uds, self.sc, dsp.DynamicStateProvider(self.sc, self.ch), sp.SynchronizationProvider(), CloudHandlerProvider(self.ch) ]) try: log.debug( mib.get( 'node.resource.state', dict(instance_id=nid, node_id="test", backend_id='lpds'))) finally: self.ch.drop_node( dict(instance_id=nid, node_id="test", backend_id='lpds'))
def test_create_using_ip(self): node_def = cfg.node_defs['node_lpds'] self.ch = CloudHandler(self.cfg) self.sc = sc.ServiceComposer.instantiate(protocol='dummy') self.uds = UDS.instantiate(protocol='dict') self.uds.kvstore.set_item('node_def:test', [node_def]) mib = ib.InfoRouter(main_info_broker=True, sub_providers=[ self.uds, self.sc, dsp.DynamicStateProvider(self.sc, self.ch), sp.SynchronizationProvider(), CloudHandlerProvider(self.ch) ]) eid = str(uuid.uuid4()) nid = str(uuid.uuid4()) node_desc = dict(infra_id=eid, node_id=nid, type='test', user_id=1, name='test') infrap = ip.InfraProcessor.instantiate('basic', self.uds, self.ch, self.sc) cmd_cre = infrap.cri_create_infrastructure(eid) cmd_crn = infrap.cri_create_node(node_desc) infrap.push_instructions(cmd_cre) node = infrap.push_instructions(cmd_crn)[0] status = mib.get('node.resource.state', node) self.drop_nodes.append(dict(instance_id=nid, node_id="test"))
def test_create_node(self): self.ch = CloudHandler(self.cfg) node_def = cfg.node_defs['node_lpds'] log.debug("node_desc: %r", node_def) nid = self.ch.create_node(node_def) log.debug("Resource acquired; node_id = %r", nid) self.drop_nodes.append( dict(instance_id=nid, node_id="test", backend_id=node_def['backend_id'])) self.update_drop_nodes()
def test_full_dryrun(self): self.ch = CloudHandler(self.cfg) with util.global_dry_run(): nid = self.ch.create_node(cfg.node_defs['node1']) self.sc = sc.ServiceComposer.instantiate(protocol='dummy') self.uds = UDS.instantiate(protocol='dict') mib = ib.real_main_info_broker = \ ib.InfoRouter(sub_providers=[ self.uds, self.sc, dsp.DynamicStateProvider(self.sc, self.ch), sp.SynchronizationProvider(), CloudHandlerProvider(self.ch) ]) try: log.debug(mib.get('node.resource.state', dict(instance_id=nid, node_id="test", backend_id='lpds'))) finally: self.ch.drop_node(dict(instance_id=nid, node_id="test", backend_id='lpds'))
def test_create_using_ip(self): node_def = cfg.node_defs['node_lpds'] self.ch = CloudHandler(self.cfg) self.sc = sc.ServiceComposer.instantiate(protocol='dummy') self.uds = UDS.instantiate(protocol='dict') self.uds.kvstore.set_item('node_def:test', [node_def]) mib = ib.InfoRouter(main_info_broker=True, sub_providers=[ self.uds, self.sc, dsp.DynamicStateProvider(self.sc, self.ch), sp.SynchronizationProvider(), CloudHandlerProvider(self.ch) ]) eid = str(uuid.uuid4()) nid = str(uuid.uuid4()) node_desc = dict( infra_id=eid, node_id=nid, type='test', user_id=1, name='test') infrap = ip.InfraProcessor.instantiate( 'basic', self.uds, self.ch, self.sc) cmd_cre = infrap.cri_create_infrastructure(eid) cmd_crn = infrap.cri_create_node(node_desc) infrap.push_instructions(cmd_cre) node = infrap.push_instructions(cmd_crn)[0] status = mib.get('node.resource.state', node) self.drop_nodes.append(dict(instance_id=nid, node_id="test"))
def test_drop_node(self): self.ch = CloudHandler(self.cfg) remaining = [] last_exception = None for i in self.drop_nodes: try: self.ch.drop_node(i) except Exception as ex: log.exception('Failure:') last_exception = ex remaining.append(i) else: log.debug('Successfully dropped node.') self.drop_nodes = remaining self.update_drop_nodes() if last_exception: raise last_exception, None, sys.exc_info()[2]
def test_create_node(self): self.ch = CloudHandler(self.cfg) node_def = cfg.node_defs['node_lpds'] log.debug("node_desc: %r", node_def) nid = self.ch.create_node(node_def) log.debug("Resource acquired; node_id = %r", nid) self.drop_nodes.append(dict(instance_id=nid, node_id="test", backend_id=node_def['backend_id'])) self.update_drop_nodes()
def test_node_status(self): self.ch = CloudHandler(self.cfg) last_exception = None self.sc = sc.ServiceComposer.instantiate(protocol='dummy') self.uds = UDS.instantiate(protocol='dict') mib = ib.InfoRouter(main_info_broker=True, sub_providers=[ self.uds, self.sc, dsp.DynamicStateProvider(self.sc, self.ch), sp.SynchronizationProvider(), CloudHandlerProvider(self.ch) ]) for i in self.drop_nodes: try: node_state = mib.get('node.resource.state', i) log.info("Status of node %r is %r", i, node_state) except Exception as ex: log.exception('Failure:') last_exception = ex if last_exception: raise last_exception, None, sys.exc_info()[2]
class DummyTest(unittest.TestCase): def setUp(self): self.cfg = common.configure() self.ch = CloudHandler(self.cfg.ch_cfgs) def test_create_node(self): nd = dict() nd['node_id'] = 'test_node_id' nd['infra_id'] = 'test_infra_id' nd['name'] = 'test_name' nd['backend_id'] = 'dummy' ch = self.ch.instantiate_ch(nd) nid = ch.cri_create_node(nd).perform(ch) self.assertIsNotNone(nid) self.assertIn(nid, ch.kvstore) self.assertIn('running', ch.kvstore[nid]) self.assertTrue(ch.kvstore[nid]['running']) def test_node_state(self): nd = dict() nd['node_id'] = 'test_node_id' nd['infra_id'] = 'test_infra_id' nd['name'] = 'test_name' nd['backend_id'] = 'dummy' nid = self.ch.create_node(nd) instance_data = dict() instance_data['instance_id'] = nid instance_data['backend_id'] = 'dummy' eq_(self.ch.get_state(instance_data), 'ready') def test_drop_node(self): nd = dict() nd['node_id'] = 'test_node_id' nd['infra_id'] = 'test_infra_id' nd['name'] = 'test_name' nd['backend_id'] = 'dummy' nid = self.ch.create_node(nd) instance_data = dict() instance_data['instance_id'] = nid instance_data['backend_id'] = 'dummy' self.ch.drop_node(instance_data) s = self.ch.get_state(instance_data) eq_(s, 'unknown')
def setUp(self): self.cfg = common.configure() self.ch = CloudHandler(self.cfg.ch_cfgs)
class BotoTest(unittest.TestCase): def setUp(self): if os.path.isfile(DROP_NODES_FILE): with open(DROP_NODES_FILE) as f: self.drop_nodes = yaml.load(f) else: self.drop_nodes = [] self.cfg = cfg.ch_cfgs cleaner = util.Cleaner(hide_keys=['password']) log.debug( 'Using Boto config:\n%s', yaml.dump(cleaner.deep_copy(self.cfg))) def test_full_dryrun(self): self.ch = CloudHandler(self.cfg) with util.global_dry_run(): nid = self.ch.create_node(cfg.node_defs['node1']) self.sc = sc.ServiceComposer.instantiate(protocol='dummy') self.uds = UDS.instantiate(protocol='dict') mib = ib.real_main_info_broker = \ ib.InfoRouter(sub_providers=[ self.uds, self.sc, dsp.DynamicStateProvider(self.sc, self.ch), sp.SynchronizationProvider(), CloudHandlerProvider(self.ch) ]) try: log.debug(mib.get('node.resource.state', dict(instance_id=nid, node_id="test", backend_id='lpds'))) finally: self.ch.drop_node(dict(instance_id=nid, node_id="test", backend_id='lpds')) def update_drop_nodes(self): with open(DROP_NODES_FILE, 'w') as f: f.write(yaml.dump(self.drop_nodes)) log.debug("Allocated nodes: %r", self.drop_nodes) @real_resource def test_create_node(self): self.ch = CloudHandler(self.cfg) node_def = cfg.node_defs['node_lpds'] log.debug("node_desc: %r", node_def) nid = self.ch.create_node(node_def) log.debug("Resource acquired; node_id = %r", nid) self.drop_nodes.append(dict(instance_id=nid, node_id="test", backend_id=node_def['backend_id'])) self.update_drop_nodes() @real_resource def test_create_using_ip(self): node_def = cfg.node_defs['node_lpds'] self.ch = CloudHandler(self.cfg) self.sc = sc.ServiceComposer.instantiate(protocol='dummy') self.uds = UDS.instantiate(protocol='dict') self.uds.kvstore.set_item('node_def:test', [node_def]) mib = ib.InfoRouter(main_info_broker=True, sub_providers=[ self.uds, self.sc, dsp.DynamicStateProvider(self.sc, self.ch), sp.SynchronizationProvider(), CloudHandlerProvider(self.ch) ]) eid = str(uuid.uuid4()) nid = str(uuid.uuid4()) node_desc = dict( infra_id=eid, node_id=nid, type='test', user_id=1, name='test') infrap = ip.InfraProcessor.instantiate( 'basic', self.uds, self.ch, self.sc) cmd_cre = infrap.cri_create_infrastructure(eid) cmd_crn = infrap.cri_create_node(node_desc) infrap.push_instructions(cmd_cre) node = infrap.push_instructions(cmd_crn)[0] status = mib.get('node.resource.state', node) self.drop_nodes.append(dict(instance_id=nid, node_id="test")) @real_resource def test_drop_node(self): self.ch = CloudHandler(self.cfg) remaining = [] last_exception = None for i in self.drop_nodes: try: self.ch.drop_node(i) except Exception as ex: log.exception('Failure:') last_exception = ex remaining.append(i) else: log.debug('Successfully dropped node.') self.drop_nodes = remaining self.update_drop_nodes() if last_exception: raise last_exception, None, sys.exc_info()[2] @real_resource def test_node_status(self): self.ch = CloudHandler(self.cfg) last_exception = None self.sc = sc.ServiceComposer.instantiate(protocol='dummy') self.uds = UDS.instantiate(protocol='dict') mib = ib.InfoRouter(main_info_broker=True, sub_providers=[ self.uds, self.sc, dsp.DynamicStateProvider(self.sc, self.ch), sp.SynchronizationProvider(), CloudHandlerProvider(self.ch) ]) for i in self.drop_nodes: try: node_state = mib.get('node.resource.state', i) log.info("Status of node %r is %r", i, node_state) except Exception as ex: log.exception('Failure:') last_exception = ex if last_exception: raise last_exception, None, sys.exc_info()[2]
class BotoTest(unittest.TestCase): def setUp(self): if os.path.isfile(DROP_NODES_FILE): with open(DROP_NODES_FILE) as f: self.drop_nodes = yaml.load(f) else: self.drop_nodes = [] self.cfg = cfg.ch_cfgs cleaner = util.Cleaner(hide_keys=['password']) log.debug('Using Boto config:\n%s', yaml.dump(cleaner.deep_copy(self.cfg))) def test_full_dryrun(self): self.ch = CloudHandler(self.cfg) with util.global_dry_run(): nid = self.ch.create_node(cfg.node_defs['node1']) self.sc = sc.ServiceComposer.instantiate(protocol='dummy') self.uds = UDS.instantiate(protocol='dict') mib = ib.real_main_info_broker = \ ib.InfoRouter(sub_providers=[ self.uds, self.sc, dsp.DynamicStateProvider(self.sc, self.ch), sp.SynchronizationProvider(), CloudHandlerProvider(self.ch) ]) try: log.debug( mib.get( 'node.resource.state', dict(instance_id=nid, node_id="test", backend_id='lpds'))) finally: self.ch.drop_node( dict(instance_id=nid, node_id="test", backend_id='lpds')) def update_drop_nodes(self): with open(DROP_NODES_FILE, 'w') as f: f.write(yaml.dump(self.drop_nodes)) log.debug("Allocated nodes: %r", self.drop_nodes) @real_resource def test_create_node(self): self.ch = CloudHandler(self.cfg) node_def = cfg.node_defs['node_lpds'] log.debug("node_desc: %r", node_def) nid = self.ch.create_node(node_def) log.debug("Resource acquired; node_id = %r", nid) self.drop_nodes.append( dict(instance_id=nid, node_id="test", backend_id=node_def['backend_id'])) self.update_drop_nodes() @real_resource def test_create_using_ip(self): node_def = cfg.node_defs['node_lpds'] self.ch = CloudHandler(self.cfg) self.sc = sc.ServiceComposer.instantiate(protocol='dummy') self.uds = UDS.instantiate(protocol='dict') self.uds.kvstore.set_item('node_def:test', [node_def]) mib = ib.InfoRouter(main_info_broker=True, sub_providers=[ self.uds, self.sc, dsp.DynamicStateProvider(self.sc, self.ch), sp.SynchronizationProvider(), CloudHandlerProvider(self.ch) ]) eid = str(uuid.uuid4()) nid = str(uuid.uuid4()) node_desc = dict(infra_id=eid, node_id=nid, type='test', user_id=1, name='test') infrap = ip.InfraProcessor.instantiate('basic', self.uds, self.ch, self.sc) cmd_cre = infrap.cri_create_infrastructure(eid) cmd_crn = infrap.cri_create_node(node_desc) infrap.push_instructions(cmd_cre) node = infrap.push_instructions(cmd_crn)[0] status = mib.get('node.resource.state', node) self.drop_nodes.append(dict(instance_id=nid, node_id="test")) @real_resource def test_drop_node(self): self.ch = CloudHandler(self.cfg) remaining = [] last_exception = None for i in self.drop_nodes: try: self.ch.drop_node(i) except Exception as ex: log.exception('Failure:') last_exception = ex remaining.append(i) else: log.debug('Successfully dropped node.') self.drop_nodes = remaining self.update_drop_nodes() if last_exception: raise last_exception, None, sys.exc_info()[2] @real_resource def test_node_status(self): self.ch = CloudHandler(self.cfg) last_exception = None self.sc = sc.ServiceComposer.instantiate(protocol='dummy') self.uds = UDS.instantiate(protocol='dict') mib = ib.InfoRouter(main_info_broker=True, sub_providers=[ self.uds, self.sc, dsp.DynamicStateProvider(self.sc, self.ch), sp.SynchronizationProvider(), CloudHandlerProvider(self.ch) ]) for i in self.drop_nodes: try: node_state = mib.get('node.resource.state', i) log.info("Status of node %r is %r", i, node_state) except Exception as ex: log.exception('Failure:') last_exception = ex if last_exception: raise last_exception, None, sys.exc_info()[2]