| import os |
| import yaml |
| import requests |
| import re |
| import pytest |
| |
| class salt_remote: |
| def cmd(self, tgt, fun, param=None,expr_form=None,tgt_type=None): |
| config = get_configuration() |
| headers = {'Accept':'application/json'} |
| login_payload = {'username':config['SALT_USERNAME'],'password':config['SALT_PASSWORD'],'eauth':'pam'} |
| accept_key_payload = {'fun': fun,'tgt':tgt,'client':'local','expr_form':expr_form,'tgt_type':tgt_type} |
| if param: |
| accept_key_payload['arg']=param |
| |
| login_request = requests.post(os.path.join(config['SALT_URL'],'login'),headers=headers,data=login_payload) |
| request = requests.post(config['SALT_URL'],headers=headers,data=accept_key_payload,cookies=login_request.cookies) |
| return request.json()['return'][0] |
| |
| |
| def init_salt_client(): |
| local = salt_remote() |
| return local |
| |
| |
| def list_to_target_string(node_list, separator): |
| result = '' |
| for node in node_list: |
| result += node + ' ' + separator + ' ' |
| return result.strip(' ' + separator + ' ') |
| |
| |
| def get_active_nodes(test=None): |
| config = get_configuration() |
| local_salt_client = init_salt_client() |
| |
| skipped_nodes = config.get('skipped_nodes') or [] |
| if test: |
| testname = test.split('.')[0] |
| if 'skipped_nodes' in config.get(testname).keys(): |
| skipped_nodes += config.get(testname)['skipped_nodes'] or [] |
| |
| if skipped_nodes: |
| print "\nNotice: {0} nodes will be skipped".format(skipped_nodes) |
| nodes = local_salt_client.cmd( |
| '* and not '+list_to_target_string(skipped_nodes, 'and not'), |
| 'test.ping', |
| expr_form='compound') |
| else: |
| nodes = local_salt_client.cmd('*', 'test.ping') |
| return nodes |
| |
| |
| def get_groups(test): |
| config = get_configuration() |
| testname = test.split('.')[0] |
| # assume that node name is like <name>.domain |
| # last 1-3 digits of name are index, e.g. 001 in cpu001 |
| # name doesn't contain dots |
| active_nodes = get_active_nodes() |
| |
| skipped_groups = config.get('skipped_groups') or [] |
| if config.get(testname): |
| if 'skipped_groups' in config.get(testname).keys(): |
| skipped_groups += config.get(testname)['skipped_groups'] or [] |
| |
| groups = [] |
| |
| for node in active_nodes: |
| index = re.search('[0-9]{1,3}$', node.split('.')[0]) |
| if index: |
| group_name = node.split('.')[0][:-len(index.group(0))] |
| else: |
| group_name = node |
| if group_name not in groups: |
| if group_name not in skipped_groups: |
| groups.append(group_name) |
| else: |
| if group_name+" - skipped" not in groups: |
| groups.append(group_name+" - skipped") |
| |
| return groups |
| |
| |
| def get_configuration(): |
| """function returns configuration for environment |
| and for test if it's specified""" |
| global_config_file = os.path.join( |
| os.path.dirname(os.path.abspath(__file__)), "../global_config.yaml") |
| with open(global_config_file, 'r') as file: |
| global_config = yaml.load(file) |
| for param in global_config.keys(): |
| if param in os.environ.keys(): |
| if ',' in os.environ[param]: |
| global_config[param]=[] |
| for item in os.environ[param].split(','): |
| global_config[param].append(item) |
| else: |
| global_config[param]=os.environ[param] |
| |
| return global_config |