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

时间:2015-01-31 21:38:09

标签: python flask flask-testing

我正在为烧瓶网络应用程序编写测试,每个测试都需要登录用户 是否有任何定义测试功能的方法并在其他测试中使用它? 这是我目前的TestCase,我想使用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)

1 个答案:

答案 0 :(得分:1)

您可以创建单独的登录功能。 E.g

  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