2017-07-01 17:48:18 -07:00
|
|
|
# kappa123/server/tests/test_main.py
|
2015-01-21 08:32:17 -08:00
|
|
|
|
|
|
|
|
|
|
|
import unittest
|
|
|
|
|
|
|
|
from base import BaseTestCase
|
|
|
|
|
|
|
|
|
|
|
|
class TestMainBlueprint(BaseTestCase):
|
|
|
|
|
|
|
|
def test_index(self):
|
|
|
|
# Ensure Flask is setup.
|
|
|
|
response = self.client.get('/', follow_redirects=True)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
2015-04-16 16:06:17 -07:00
|
|
|
self.assertIn(b'Welcome!', response.data)
|
|
|
|
self.assertIn(b'Register/Login', response.data)
|
2015-01-21 08:32:17 -08:00
|
|
|
|
|
|
|
def test_about(self):
|
|
|
|
# Ensure about route behaves correctly.
|
|
|
|
response = self.client.get('/about', follow_redirects=True)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
2015-04-16 16:06:17 -07:00
|
|
|
self.assertIn(b'About', response.data)
|
2015-01-21 08:32:17 -08:00
|
|
|
|
|
|
|
def test_404(self):
|
|
|
|
# Ensure 404 error is handled.
|
|
|
|
response = self.client.get('/404')
|
|
|
|
self.assert404(response)
|
|
|
|
self.assertTemplateUsed('errors/404.html')
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
unittest.main()
|