unittest写个登录的用例
时间: 2023-08-16 14:06:41 浏览: 35
好的,下面是一个简单的登录用例示例:
```python
import unittest
class TestLogin(unittest.TestCase):
def test_valid_login(self):
# 假设用户名和密码正确,登录成功
result = login('username', 'password')
self.assertTrue(result)
def test_invalid_login(self):
# 假设用户名和密码不正确,登录失败
result = login('invalid_username', 'invalid_password')
self.assertFalse(result)
def login(username, password):
# 实现登录逻辑,返回True表示登录成功,False表示登录失败
if username == 'username' and password == 'password':
return True
else:
return False
if __name__ == '__main__':
unittest.main()
```
在上面的示例中,我们定义了一个TestLogin类,其中包含了两个测试方法:test_valid_login和test_invalid_login。这两个方法分别测试了登录成功和登录失败的情况。在每个测试方法中,我们调用login函数进行登录,并使用assertTrue和assertFalse断言来验证登录结果是否符合我们的预期。
最后,在if __name__ == '__main__':这一行代码中,我们使用unittest.main()方法来运行测试用例。