Pytest登录测试断言错误：正确凭据返回False问题排查

嘿，咱们来拆解下你的问题，看看到底哪里出了问题：你写的两个登录测试里，第一个用正确凭据的断言明明通过了，但第二个用错误凭据的断言却失败了，而且报错信息看起来像是正确凭据的测试突然返回了False——这其实是个小迷惑点，真正的问题根源在你的login_user方法里的判断逻辑上。

核心问题出在判断逻辑上 #

你原来的判断代码是这样的：

if request.headers['Set-Cookie'] in request.headers: 
    return True 
return False

这里有两个致命的逻辑错误：

1. 直接取键会触发异常：如果request.headers里没有Set-Cookie这个键，你直接写request.headers['Set-Cookie']会抛出KeyError，根本走不到返回False的分支
2. 判断逻辑完全偏离需求：request.headers['Set-Cookie'] in request.headers这句话的意思是「检查Set-Cookie对应的值是否是请求头的键」，这完全不是你想要的「检查Set-Cookie这个键是否存在于头信息里」的逻辑！

举个例子，假设头信息里有Set-Cookie: sessionid=xxx，那request.headers['Set-Cookie']的值是sessionid=xxx，你去检查这个字符串是不是头信息的键，结果肯定是False——这就是为什么你看到的报错里，正确凭据的测试会返回False。

一步步排查的思路 #

1. 单独验证判断逻辑：把判断逻辑抽出来写个小脚本，一眼就能看出问题：

   # 模拟登录成功的头信息（注意：正常应该是响应头，不是请求头！）
   success_headers = {'Set-Cookie': 'sessionid=abc123'}
   # 原来的错误逻辑
   print(success_headers['Set-Cookie'] in success_headers)  # 输出False，因为'sessionid=abc123'不是键
   # 正确的逻辑
   print('Set-Cookie' in success_headers)  # 输出True，这才是你要的判断
   
   # 模拟登录失败的情况
   fail_headers = {}
   # 原来的逻辑会直接抛KeyError
   try:
       print(fail_headers['Set-Cookie'] in fail_headers)
   except KeyError:
       print("触发了KeyError！")
   # 正确的逻辑
   print('Set-Cookie' in fail_headers)  # 输出False
   

2. 仔细解读测试报错：你看到的报错信息其实是测试框架的排版小问题，它可能重复显示了第一个断言的代码，但实际是第二个断言执行时，错误的判断逻辑（或KeyError被捕获后返回了错误值）导致断言失败。

已验证的解决方案 #

把你的判断逻辑改成直接检查键是否存在：

if 'Set-Cookie' in request.headers:
    return True
return False

或者更简洁的写法：

return 'Set-Cookie' in request.headers

另外提个小细节：从你提供的login接口代码来看，登录成功时是在响应对象里设置Cookie，正常来说你应该检查的是response.headers而不是request.headers——不过你说修改判断条件后问题解决了，可能你的代码里把响应对象命名成了request？这个小细节后续可以留意下，避免混淆。

修改后的判断逻辑已经能正确识别是否存在Cookie，你的两个测试就能正常通过了。

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