class MaasClientTest(unittest.TestCase): def setUp(self): self.c = MaasClient(AUTH) self.tag = 'a-test-tag-' + randomString() res = self.c.tag_new(self.tag) self.assertTrue(res) def tearDown(self): res = self.c.tag_delete(self.tag) self.assertTrue(res) def test_get_tags(self): self.assertTrue(self.tag in [t['name'] for t in self.c.tags])
class MaasClientZoneTest(unittest.TestCase): def setUp(self): self.c = MaasClient(AUTH) self.zone_name = 'testzone-' + randomString() res = self.c.zone_new(self.zone_name, 'zone created in unittest') self.assertTrue(res) def tearDown(self): res = self.c.zone_delete(self.zone_name) self.assertTrue(res) def test_get_zones(self): self.assertTrue(self.zone_name in [z['name'] for z in self.c.zones])
def poll_state(): """ Polls current state of Juju and MAAS :returns: list of Machine() and the Juju state :rtype: tuple (JujuState(), MaasState()) """ # Capture Juju state cmd = utils.get_command_output('juju status', combine_output=False) if cmd['ret']: raise SystemExit("Error connecting to juju: stderr: {e}".format( e=cmd['stderr'])) juju = JujuState(cmd['stdout']) maas = None if MULTI_SYSTEM: # Login to MAAS auth = MaasAuth() auth.get_api_key('root') # auth.login() # Load Client routines c = MaasClient(auth) # Capture Maas state maas = MaasState(c.nodes) c.tag_fpi(maas) c.nodes_accept_all() c.tag_name(maas) update_machine_info(juju, maas) return (juju, maas)
def poll_state(): """ Polls current state of Juju and MAAS :returns: list of Machine() and the Juju state :rtype: tuple (JujuState(), MaasState()) """ # Capture Juju state (ret, juju_stdout, juju_stderr, _) = utils.get_command_output('juju status', combine_output=False) if ret: log.debug("Juju state unknown, will re-poll in " "case bootstrap is taking a little longer to come up.") log.debug("Juju status output: {o} \n stderr: {e}".format( o=juju_stdout, e=juju_stderr)) # Stub out a juju status for now juju = JujuState('environment: local\nmachines:') else: try: juju = JujuState(juju_stdout) except: log.exception("Ignoring exception in parsing juju state.") juju = JujuState('environment: local\nmachines:') maas = None if MULTI_SYSTEM: # Login to MAAS auth = MaasAuth() auth.get_api_key('root') # auth.login() # Load Client routines c = MaasClient(auth) # Capture Maas state maas = MaasState(c.nodes) c.tag_fpi(maas) c.nodes_accept_all() c.tag_name(maas) update_machine_info(juju, maas) return (juju, maas)
def setUp(self): self.c = MaasClient(AUTH) self.zone_name = 'testzone-' + randomString() res = self.c.zone_new(self.zone_name, 'zone created in unittest') self.assertTrue(res)
def setUp(self): self.c = MaasClient(AUTH) self.tag = 'a-test-tag-' + randomString() res = self.c.tag_new(self.tag) self.assertTrue(res)