class UserTest(BaseTest):
def setUp(self):
super().setUp()
with self.app_context():
self.user = UserModel('test_admin', '0123456789', 'admin')
self.user.save_to_db()
def test_create_user(self):
with self.app_context():
self.user.save_to_db()
user = UserModel.find_by_id(self.user.id)
self.assertEqual(user.username, 'test_admin')
self.assertEqual(user.password, '0123456789')
def test_fail_create_user(self):
with self.assertRaises(BaseException):
UserModel()
def test_update_user(self):
with self.app_context():
self.user.username = '******'
self.user.save_to_db()
user = UserModel.find_by_id(self.user.id)
self.assertEqual(user.username, 'administrator')
def test_fail_update_user(self):
with self.app_context():
with self.assertRaises(BaseException):
self.user = None
self.user.save_to_db()
def test_delete_user(self):
with self.app_context():
self.user.delete_from_db()
user = UserModel.find_by_id(self.user.id)
self.assertIsNone(user)
def test_crud(self):
with self.app_context():
# Create
user = UserModel('test_user', 'password', 'admin')
user.save_to_db()
# Read
user = UserModel.find_by_username('test_user')
self.assertIsNotNone(user)
user = UserModel.find_by_id(1)
self.assertIsNotNone(user)
user = next(iter(UserModel.find_by_role('admin')))
self.assertIsNotNone(user)
# Update
user.username = '******'
user.save_to_db()
user = UserModel.find_by_username('administrator')
self.assertIsNotNone(user)
# Delete
user.delete_from_db()
self.assertIsNone(UserModel.find_by_username('administrator'))
