blob: 1ba403d0ed69affde8b1d48f33ac9bbff735384e [file] [log] [blame]
adminbae64d82013-08-01 10:50:15 -07001#!/usr/bin/env python
2'''
3Created on 23-Oct-2012
4
5@authors: Anil Kumar (anilkumar.s@paxterrasolutions.com),
6 Raghav Kashyap(raghavkashyap@paxterrasolutions.com)
7
8
9
10 TestON is free software: you can redistribute it and/or modify
11 it under the terms of the GNU General Public License as published by
12 the Free Software Foundation, either version 2 of the License, or
13 (at your option) any later version.
14
15 TestON is distributed in the hope that it will be useful,
16 but WITHOUT ANY WARRANTY; without even the implied warranty of
17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 GNU General Public License for more details.
19
20 You should have received a copy of the GNU General Public License
21 along with TestON. If not, see <http://www.gnu.org/licenses/>.
22
23
24Utilities will take care about the basic functions like :
25 * Extended assertion,
26 * parse_args for key-value pair handling
27 * Parsing the params or topology file.
28
29'''
30import re
31from configobj import ConfigObj
adminbae64d82013-08-01 10:50:15 -070032from core import ast as ast
33import smtplib
34
adminbae64d82013-08-01 10:50:15 -070035import email
36import os
37import email.mime.application
38
39class Utilities:
40 '''
41 Utilities will take care about the basic functions like :
42 * Extended assertion,
43 * parse_args for key-value pair handling
44 * Parsing the params or topology file.
45 '''
46
47 def __init__(self):
48 self.wrapped = sys.modules[__name__]
49
50 def __getattr__(self, name):
51 '''
52 This will invoke, if the attribute wasn't found the usual ways.
53 Here it will look for assert_attribute and will execute when AttributeError occurs.
54 It will return the result of the assert_attribute.
55 '''
56 try:
57 return getattr(self.wrapped, name)
58 except AttributeError:
59 def assertHandling(**kwargs):
60 nameVar = re.match("^assert",name,flags=0)
61 matchVar = re.match("assert(_not_|_)(equals|matches|greater|lesser)",name,flags=0)
62 notVar = 0
63 operators = ""
64
65 try :
66 if matchVar.group(1) == "_not_" and matchVar.group(2) :
67 notVar = 1
68 operators = matchVar.group(2)
69 elif matchVar.group(1) == "_" and matchVar.group(2):
70 operators = matchVar.group(2)
adminbae64d82013-08-01 10:50:15 -070071 except AttributeError:
72 if matchVar==None and nameVar:
73 operators ='equals'
Jon Hall79bec222015-04-30 16:23:30 -070074 result = self._assert(NOT=notVar,operator=operators,**kwargs)
adminbae64d82013-08-01 10:50:15 -070075 if result == main.TRUE:
76 main.log.info("Assertion Passed")
Jon Hall79bec222015-04-30 16:23:30 -070077 main.STEPRESULT = main.TRUE
adminbae64d82013-08-01 10:50:15 -070078 elif result == main.FALSE:
79 main.log.warn("Assertion Failed")
Jon Hall79bec222015-04-30 16:23:30 -070080 main.STEPRESULT = main.FALSE
81 else:
adminbae64d82013-08-01 10:50:15 -070082 main.log.error("There is an Error in Assertion")
Jon Hall79bec222015-04-30 16:23:30 -070083 main.STEPRESULT = main.ERROR
adminbae64d82013-08-01 10:50:15 -070084 return result
adminbae64d82013-08-01 10:50:15 -070085 return assertHandling
Jon Hall79bec222015-04-30 16:23:30 -070086
Jon Hall8ce34e82015-06-05 10:41:45 -070087 def _assert (self,**assertParam):
adminbae64d82013-08-01 10:50:15 -070088 '''
89 It will take the arguments :
Jon Hall8ce34e82015-06-05 10:41:45 -070090 expect:'Expected output'
91 actual:'Actual output'
adminbae64d82013-08-01 10:50:15 -070092 onpass:'Action or string to be triggered or displayed respectively when the assert passed'
93 onfail:'Action or string to be triggered or displayed respectively when the assert failed'
94 not:'optional argument to specify the negation of the each assertion type'
95 operator:'assertion type will be defined by using operator. Like equal , greater, lesser, matches.'
Jon Hall8ce34e82015-06-05 10:41:45 -070096
adminbae64d82013-08-01 10:50:15 -070097 It will return the assertion result.
Jon Hall8ce34e82015-06-05 10:41:45 -070098
adminbae64d82013-08-01 10:50:15 -070099 '''
Jon Hall8ce34e82015-06-05 10:41:45 -0700100
adminbae64d82013-08-01 10:50:15 -0700101 arguments = self.parse_args(["EXPECT","ACTUAL","ONPASS","ONFAIL","NOT","OPERATOR"],**assertParam)
Jon Hall8ce34e82015-06-05 10:41:45 -0700102
adminbae64d82013-08-01 10:50:15 -0700103 result = 0
104 valuetype = ''
105 operation = "not "+ str(arguments["OPERATOR"]) if arguments['NOT'] and arguments['NOT'] == 1 else arguments["OPERATOR"]
106 operators = {'equals':{'STR':'==','NUM':'=='}, 'matches' : '=~', 'greater':'>' ,'lesser':'<'}
Jon Hall8ce34e82015-06-05 10:41:45 -0700107
adminbae64d82013-08-01 10:50:15 -0700108 expectMatch = re.match('^\s*[+-]?0(e0)?\s*$', str(arguments["EXPECT"]), re.I+re.M)
109 if not ((not expectMatch) and (arguments["EXPECT"]==0)):
110 valuetype = 'NUM'
111 else :
112 if arguments["OPERATOR"] == 'greater' or arguments["OPERATOR"] == 'lesser':
113 main.log.error("Numeric comparison on strings is not possibele")
114 return main.ERROR
Jon Hall8ce34e82015-06-05 10:41:45 -0700115
adminbae64d82013-08-01 10:50:15 -0700116 valuetype = 'STR'
117 arguments["ACTUAL"] = str(arguments["ACTUAL"])
118 if arguments["OPERATOR"] != 'matches':
119 arguments["EXPECT"] = str(arguments["EXPECT"])
Jon Hall8ce34e82015-06-05 10:41:45 -0700120
adminbae64d82013-08-01 10:50:15 -0700121 try :
122 opcode = operators[str(arguments["OPERATOR"])][valuetype] if arguments["OPERATOR"] == 'equals' else operators[str(arguments["OPERATOR"])]
Jon Hall8ce34e82015-06-05 10:41:45 -0700123
adminbae64d82013-08-01 10:50:15 -0700124 except KeyError:
125 print "Key Error in assertion"
126 return main.FALSE
Jon Hall8ce34e82015-06-05 10:41:45 -0700127
adminbae64d82013-08-01 10:50:15 -0700128 if opcode == '=~':
129 try:
130 assert re.search(str(arguments["EXPECT"]),str(arguments["ACTUAL"]))
131 result = main.TRUE
132 except AssertionError:
133 try :
Jon Hall8ce34e82015-06-05 10:41:45 -0700134 assert re.match(str(arguments["EXPECT"]),str(arguments["ACTUAL"]))
adminbae64d82013-08-01 10:50:15 -0700135 result = main.TRUE
136 except AssertionError:
137 main.log.error("Assertion Failed")
138 result = main.FALSE
adminbae64d82013-08-01 10:50:15 -0700139 else :
140 try:
141 if str(opcode)=="==":
142 main.log.info("Verifying the Expected is equal to the actual or not using assert_equal")
143 if (arguments["EXPECT"] == arguments["ACTUAL"]):
144 result = main.TRUE
145 else :
146 result = main.FALSE
adminbae64d82013-08-01 10:50:15 -0700147 elif str(opcode) == ">":
148 main.log.info("Verifying the Expected is Greater than the actual or not using assert_greater")
149 if (ast.literal_eval(arguments["EXPECT"]) > ast.literal_eval(arguments["ACTUAL"])) :
150 result = main.TRUE
151 else :
152 result = main.FALSE
adminbae64d82013-08-01 10:50:15 -0700153 elif str(opcode) == "<":
154 main.log.info("Verifying the Expected is Lesser than the actual or not using assert_lesser")
155 if (ast.literal_eval(arguments["EXPECT"]) < ast.literal_eval(arguments["ACTUAL"])):
156 result = main.TRUE
157 else :
158 result = main.FALSE
adminbae64d82013-08-01 10:50:15 -0700159 except AssertionError:
160 main.log.error("Assertion Failed")
161 result = main.FALSE
adminbae64d82013-08-01 10:50:15 -0700162 result = result if result else 0
163 result = not result if arguments["NOT"] and arguments["NOT"] == 1 else result
164 resultString = ""
165 if result :
166 resultString = str(resultString) + "PASS"
167 main.log.info(arguments["ONPASS"])
168 else :
169 resultString = str(resultString) + "FAIL"
170 if not isinstance(arguments["ONFAIL"],str):
171 eval(str(arguments["ONFAIL"]))
172 else :
173 main.log.error(arguments["ONFAIL"])
174 main.log.report(arguments["ONFAIL"])
Jon Hall90627612015-06-09 14:57:02 -0700175 main.onFailMsg = arguments[ 'ONFAIL' ]
Jon Hall8ce34e82015-06-05 10:41:45 -0700176
adminbae64d82013-08-01 10:50:15 -0700177 msg = arguments["ON" + str(resultString)]
178
179 if not isinstance(msg,str):
180 try:
181 eval(str(msg))
182 except SyntaxError:
183 print "functin definition is not write"
184
185 main.last_result = result
186 return result
Jon Hall8ce34e82015-06-05 10:41:45 -0700187
adminbae64d82013-08-01 10:50:15 -0700188 def parse_args(self,args, **kwargs):
189 '''
190 It will accept the (key,value) pair and will return the (key,value) pairs with keys in uppercase.
191 '''
192 newArgs = {}
193 for key,value in kwargs.iteritems():
194 #currentKey = str.upper(key)
195 if isinstance(args,list) and str.upper(key) in args:
196 for each in args:
197 if each==str.upper(key):
198 newArgs [str(each)] = value
199 elif each != str.upper(key) and (newArgs.has_key(str(each)) == False ):
200 newArgs[str(each)] = None
201
202
203
204 return newArgs
205
206 def send_mail(self):
207 # Create a text/plain message
208 msg = email.mime.Multipart.MIMEMultipart()
209 try :
210 if main.test_target:
211 sub = "Result summary of \""+main.TEST+"\" run on component \""+main.test_target+"\" Version \""+vars(main)[main.test_target].get_version()+"\": "+str(main.TOTAL_TC_SUCCESS)+"% Passed"
212 else :
213 sub = "Result summary of \""+main.TEST+"\": "+str(main.TOTAL_TC_SUCCESS)+"% Passed"
214 except KeyError,AttributeError:
215 sub = "Result summary of \""+main.TEST+"\": "+str(main.TOTAL_TC_SUCCESS)+"% Passed"
216
217 msg['Subject'] = sub
218 msg['From'] = 'paxweb@paxterrasolutions.com'
219 msg['To'] = main.mail
220 #msg['Cc'] = 'paxweb@paxterrasolutions.com'
221
222 # The main body is just another attachment
223 body = email.mime.Text.MIMEText(main.logHeader+"\n"+main.testResult)
224 msg.attach(body)
225
226 # Attachment
227 for filename in os.listdir(main.logdir):
228 filepath = main.logdir+"/"+filename
229 fp=open(filepath,'rb')
230 att = email.mime.application.MIMEApplication(fp.read(),_subtype="")
231 fp.close()
232 att.add_header('Content-Disposition','attachment',filename=filename)
233 msg.attach(att)
234
235 smtp = smtplib.SMTP('198.57.211.46')
236 smtp.starttls()
237 smtp.login('paxweb@paxterrasolutions.com','pax@peace')
238 smtp.sendmail(msg['From'],[msg['To']], msg.as_string())
239 smtp.quit()
240 return main.TRUE
241
242
243 def parse(self,fileName):
244 '''
245 This will parse the params or topo or cfg file and return content in the file as Dictionary
246 '''
247 self.fileName = fileName
248 matchFileName = re.match(r'(.*)\.(cfg|params|topo)',self.fileName,re.M|re.I)
249 if matchFileName:
250 try :
251 parsedInfo = ConfigObj(self.fileName)
252 return parsedInfo
Jon Hallfebb1c72015-03-05 13:30:09 -0800253 except Exception:
adminbae64d82013-08-01 10:50:15 -0700254 print "There is no such file to parse "+fileName
255 else:
256 return 0
257
258
259if __name__ != "__main__":
260 import sys
261
262 sys.modules[__name__] = Utilities()