Pytest lambda 处理程序传递事件和上下文

Question

我正在使用pytest为我的lambda函数编写单元测试。我不知道应该如何将事件参数传递给函数调用。我了解到可以使用@pytest.fixture来实现。我对Python和pytest非常陌生。相信我以错误的方式使用固定装置。请帮帮我！！

下面是我的 lambda 处理程序：

lambda_service.py

def lambda_handler(event, context):   
logger.info('Event received: ' + json.dumps(event))
try:
sort = (event['sort'])
size = int(event['size'])
page = int(event['page'])
list_response = MyService().get_people_list(sort, size, page)
logger.info(list_response)
except Exception as e:
logger.error("Unable to fetch details")
logger.exception(e)
return list_response

这是我的测试课——

class TestServiceHandler:
@pytest.fixture
def event(self):
return {
"sort": "asc",
"size": 5,
"page": 0
}
@pytest.fixture
def context(self):
return None
def test_lambda_handler(self):
result = lambda_service.lambda_handler(self.event, self.context)
assert_valid_schema(result, 'vendor_list.json')

运行此测试时我得到以下错误

line 17, in lambda_handler
sort = (event['sort'])nTypeError: 'method' object is not subscriptable"

虽然我在赛程中传递事件和背景，但它仍然指的是lambda_handler内部的event[sort]。

Answer 1

您已经正确定义了夹具，但使用错误。若要修复，请将参数添加到与夹具完全相同的test_lambda_handler方法。运行测试时，pytest将分析每个参数并插入夹具返回值(如果它可以找到具有该名称的夹具(。例：

class TestServiceHandler:
@pytest.fixture
def event(self):
...
@pytest.fixture
def context(self):
...
def test_lambda_handler(self, event, context):
result = lambda_service.lambda_handler(event, context)
assert ...