def test_change_names(self):
sqlauth = SQLAuth("sqlite://", logging=False)
sqlauth.create_table()
sqlauth.add_user("test_user", "123", fullname="Test User")
guid = sqlauth.get_guid("test_user")
sqlauth.modify_user("test_user", name_new="test_2", fullname="Test 2")
with sqlauth.session_scope() as session:
user = session.query(User).filter_by(id=guid).scalar()
self.assertEqual(user.name, "test_2")
self.assertEqual(user.fullname, "Test 2")
def test_delete_user(self):
sqlauth = SQLAuth("sqlite://", logging=False)
sqlauth.create_table()
sqlauth.add_user("test_user", "123", fullname="Test User")
with sqlauth.session_scope() as session:
user = session.query(User).filter_by(name="test_user").scalar()
self.assertIsNotNone(user)
sqlauth.delete_user("test_user")
user = session.query(User).filter_by(name="test_user").scalar()
self.assertIsNone(user)
with self.assertRaisesRegex(ValueError, r".* not found"):
# deleting again should fail
sqlauth.delete_user("test_user")
def test_add_user(self):
sqlauth = SQLAuth("sqlite://", logging=False)
sqlauth.create_table()
with self.assertRaises(IntegrityError):
sqlauth.add_user(None, "123") # NULL username
with self.assertRaises(ValueError):
sqlauth.add_user("", "123") # empty username
with self.assertRaises(ValueError):
sqlauth.add_user("test_user", "") # empty pw
sqlauth.add_user("test_user", "123", fullname="Test User")
with self.assertRaises(IntegrityError):
# adding again should fail
sqlauth.add_user("test_user", "123", fullname="Test User")
with sqlauth.session_scope() as session:
user = session.query(User).filter_by(name="test_user").scalar()
self.assertEqual(user.name, "test_user")
self.assertEqual(user.fullname, "Test User")