From 4a5b6f807ba8ed941b286f04d712f88734ff3c8e Mon Sep 17 00:00:00 2001
From: zy7y <13271962515@163.com>
Date: Mon, 17 May 2021 16:14:20 +0800
Subject: [PATCH] =?UTF-8?q?update=20tools/data=5Fprocess.py.=20sql?=
 =?UTF-8?q?=E6=A0=8F=E6=94=AF=E6=8C=81=E5=A4=9Asql=E8=AF=AD=E5=8F=A5?=
 =?UTF-8?q?=E6=89=A7=E8=A1=8C=EF=BC=88=E6=9F=A5=E8=AF=A2=EF=BC=89=EF=BC=8C?=
 =?UTF-8?q?=E5=A4=9A=E6=9D=A1sql=E4=B9=8B=E9=97=B4=E4=BD=BF=E7=94=A8?=
 =?UTF-8?q?=E8=8B=B1=E6=96=87=20;=20=E7=BB=93=E5=B0=BE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tools/data_process.py | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/tools/data_process.py b/tools/data_process.py
index 4ea2632..5842380 100644
--- a/tools/data_process.py
+++ b/tools/data_process.py
@@ -87,14 +87,19 @@ class DataProcess:
         allure_step('运行sql', sql)
         logger.info(sql)
         if sql is not None:
-            # 查后置sql
-            result = db.fetch_one(sql)
-            allure_step('sql执行结果', result)
-            logger.info(f'结果：{result}')
-            if result is not None:
-                # 将查询结果添加到响应字典里面，作用在，接口响应的内容某个字段 直接和数据库某个字段比对，在预期结果中
-                # 使用同样的语法提取即可
-                DataProcess.response_dict.update(result)
+            # 多条sql，在用例中填写方式如下select * from user; select * from goods 每条sql语句之间需要使用 ; 来分割
+            for sql in sql.split(";"):
+                sql = sql.strip()
+                if sql == '':
+                    continue
+                # 查后置sql
+                result = db.fetch_one(sql)
+                allure_step('sql执行结果', result)
+                logger.info(f'结果：{result}')
+                if result is not None:
+                    # 将查询结果添加到响应字典里面，作用在，接口响应的内容某个字段 直接和数据库某个字段比对，在预期结果中
+                    # 使用同样的语法提取即可
+                    DataProcess.response_dict.update(result)
 
     @classmethod
     def assert_result(cls, response: dict, expect_str: str):