class AWSResourceManager_test(unittest.TestCase): def setUp(self): set_debug(True) self.mgr = AWSResourceManager('us-east-1') # east coast def tearDown(self): pass def test_get_instance_list(self): vm_list = self.mgr.get_instance_list() print "%d instances" % len(vm_list) for inst in vm_list : if inst: msg("\tinst: " + inst) else: msg("\tinst: <null>") def test_stop_instance(self): self.mgr.stop_instance(aws_id) pass def test_start_instance(self): self.mgr.start_instance(aws_id) pass def test_get_instance_list_key_name(self): vm_list = self.mgr.get_instance_list(key_name='bogus') assert len(vm_list) == 0
ip_address = instance.ip_address while True: (s,o) = os_cmd("checkport.py " + ip_address + " 12345") if s == 0 : break BaseUtil.sleep(20) if s > 0 : msg("The amazon VM is not reachable - please make sure AWS instances are running and " + "'amazonhost' is configured properly.") msg("Skipping this part of testing") sys.exit(0) (s, pubKey) = os_cmd("cat ~/.ssh/id_rsa.pub") pubKey.rstrip('\n') #print "pubKey: " + pubKey if s == 0: #addSudoer("54.80.11.60", "sudoer", "ALL", pubKey = pubKey ) add_sudoer( 'ssh -i ' + os.environ['BS_HOME'] + '/info/vault/TheGreatKeyPair_East.pem.unlocked ubuntu@amazonhost', "amazonhost", "sudoer", remove_user = True, root_services = "ALL", pub_key = pubKey ) mgr.stop_instance('i-ba73ff54')