def validate_credentials(self, email, password): user = User.select().where(fn.lower(User.email) == email.lower()).first() if not user: raise self.UserDoesNotExist if not self.password_match(user, password): raise self.InvalidPassword return user
def there_are_users(self): return User.select().exists()
def email_taken(self, email): return User.select().where( fn.lower(User.email) == email.lower()).exists()
def register_user(self, email, password): self.validate_data(email, password) hashed_password = self.hash_password(password) type = User.ADMIN if not self.there_are_users() else User.EMPLOYEE return User.create(email=email, password=hashed_password, type=type)
def tearDown(self): User.delete().execute()
def get_current_user(self, token): session = Session.select().where(Session.token == token).first() if session: return User.get(User.id == session.user_id) return None
def tearDownClass(self): super(UserTest, self).tearDownClass() User.delete().execute()
def email_taken(self, email): return User.select().where(fn.lower(User.email) == email.lower()).exists()
def tearDown(self): Session.delete().execute() User.delete().execute()