Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from flask.ext.testing import TestCase
- from flask import Flask
- from Shares import db, app
- from Shares.models import User
- import manage
- class test(TestCase):
- def create_app(self):
- app = Flask(__name__)
- app.config['TESTING'] = True
- app.config['WTF_CSRF_ENABLED'] = False
- return app
- SQLALCHEMY_DATABASE_URI = "sqlite://"
- TESTING = True
- def setUp(self):
- manage.initdb()
- print self.login('lucas', 'test').data
- def tearDown(self):
- db.session.remove()
- db.drop_all()
- def login(self, username, password):
- return self.client.post('/login', data=dict(
- username=username,
- password=password
- ), follow_redirects=True)
- def logout(self):
- return self.client.get('/logout', follow_redirects=True)
- def test_adduser(self):
- lucas=User(username="lucas", email="lucas@example.com", password="test")
- user2 = User(username="lucas", email="lucas@test.com")
- db.session.add(lucas)
- db.session.commit()
- assert lucas in db.session
- assert user2 not in db.session
- def test_login(self):
- rv = self.login('lucas', 'test')
- assert 'Welcome' in rv.data
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement