test.py 920 B

12345678910111213141516171819202122232425262728293031323334353637
  1. #!/usr/bin/env python
  2. import unittest
  3. from app import app
  4. class TestApp(unittest.TestCase):
  5. def setUp(self):
  6. self.app = app.test_client()
  7. def test_home_page_works(self):
  8. rv = self.app.get('/')
  9. self.assertTrue(rv.data)
  10. self.assertEquals(rv.status_code, 200)
  11. def test_about_page_works(self):
  12. rv = self.app.get('/about/')
  13. self.assertTrue(rv.data)
  14. self.assertEquals(rv.status_code, 200)
  15. def test_default_redirecting(self):
  16. rv = self.app.get('/about')
  17. self.assertEquals(rv.status_code, 301)
  18. def test_404_page(self):
  19. rv = self.app.get('/i-am-not-found/')
  20. self.assertEquals(rv.status_code, 404)
  21. def test_static_text_file_request(self):
  22. rv = self.app.get('/robots.txt')
  23. self.assertTrue(rv.data)
  24. self.assertEquals(rv.status_code, 200)
  25. if __name__ == '__main__':
  26. unittest.main()