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 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 there_are_users(self):
return User.select().exists()
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 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 tearDownClass(self):
super(UserTest, self).tearDownClass()
User.delete().execute()
def tearDown(self):
Session.delete().execute()
User.delete().execute()
def tearDown(self):
Session.delete().execute()
User.delete().execute()
