QueryRules mock in tests