更新query函数，添加query测试