KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use _cleanup_labels method in autotest

Best Python code snippet using autotest_python

host.py

Source:host.py

...48            self.messages.append('Locked host')49        elif options.unlock:50            self.data['locked'] = False51            self.messages.append('Unlocked host')52    def _cleanup_labels(self, labels, platform=None):53        """Removes the platform label from the overall labels"""54        if platform:55            return [label for label in labels56                    if label != platform]57        else:58            try:59                return [label for label in labels60                        if not label['platform']]61            except TypeError:62                # This is a hack - the server will soon63                # do this, so all this code should be removed.64                return labels65    def get_items(self):66        return self.hosts67class host_help(host):68    """Just here to get the atest logic working.69    Usage is set by its parent"""70    pass71class host_list(action_common.atest_list, host):72    """atest host list [--mlist <file>|<hosts>] [--label <label>]73       [--status <status1,status2>] [--acl <ACL>] [--user <user>]"""74    def __init__(self):75        super(host_list, self).__init__()76        self.parser.add_option('-b', '--label',77                               default='',78                               help='Only list hosts with all these labels '79                               '(comma separated)')80        self.parser.add_option('-s', '--status',81                               default='',82                               help='Only list hosts with any of these '83                               'statuses (comma separated)')84        self.parser.add_option('-a', '--acl',85                               default='',86                               help='Only list hosts within this ACL')87        self.parser.add_option('-u', '--user',88                               default='',89                               help='Only list hosts available to this user')90        self.parser.add_option('-N', '--hostnames-only', help='Only return '91                               'hostnames for the machines queried.',92                               action='store_true')93        self.parser.add_option('--locked',94                               default=False,95                               help='Only list locked hosts',96                               action='store_true')97        self.parser.add_option('--unlocked',98                               default=False,99                               help='Only list unlocked hosts',100                               action='store_true')101    def parse(self):102        """Consume the specific options"""103        label_info = topic_common.item_parse_info(attribute_name='labels',104                                                  inline_option='label')105        (options, leftover) = super(host_list, self).parse([label_info])106        self.status = options.status107        self.acl = options.acl108        self.user = options.user109        self.hostnames_only = options.hostnames_only110        if options.locked and options.unlocked:111            self.invalid_syntax('--locked and --unlocked are '112                                'mutually exclusive')113        self.locked = options.locked114        self.unlocked = options.unlocked115        return (options, leftover)116    def execute(self):117        filters = {}118        check_results = {}119        if self.hosts:120            filters['hostname__in'] = self.hosts121            check_results['hostname__in'] = 'hostname'122        if self.labels:123            if len(self.labels) == 1:124                # This is needed for labels with wildcards (x86*)125                filters['labels__name__in'] = self.labels126                check_results['labels__name__in'] = None127            else:128                filters['multiple_labels'] = self.labels129                check_results['multiple_labels'] = None130        if self.status:131            statuses = self.status.split(',')132            statuses = [status.strip() for status in statuses133                        if status.strip()]134            filters['status__in'] = statuses135            check_results['status__in'] = None136        if self.acl:137            filters['aclgroup__name'] = self.acl138            check_results['aclgroup__name'] = None139        if self.user:140            filters['aclgroup__users__login'] = self.user141            check_results['aclgroup__users__login'] = None142        if self.locked or self.unlocked:143            filters['locked'] = self.locked144            check_results['locked'] = None145        return super(host_list, self).execute(op='get_hosts',146                                              filters=filters,147                                              check_results=check_results)148    def output(self, results):149        if results:150            # Remove the platform from the labels.151            for result in results:152                result['labels'] = self._cleanup_labels(result['labels'],153                                                        result['platform'])154        if self.hostnames_only:155            self.print_list(results, key='hostname')156        else:157            super(host_list, self).output(results, keys=['hostname', 'status',158                                                         'locked', 'platform', 'labels'])159class host_stat(host):160    """atest host stat --mlist <file>|<hosts>"""161    usage_action = 'stat'162    def execute(self):163        results = []164        # Convert wildcards into real host stats.165        existing_hosts = []166        for host in self.hosts:167            if host.endswith('*'):168                stats = self.execute_rpc('get_hosts',169                                         hostname__startswith=host.rstrip('*'))170                if len(stats) == 0:171                    self.failure('No hosts matching %s' % host, item=host,172                                 what_failed='Failed to stat')173                    continue174            else:175                stats = self.execute_rpc('get_hosts', hostname=host)176                if len(stats) == 0:177                    self.failure('Unknown host %s' % host, item=host,178                                 what_failed='Failed to stat')179                    continue180            existing_hosts.extend(stats)181        for stat in existing_hosts:182            host = stat['hostname']183            # The host exists, these should succeed184            acls = self.execute_rpc('get_acl_groups', hosts__hostname=host)185            labels = self.execute_rpc('get_labels', host__hostname=host)186            results.append([[stat], acls, labels])187        return results188    def output(self, results):189        for stats, acls, labels in results:190            print '-' * 5191            self.print_fields(stats,192                              keys=['hostname', 'platform',193                                    'status', 'locked', 'locked_by',194                                    'lock_time', 'protection', ])195            self.print_by_ids(acls, 'ACLs', line_before=True)196            labels = self._cleanup_labels(labels)197            self.print_by_ids(labels, 'Labels', line_before=True)198class host_jobs(host):199    """atest host jobs [--max-query] --mlist <file>|<hosts>"""200    usage_action = 'jobs'201    def __init__(self):202        super(host_jobs, self).__init__()203        self.parser.add_option('-q', '--max-query',204                               help='Limits the number of results '205                               '(20 by default)',206                               type='int', default=20)207    def parse(self):208        """Consume the specific options"""209        (options, leftover) = super(host_jobs, self).parse()210        self.max_queries = options.max_query...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.