python - 测试登录所需的 Flask 应用程序

Question

我正在为烧瓶 Web 应用程序编写测试，并且每个测试都需要用户登录是否有任何方法可以定义测试功能并在其他测试中使用它？这是我当前的测试用例，我想只使用 post to API URL 来实现登录功能

class TestLogin(unittest.TestCase):
   def setUp(self):
       self.headers = {'Content-type':'application/json','Accept':'tex/plain')
   def test_login(self):
      s=request.Session()
      url = 'http://localhost/api/v1/user/login/'
      data = {'username' = 'admin','password' :'123123'}
      r=s.post(url,data=json.dumps(data),headers=self.headers)
      self.assertEqual(r.status_code,200)

Answer 1

您可以创建一个单独的登录功能。例如

  def login():
      s = request.Session()
      url = 'http://localhost/api/v1/user/login/'
      data = {'username' = 'admin','password' :'123123'}
      r=s.post(url,data=json.dumps(data),headers=self.headers)
      return (r.status_code == 200)

然后从您的测试中调用它：

 def test_something(self):
      if(login()):
         #Test stuff