def hosts(self): """Obtain the managed hosts.""" hosts = [] jsonresult = {} page = 0 per_page = 100 credential = utils.get_credential(self.connect_scan_task) while (page == 0 or int(jsonresult.get(PER_PAGE, 0)) == len(jsonresult.get(RESULTS, []))): page += 1 params = {PAGE: page, PER_PAGE: per_page, THIN: 1} response, url = utils.execute_request(self.connect_scan_task, url=HOSTS_V2_URL, query_params=params) # pylint: disable=no-member if response.status_code != requests.codes.ok: raise SatelliteException('Invalid response code %s' ' for url: %s' % (response.status_code, url)) jsonresult = response.json() for result in jsonresult.get(RESULTS, []): host_name = result.get(NAME) host_id = result.get(ID) if host_name is not None and host_id is not None: unique_name = '%s_%s' % (host_name, host_id) hosts.append(unique_name) self.record_conn_result(unique_name, credential) return hosts
def hosts(self): """Obtain the managed hosts.""" hosts = [] credential = utils.get_credential(self.scan_task) client, user, password = utils.get_sat5_client(self.scan_task) try: key = client.auth.login(user, password) systems = client.system.list_user_systems(key) client.auth.logout(key) for system in systems: name = system.get(NAME) if name is not None: hosts.append(name) self.record_conn_result(name, credential) except xmlrpc.client.Fault as xml_error: raise SatelliteException(str(xml_error)) return hosts
def test_get_credential(self): """Test the method to extract credential.""" cred = get_credential(self.scan_task) self.assertEqual(cred, self.cred)