import argparse from WFAUtil import WFAUtil import sys import traceback parser = argparse.ArgumentParser(description='Test python code for all WFAUtil libs') parser.add_argument('--host', help='String Parameter', required=True) parser.add_argument('--param_int', help='Integer Parameter', required=True, type=int) parser.add_argument('--param_float', help='Float Parameter') parser.add_argument('--param_password', help='Password Parameter', required=True) parser.add_argument('--nohost', help='String Parameter', required=True) args = vars(parser.parse_args()) wfa=WFAUtil() ## Test send_log wfa.send_log(message="=== Testing Method send_log for all log levels ===", level='INFO') wfa.send_log(message="=== INFO Message ===", level='INFO') wfa.send_log(message="=== Warning Message ===", level='WARN') wfa.send_log(message="=== Error Message ===", level='ERROR') wfa.send_log(message="=== Debug Message ===") #Test gre_credentials wfa.send_log(message="=== Testing Method get Credentials for a saved host. ===", level='INFO') creds = wfa.get_credentials(args['host']) wfa.send_log(level='INFO', message='Username: ' + creds['userName']) wfa.send_log(level='INFO', message='Password: '+ creds['password']) wfa.send_log(message="=== Testing Method get Credentials for a host NOT SAVED in WFA ===", level='INFO') try: creds = wfa.get_credentials(args['nohost']) except: error = traceback.format_exc() wfa.send_log(message=error, level='ERROR') #Test get connection details: wfa.send_log(message="=== Testing Method get_connection_details for a saved host. ===", level='INFO') details = wfa.get_connection_details(args['host']) wfa.send_log(level='INFO', message=details ) wfa.send_log(message="=== Testing Method get_wfa_input_password. ===", level='INFO') pass1 = wfa.get_wfa_input_password(args['param_password']) wfa.send_log( level='INFO', message='Password in plain text: ' + pass1) #Test Add WFA Workflow params. wfa.send_log(message="=== Testing Method add_wfa_workflow_parameter and get_wfa_workflow_parameter ===", level='INFO') wfa.add_wfa_workflow_parameter('name1', 'value1', add_return=True) val = wfa.get_wfa_workflow_parameter('name1') wfa.send_log( level='INFO', message='Name %s, Value %s' % ('name1', val)) wfa.send_log(message="Overwriting the value of name1", level='INFO') wfa.add_wfa_workflow_parameter('name1', 'value2', add_return=True) val = wfa.get_wfa_workflow_parameter('name1') wfa.send_log( level='INFO', message='Name %s, Value %s' % ('name1', val)) wfa.send_log(message="Adding another Parameter name3, value3.", level='INFO') wfa.add_wfa_workflow_parameter('name3', 'value3', add_return=True) #Now again obtain the values of name1, name3 val = wfa.get_wfa_workflow_parameter('name1') wfa.send_log( level='INFO', message='Name %s, Value %s' % ('name1', val)) val = wfa.get_wfa_workflow_parameter('name3') wfa.send_log( level='INFO', message='Name %s, Value %s' % ('name3', val)) wfa.send_log(message="=== Testing Method get_wfa_rest_parameter. ===", level='INFO') workflowName = wfa.get_wfa_rest_parameter('workflowName') jobId = wfa.get_wfa_rest_parameter('workflowName') userName = wfa.get_wfa_rest_parameter('userName') commandName = wfa.get_wfa_rest_parameter('commandName') workflowId = wfa.get_wfa_rest_parameter('workflowId') wfa.send_log( level='INFO', message='workflowName: %s, jobId: %s, userName: %s, commandName: %s, workflowId: %s' % (workflowName, jobId, userName, commandName, workflowId)) wfa.send_log(message="=== Testing Method get_wfa_version. ===", level='INFO') wfaver = wfa.get_wfa_version(full=True) wfa.send_log(message='FULL VERSION: ' + str(wfaver), level='INFO') wfaver_short = wfa.get_wfa_version() wfa.send_log(message='Short VERSION: ' + str(wfaver_short), level='INFO') wfa.send_log(message="=== Testing Method get_wfa_vendor. ===", level='INFO') wfa.send_log(message='VENDOR: ' + str(wfa.get_wfa_vendor()), level='INFO') wfa.send_log(message="=== Testing Method get_wfa_global_configuration. ===", level='INFO') conf = wfa.get_wfa_global_configuration('company') wfa.send_log( level='INFO', message=conf)