def test_check_user_password(self):
Session = sessionmaker(bind=Database.create_sqlalchemy_engine())
session = Session()
User.add_user_account(session=session, user_id="apple", password="******")
self.assertTrue(User.check_user_account_password(session=session, user_id="apple", password="******"))
self.assertFalse(User.check_user_account_password(session=session, user_id="apple", password="******"))
self.assertFalse(User.check_user_account_password(session=session, user_id="xxx", password="******"))
session.close()
def test_remove_user(self):
cleanup = Cleanup()
session = Database.create_sqlalchemy_session_push(cleanup)
self.assertFalse(User.check_user_account_exist(session=session, user_id="apple"))
User.add_user_account(session=session, user_id="apple", password="******")
self.assertTrue(User.check_user_account_exist(session=session, user_id="apple"))
User.remove_user_account(session=session, user_id="apple")
self.assertFalse(User.check_user_account_exist(session=session, user_id="apple"))
def test_get_user_permission(self):
cleanup = Cleanup()
session = Database.create_sqlalchemy_session()
cleanup.push(session.close)
User.add_user_account(session, "u0", "up0")
Group.add(session, "g0", "g0")
UserGroup.join(session, "u0", "g0")
self.assertEqual(Permission.get_user_permission(session, "u0", "p0"), False)
GroupPermission.set(session, "g0", "p0", 10, True)
self.assertEqual(Permission.get_user_permission(session, "u0", "p0"), True)
GroupPermission.set(session, "g0", "p0", 20, False)
self.assertEqual(Permission.get_user_permission(session, "u0", "p0"), False)
Group.add(session, "g1", "g1")
UserGroup.join(session, "u0", "g1")
GroupPermission.set(session, "g1", "p0", 30, True)
self.assertEqual(Permission.get_user_permission(session, "u0", "p0"), True)
self.assertEqual(Permission.get_user_permission(session, "u1", "p0"), False)
self.assertEqual(Permission.get_user_permission(session, "u0", "p1"), False)
