Guest User

Untitled

a guest
Oct 16th, 2017
73
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.48 KB | None | 0 0
  1. def TestLoginSuccessful(self):
  2. with mock.patch('CaseLoader_Fl.models.ndems.User') as mocksql:
  3. mocksql.query.filter_by('').first.return_value = ('test_user','test_password')
  4. is_validated = login_controller.validate_user('test_user','test_password')
  5. self.assertTrue(is_validated)
  6.  
  7. def validate_user(username,password):
  8. user = User.query.filter_by(username=username).first()
  9. print(user)
  10. if user:
  11. if user.check_password(password):
  12. return login_user(user)
  13. else:
  14. print('wrong pass')
  15. return False
  16. else:
  17. print('no user')
  18. return False
  19.  
  20. class User(export_db.Model):
  21. __bind_key__ = 'ndems'
  22. username = export_db.Column(export_db.String(80), primary_key=True, unique=True)
  23. password = export_db.Column(export_db.String(80))
  24. admin = export_db.Column(export_db.Boolean())
  25.  
  26. def __init__(self, username, password, admin):
  27. self.username = username
  28. self.password = self.set_password(password)
  29. self.admin = admin
  30.  
  31. def __repr__(self):
  32. return '<User %r>' % self.username
  33.  
  34. def set_password(self, password):
  35. return generate_password_hash(password)
  36.  
  37. def check_password(self, password):
  38. return check_password_hash(self.password, password)
  39.  
  40. def is_authenticated(self):
  41. return True
  42.  
  43. def is_active(self):
  44. return True
  45.  
  46. def is_anonymous(self):
  47. return False
  48.  
  49. def get_id(self):
  50. return str(self.username)
Add Comment
Please, Sign In to add comment