def addToken(self, params):
try:
params = json.loads(params)
params.update(dict(token=app.config['TOKEN']))
params = dumps(params)
except:
pass
return params
def prepare(self, sql, case):
if sql:
obj = MysqlDb(app.config['DBNAME'])
value = getattr(obj, sql)()
pre_data = case.get('pre_data')
params = json.loads(case.get('params'))
requestData = params.get('requestData')
requestData.update({pre_data: value})
params.update(dict(requestData=requestData))
case.update(dict(params=dumps(params)))
del obj
def transfer_data(self, case):
try:
params = self.addToken(case.get('params'))
enable = case.get('enable')
code = int(case.get('code'))
headers = case.get('headers')
expected = self.get_response(case.get('expected'))
if params != '' and isinstance(params, str):
params = self.form_to_dict(
params) if 'form' in headers else dumps(
self.data_to_json(params))
else:
params = dumps(params)
headers = self.data_to_json(headers)
case.update(
dict(headers=headers,
params=params,
enable=enable,
code=code,
expected=expected))
except Exception as err:
logging.error("{} error: {}".format(Operator.get_current_func(),
str(err)))
return case