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)