Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def TestLoginSuccessful(self):
- with mock.patch('CaseLoader_Fl.models.ndems.User') as mocksql:
- mocksql.query.filter_by('').first.return_value = ('test_user','test_password')
- is_validated = login_controller.validate_user('test_user','test_password')
- self.assertTrue(is_validated)
- def validate_user(username,password):
- user = User.query.filter_by(username=username).first()
- print(user)
- if user:
- if user.check_password(password):
- return login_user(user)
- else:
- print('wrong pass')
- return False
- else:
- print('no user')
- return False
- class User(export_db.Model):
- __bind_key__ = 'ndems'
- username = export_db.Column(export_db.String(80), primary_key=True, unique=True)
- password = export_db.Column(export_db.String(80))
- admin = export_db.Column(export_db.Boolean())
- def __init__(self, username, password, admin):
- self.username = username
- self.password = self.set_password(password)
- self.admin = admin
- def __repr__(self):
- return '<User %r>' % self.username
- def set_password(self, password):
- return generate_password_hash(password)
- def check_password(self, password):
- return check_password_hash(self.password, password)
- def is_authenticated(self):
- return True
- def is_active(self):
- return True
- def is_anonymous(self):
- return False
- def get_id(self):
- return str(self.username)
Add Comment
Please, Sign In to add comment