如何Mock AWSwrangler 对Lambda的Athena查询逻辑进行单元测试

方案1：直接Mock AWSwrangler的Athena查询方法（最轻量，推荐单元测试用） #

不用依赖moto这类AWS模拟工具，单测时直接替换wr.athena.read_sql_query方法的返回值即可，完全不会触发真实AWS请求。
以下是用pytest + pytest-mock实现的测试代码示例：

import pandas as pd
from your_module import athena_query # 导入你自己的业务函数

def test_athena_query(mocker):
    # 构造期望返回的模拟结果数据
    mock_df = pd.DataFrame([
        {"contactid": "123", "col1": "test_value1", "col2": "test_value2"}
    ])
    # 替换awswrangler的查询方法，固定返回上面构造的模拟数据
    mock_read_sql = mocker.patch("awswrangler.athena.read_sql_query", return_value=mock_df)
    
    # 调用业务函数执行测试
    result = athena_query("table_name", "123", "s3://bucket", None)
    
    # 断言查询方法的入参符合预期
    mock_read_sql.assert_called_once()
    call_params = mock_read_sql.call_args[1]
    assert call_params["params"]["contactid"] == "'123'"
    assert call_params["params"]["dbtable"] == "table_name"
    # 断言返回结果符合预期
    assert result["contactid"] == "123"
    assert result["col1"] == "test_value1"

该方案不需要启动任何额外服务，单测运行速度最快，完全隔离外部依赖。

方案2：用LocalStack模拟完整AWS服务链路（适合集成测试用） #

如果需要测试完整的Athena查询、S3输出、元数据读取等全链路逻辑，可以用LocalStack代替moto：

1. 本地启动LocalStack服务，开启Athena、S3、Glue组件
2. 测试时构造的boto3 session指定endpoint_url为LocalStack的本地地址，通常为http://localhost:4566
3. 提前在LocalStack中创建对应的Glue库、表、写入模拟数据，再调用业务函数，AWSwrangler会自动把请求打到LocalStack而不是真实AWS服务

额外注意 #

你贴的业务代码中SQL语法存在错误：FROM :dbtable;后多了一个分号，会打断SQL结构导致查询报错，建议去掉该分号。

内容的提问来源于stack exchange，提问作者naresh_321