Advertisement
Guest User

Untitled

a guest
Jan 22nd, 2016
118
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.10 KB | None | 0 0
  1. from flask.ext.testing import TestCase
  2. from flask import Flask
  3. from Shares import db, app
  4. from Shares.models import User
  5. import manage
  6.  
  7.  
  8. class test(TestCase):
  9.  
  10. def create_app(self):
  11. app = Flask(__name__)
  12. app.config['TESTING'] = True
  13. app.config['WTF_CSRF_ENABLED'] = False
  14. return app
  15.  
  16. SQLALCHEMY_DATABASE_URI = "sqlite://"
  17. TESTING = True
  18.  
  19. def setUp(self):
  20.  
  21. manage.initdb()
  22. print self.login('lucas', 'test').data
  23.  
  24. def tearDown(self):
  25.  
  26. db.session.remove()
  27. db.drop_all()
  28.  
  29. def login(self, username, password):
  30. return self.client.post('/login', data=dict(
  31. username=username,
  32. password=password
  33. ), follow_redirects=True)
  34.  
  35. def logout(self):
  36. return self.client.get('/logout', follow_redirects=True)
  37.  
  38. def test_adduser(self):
  39.  
  40. lucas=User(username="lucas", email="lucas@example.com", password="test")
  41. user2 = User(username="lucas", email="lucas@test.com")
  42.  
  43. db.session.add(lucas)
  44. db.session.commit()
  45.  
  46. assert lucas in db.session
  47. assert user2 not in db.session
  48.  
  49. def test_login(self):
  50.  
  51. rv = self.login('lucas', 'test')
  52. assert 'Welcome' in rv.data
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement