70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
|
import unittest
|
||
|
|
from wanikaniburned import app, db, User
|
||
|
|
from flask.ext.testing import TestCase
|
||
|
|
|
||
|
|
|
||
|
|
class BaseTestCase(TestCase):
|
||
|
|
|
||
|
|
def create_app(self):
|
||
|
|
app.config.update(
|
||
|
|
TESTING=True,
|
||
|
|
SQLALCHEMY_DATABASE_URI='sqlite:///:memory:',
|
||
|
|
SECRET_KEY="testkey",
|
||
|
|
WTF_CSRF_ENABLED=False,
|
||
|
|
PRESERVE_CONTEXT_ON_EXCEPTION=False,
|
||
|
|
HASH_ROUNDS=1
|
||
|
|
)
|
||
|
|
return app
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
db.create_all()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
db.session.remove()
|
||
|
|
db.drop_all()
|
||
|
|
|
||
|
|
|
||
|
|
class TestLogin(BaseTestCase):
|
||
|
|
|
||
|
|
def login(self):
|
||
|
|
user = User('test', False)
|
||
|
|
return self.client.post('/', data={'api_key': user.api_key})
|
||
|
|
|
||
|
|
def test_user_login(self):
|
||
|
|
response = self.login()
|
||
|
|
self.assertRedirects(response, '/quiz')
|
||
|
|
|
||
|
|
def test_user_logout(self):
|
||
|
|
response = self.login()
|
||
|
|
self.assertRedirects(response, '/quiz')
|
||
|
|
response = self.client.get('/logout')
|
||
|
|
self.assertRedirects(response, '/')
|
||
|
|
|
||
|
|
def test_homepage_without_login(self):
|
||
|
|
response = self.client.get('/')
|
||
|
|
self.assert200(response)
|
||
|
|
self.assertTemplateUsed("welcome.html")
|
||
|
|
|
||
|
|
def test_quiz_inaccessible_without_login(self):
|
||
|
|
response = self.client.get('/quiz')
|
||
|
|
self.assertRedirects(response, '/')
|
||
|
|
|
||
|
|
def test_user_items_inaccessible_without_login(self):
|
||
|
|
response = self.client.get('/user_items')
|
||
|
|
self.assertRedirects(response, '/')
|
||
|
|
|
||
|
|
def test_homepage_with_existing_login(self):
|
||
|
|
response = self.login() # Login
|
||
|
|
self.assertRedirects(response, '/quiz')
|
||
|
|
response = self.client.get('/') # Try to go back to the homepage
|
||
|
|
self.assertRedirects(response, '/quiz')
|
||
|
|
|
||
|
|
def test_user_items_with_existing_login(self):
|
||
|
|
response = self.login() # Login
|
||
|
|
self.assertRedirects(response, '/quiz')
|
||
|
|
response = self.client.get('/user_items')
|
||
|
|
self.assert200(response)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|