如何在自动化集成测试中编写针对SQL注入的Unit Test?

胖墨大大_8783

胖墨大大_8783

2026-06-24

240人浏览

原创

sqlmapapi.py不能用于单元测试,因其依赖真实环境且违背单元测试的毫秒级、零依赖、可预测原则;应直接测试dao层参数化逻辑,用mock隔离数据库并验证sql模板与参数分离。

如何在自动化集成测试中编写针对sql注入的unit test?

sqlmapapi.py 不是单元测试工具,它跑的是真实扫描任务,和单元测试目标完全冲突。想在 UnitTest 里验证 SQL 注入防护逻辑,必须剥离数据库、网络、外部命令等所有真实依赖。

为什么不能用 sqlmapapi.py 做单元测试

单元测试要求「毫秒级执行」「零外部依赖」「可重复、可预测」。sqlmapapi.py 启动的是完整 HTTP 服务,依赖 Python 环境、Sqlmap 二进制、网络端口、HTTP 轮询、结果解析——这些全是集成行为。哪怕你 mock 了 requests.post,也掩盖了真正要测的点:你的代码是否安全拼接了用户输入。

真正该测的是 DAO 层的参数化逻辑

SQL 注入防护失效,99% 发生在数据访问层(DAO)对用户输入的处理上。单元测试要直接覆盖这一层,而不是绕到外部工具去“间接验证”。

  • 测 get_user_by_id 方法是否用了 cursor.execute("SELECT * FROM users WHERE id = %s", (user_id,)),而不是 f"WHERE id = {user_id}"
  • 测 search_products 是否把 keyword 传给了 execute 的参数元组,而不是字符串格式化
  • 测任何包含 format、%、f-string 拼接 SQL 的函数——它们应该被标记为 禁止出现,并在 CI 中用静态检查(如 bandit)拦截

用 Mock 隔离数据库驱动,只验证 SQL 构造行为

你不需连接真实 MySQL,只需确认你的 DAO 方法调用了正确的 API,并传入了预期参数:

倍塔塞司
倍塔塞司

一款聚焦职业发展与职业测评的 AI 服务平台,提供职业规划、个性化测评及相关 AI 工具,帮助用户梳理职业方向与发展需求。

下载
def test_get_user_safe_with_parametrized_query():
    mock_cursor = MagicMock()
    mock_conn = MagicMock()
    mock_conn.cursor.return_value = mock_cursor
<pre class="brush:php;toolbar:false;"># 模拟用户输入
user_input = "1' OR '1'='1"

# 执行被测方法
result = get_user_by_id(mock_conn, user_input)

# 断言:SQL 字符串里没有 user_input 的原始值
mock_cursor.execute.assert_called_once()
executed_sql = mock_cursor.execute.call_args[0][0]
assert "OR" not in executed_sql  # 不是拼接,是占位符
assert mock_cursor.execute.call_args[0][1] == (user_input,)  # 参数单独传入

关键点:mock_cursor.execute.call_args[0][1] 是第二个参数(即参数元组),它必须包含原始输入;而第一个参数 executed_sql 必须是纯模板,不含任何用户数据。

边界输入必须覆盖,但不用 sqlmap

单元测试不模拟攻击,而是验证防御逻辑是否覆盖常见注入载荷:

  • "admin' -- " → 应触发参数化,不报语法错误
  • "1 OR 1=1" → 查询应返回 0 条或 1 条(取决于是否存在 ID=1 的记录),而非全表
  • "'; DROP TABLE users; -- " → 不应执行第二条语句,也不应抛出 ProgrammingError(说明未参数化)

这些输入直接喂给 DAO 方法,观察返回值或异常类型即可。不需要启动服务、发 HTTP 请求、解析 json 结果——那已经是集成测试的事了。

真正的难点不在写断言,而在于让团队接受:SQL 注入不是靠外部扫描器“发现”,而是靠代码层的参数化约定 + 单元测试强制兜底。一旦允许字符串拼接进 SQL,再强的 sqlmap 报告也只是马后炮。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

sql注入

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
数据分析工具有哪些
数据分析工具有哪些

数据分析工具有Excel、SQL、Python、R、Tableau、Power BI、SAS、SPSS和MATLAB等。详细介绍:1、Excel,具有强大的计算和数据处理功能;2、SQL,可以进行数据查询、过滤、排序、聚合等操作;3、Python,拥有丰富的数据分析库;4、R,拥有丰富的统计分析库和图形库;5、Tableau,提供了直观易用的用户界面等等。

2023.10.12

4143

8

SQL中distinct的用法
SQL中distinct的用法

SQL中distinct的语法是“SELECT DISTINCT column1, column2,...,FROM table_name;”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.27

891

4

SQL中months_between使用方法
SQL中months_between使用方法

在SQL中,MONTHS_BETWEEN 是一个常见的函数,用于计算两个日期之间的月份差。想了解更多SQL的相关内容,可以阅读本专题下面的文章。

2024.02.23

1069

5

SQL出现5120错误解决方法
SQL出现5120错误解决方法

SQL Server错误5120是由于没有足够的权限来访问或操作指定的数据库或文件引起的。想了解更多sql错误的相关内容,可以阅读本专题下面的文章。

2024.03.06

6041

10

sql procedure语法错误解决方法
sql procedure语法错误解决方法

sql procedure语法错误解决办法:1、仔细检查错误消息;2、检查语法规则;3、检查括号和引号;4、检查变量和参数;5、检查关键字和函数;6、逐步调试;7、参考文档和示例。想了解更多语法错误的相关内容,可以阅读本专题下面的文章。

2024.03.06

2923

4

oracle数据库运行sql方法
oracle数据库运行sql方法

运行sql步骤包括:打开sql plus工具并连接到数据库。在提示符下输入sql语句。按enter键运行该语句。查看结果,错误消息或退出sql plus。想了解更多oracle数据库的相关内容,可以阅读本专题下面的文章。

2024.04.07

6020

11

sql中where的含义
sql中where的含义

sql中where子句用于从表中过滤数据,它基于指定条件选择特定的行。想了解更多where的相关内容,可以阅读本专题下面的文章。

2024.04.29

8061

6

sql中删除表的语句是什么
sql中删除表的语句是什么

sql中用于删除表的语句是drop table。语法为drop table table_name;该语句将永久删除指定表的表和数据。想了解更多sql的相关内容,可以阅读本专题下面的文章。

2024.04.29

1090

5

sql中删除一列的命令是什么
sql中删除一列的命令是什么

在sql中,使用alter table语句可以删除一列,语法为:alter table table_name drop column column_name。想了解更多sql的相关内容,可以阅读本专题下面的文章。

2024.04.29

952

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
PHP开发基础之数据库篇(PDO)
PHP开发基础之数据库篇(PDO)

共10课时 | 2.3万人学习

文心快码功能介绍
文心快码功能介绍

共0课时 | 0人学习

文心快码Agent使用手册
文心快码Agent使用手册

共0课时 | 0人学习