def test_check_args_path(self):
"""
测试路径
:return:
"""
args = {
"online-dsn": '',
"test-dsn": '',
"allow-online-as-test": "false",
"report-type": "markdown",
"query": "select 1;"
}
self.sys_config.set('soar', '')
self.sys_config.get_all_config()
soar = Soar()
args_check_result = soar.check_args(args)
self.assertDictEqual(args_check_result, {
'status': 1,
'msg': '可执行文件路径不能为空！',
'data': {}
})
# 路径不为空
self.sys_config.set('soar', '/opt/archery/src/plugins/soar')
self.sys_config.get_all_config()
soar = Soar()
args_check_result = soar.check_args(args)
self.assertDictEqual(args_check_result, {
'status': 0,
'msg': 'ok',
'data': {}
})
def test_check_args_required(self):
"""
测试必选参数
:return:
"""
args = {
"online-dsn": '',
"test-dsn": '',
"allow-online-as-test": "false",
"report-type": "markdown",
}
self.sys_config.set('soar', '/opt/archery/src/plugins/soar')
self.sys_config.get_all_config()
soar = Soar()
soar.required_args = ['query']
args_check_result = soar.check_args(args)
self.assertDictEqual(args_check_result, {
'status': 1,
'msg': '必须指定query参数',
'data': {}
})
args['query'] = ""
args_check_result = soar.check_args(args)
self.assertDictEqual(args_check_result, {
'status': 1,
'msg': 'query参数值不能为空',
'data': {}
})
def optimize_soar(request):
instance_name = request.POST.get('instance_name')
db_name = request.POST.get('db_name')
sql = request.POST.get('sql')
result = {'status': 0, 'msg': 'ok', 'data': []}
# 服务器端参数验证
if not (instance_name and db_name and sql):
result['status'] = 1
result['msg'] = '页面提交参数可能为空'
return HttpResponse(json.dumps(result), content_type='application/json')
try:
user_instances(request.user, type='all', db_type='mysql').get(instance_name=instance_name)
except Exception:
result['status'] = 1
result['msg'] = '你所在组未关联该实例'
return HttpResponse(json.dumps(result), content_type='application/json')
# 检查测试实例的连接信息和soar程序路径
soar_test_dsn = SysConfig().get('soar_test_dsn')
soar_path = SysConfig().get('soar')
if not (soar_path and soar_test_dsn):
result['status'] = 1
result['msg'] = '请配置soar_path和test_dsn！'
return HttpResponse(json.dumps(result), content_type='application/json')
# 目标实例的连接信息
instance_info = Instance.objects.get(instance_name=instance_name)
online_dsn = "{user}:{pwd}@{host}:{port}/{db}".format(user=instance_info.user,
pwd=instance_info.raw_password,
host=instance_info.host,
port=instance_info.port,
db=db_name)
# 提交给soar获取分析报告
soar = Soar()
# 准备参数
args = {"online-dsn": online_dsn,
"test-dsn": soar_test_dsn,
"allow-online-as-test": "false",
"report-type": "markdown",
"query": sql.strip().replace('"', '\\"').replace('`', '').replace('\n', ' ')
}
# 参数检查
args_check_result = soar.check_args(args)
if args_check_result['status'] == 1:
return HttpResponse(json.dumps(args_check_result), content_type='application/json')
# 参数转换
cmd_args = soar.generate_args2cmd(args, shell=True)
# 执行命令
try:
stdout, stderr = soar.execute_cmd(cmd_args, shell=True).communicate()
result['data'] = stdout if stdout else stderr
except RuntimeError as e:
result['status'] = 1
result['msg'] = str(e)
return HttpResponse(json.dumps(result), content_type='application/json')
def test_check_args_disable(self):
"""
测试禁用参数
:return:
"""
args = {"online-dsn": '',
"test-dsn": '',
"allow-online-as-test": "false",
"report-type": "markdown",
"query": "select 1;"
}
self.sys_config.set('soar', '/opt/archery/src/plugins/soar')
self.sys_config.get_all_config()
soar = Soar()
soar.disable_args = ['allow-online-as-test']
args_check_result = soar.check_args(args)
self.assertDictEqual(args_check_result, {'status': 1, 'msg': 'allow-online-as-test参数已被禁用', 'data': {}})