python - 装饰两个功能？

Question

是否可以通过使用装饰器或其他任何方法来缩短此类测试用例？

    def test_login_invalid_pwd(self):
        password = '12345'
        response = self._login(pwd=password)
        self.assertEqual(status_code, 200)
        self.assertEqual(response['resultText'],
                         'invalid password or login')
        self.assertEqual(response['resultCode'], 55)

    def test_web_login_invalid_login(self):
        login = 'my_1258@'
        response = self._login(login=login)
        self.assertEqual(status_code, 200)
        self.assertEqual(response['resultText'],
                         'invalid password or login')
        self.assertEqual(response['resultCode'], 55)

Answer 1

是的。尝试这个：

def helper(self, response):
    self.assertEqual(status_code, 200)
    self.assertEqual(response['resultText'],
                     'invalid password or login')
    self.assertEqual(response['resultCode'], 55)

def test_login_invalid_pwd(self):
    password = '12345'
    response = self._login(pwd=password)
    self.helper(response)

def test_web_login_invalid_login(self):
    login = 'my_1258@'
    response = self._login(login=login)
    self.helper(response)

或者，取决于您的测试需要的粒度，并假设您的._login()方法None用作默认值：

def test_login_invalid(self):
    for login, pwd in (('my_1258@', None), (None, '12345')):
        response = self._login(login=login, pwd=pwd)
        self.assertEqual(status_code, 200)
        self.assertEqual(response['resultText'],
                         'invalid password or login')
        self.assertEqual(response['resultCode'], 55)