Lines
82.38 %
Functions
100 %
Branches
/*
* Hurl (https://hurl.dev)
* Copyright (C) 2024 Orange
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
use crate::ast::{
Predicate, PredicateFunc, PredicateFuncValue, PredicateValue, SourceInfo, Whitespace,
};
use crate::combinator::choice;
use crate::parser::predicate_value::predicate_value;
use crate::parser::primitives::{one_or_more_spaces, try_literal, zero_or_more_spaces};
use crate::parser::{ParseError, ParseErrorKind, ParseResult};
use crate::reader::Reader;
pub fn predicate(reader: &mut Reader) -> ParseResult<Predicate> {
let (not, space0) = predicate_not(reader);
let func = predicate_func(reader)?;
Ok(Predicate {
not,
space0,
predicate_func: func,
})
}
// can not fail
fn predicate_not(reader: &mut Reader) -> (bool, Whitespace) {
let save = reader.cursor();
let no_whitespace = Whitespace {
value: String::new(),
source_info: SourceInfo {
start: save.pos,
end: save.pos,
},
if try_literal("not", reader).is_ok() {
match one_or_more_spaces(reader) {
Ok(space) => (true, space),
Err(_) => {
reader.seek(save);
(false, no_whitespace)
} else {
fn predicate_func(reader: &mut Reader) -> ParseResult<PredicateFunc> {
let start = reader.cursor();
let value = predicate_func_value(reader)?;
let end = reader.cursor();
Ok(PredicateFunc {
source_info: SourceInfo::new(start.pos, end.pos),
value,
fn predicate_func_value(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
match choice(
&[
equal_predicate,
not_equal_predicate,
greater_or_equal_predicate,
greater_predicate,
less_or_equal_predicate,
less_predicate,
start_with_predicate,
end_with_predicate,
contain_predicate,
include_predicate,
match_predicate,
integer_predicate,
float_predicate,
boolean_predicate,
string_predicate,
collection_predicate,
date_predicate,
iso_date_predicate,
exist_predicate,
is_empty_predicate,
is_number_predicate,
],
reader,
) {
Err(ParseError {
recoverable: true, ..
}) => Err(ParseError::new(start.pos, false, ParseErrorKind::Predicate)),
x => x,
impl PredicateValue {
pub fn is_number(&self) -> bool {
matches!(self, PredicateValue::Number(_))
pub fn is_string(&self) -> bool {
matches!(self, PredicateValue::String(_))
pub fn is_bytearray(&self) -> bool {
matches!(self, PredicateValue::Hex(_)) | matches!(self, PredicateValue::Base64(_))
pub fn is_expression(&self) -> bool {
matches!(self, PredicateValue::Placeholder(_))
fn equal_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("==", reader)?;
let space0 = zero_or_more_spaces(reader)?;
let value = predicate_value(reader)?;
Ok(PredicateFuncValue::Equal { space0, value })
fn not_equal_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("!=", reader)?;
Ok(PredicateFuncValue::NotEqual { space0, value })
fn greater_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal(">", reader)?;
if value.is_number() || value.is_string() || value.is_expression() {
Ok(PredicateFuncValue::GreaterThan { space0, value })
Err(ParseError::new(
start.pos,
false,
ParseErrorKind::PredicateValue,
))
fn greater_or_equal_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal(">=", reader)?;
Ok(PredicateFuncValue::GreaterThanOrEqual { space0, value })
fn less_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("<", reader)?;
Ok(PredicateFuncValue::LessThan { space0, value })
fn less_or_equal_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("<=", reader)?;
Ok(PredicateFuncValue::LessThanOrEqual { space0, value })
fn start_with_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("startsWith", reader)?;
let space0 = one_or_more_spaces(reader)?;
if !value.is_string() && !value.is_bytearray() {
return Err(ParseError::new(
save.pos,
));
Ok(PredicateFuncValue::StartWith { space0, value })
fn end_with_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("endsWith", reader)?;
Ok(PredicateFuncValue::EndWith { space0, value })
fn contain_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("contains", reader)?;
Ok(PredicateFuncValue::Contain { space0, value })
fn include_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("includes", reader)?;
Ok(PredicateFuncValue::Include { space0, value })
fn match_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("matches", reader)?;
if !matches!(value, PredicateValue::String(_)) && !matches!(value, PredicateValue::Regex(_)) {
Ok(PredicateFuncValue::Match { space0, value })
fn integer_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("isInteger", reader)?;
Ok(PredicateFuncValue::IsInteger)
fn float_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("isFloat", reader)?;
Ok(PredicateFuncValue::IsFloat)
fn boolean_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("isBoolean", reader)?;
Ok(PredicateFuncValue::IsBoolean)
fn string_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("isString", reader)?;
Ok(PredicateFuncValue::IsString)
fn collection_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("isCollection", reader)?;
Ok(PredicateFuncValue::IsCollection)
fn date_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("isDate", reader)?;
Ok(PredicateFuncValue::IsDate)
fn iso_date_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("isIsoDate", reader)?;
Ok(PredicateFuncValue::IsIsoDate)
fn exist_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("exists", reader)?;
Ok(PredicateFuncValue::Exist)
fn is_empty_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("isEmpty", reader)?;
Ok(PredicateFuncValue::IsEmpty)
fn is_number_predicate(reader: &mut Reader) -> ParseResult<PredicateFuncValue> {
try_literal("isNumber", reader)?;
Ok(PredicateFuncValue::IsNumber)
#[cfg(test)]
mod tests {
use super::*;
Expr, ExprKind, Float, Number, Placeholder, Template, TemplateElement, Variable, I64,
use crate::reader::Pos;
#[test]
fn test_predicate_not() {
let mut reader = Reader::new("notXX");
assert_eq!(
predicate_not(&mut reader),
(
Whitespace {
source_info: SourceInfo::new(Pos::new(1, 1), Pos::new(1, 1)),
)
);
assert_eq!(reader.cursor().pos, Pos { line: 1, column: 1 });
let mut reader = Reader::new("not XX");
true,
value: String::from(" "),
source_info: SourceInfo::new(Pos::new(1, 4), Pos::new(1, 5)),
assert_eq!(reader.cursor().pos, Pos { line: 1, column: 5 });
fn test_predicate() {
let mut reader = Reader::new("not == true");
predicate(&mut reader).unwrap(),
Predicate {
not: true,
space0: Whitespace {
predicate_func: PredicateFunc {
source_info: SourceInfo::new(Pos::new(1, 5), Pos::new(1, 12)),
value: PredicateFuncValue::Equal {
source_info: SourceInfo::new(Pos::new(1, 7), Pos::new(1, 8)),
value: PredicateValue::Bool(true),
fn test_predicate_func() {
let mut reader = Reader::new("tata == 1");
let error = predicate_func(&mut reader).err().unwrap();
assert_eq!(error.pos, Pos { line: 1, column: 1 });
assert!(!error.recoverable);
assert_eq!(error.kind, ParseErrorKind::Predicate);
fn test_equal_predicate() {
let mut reader = Reader::new("== true");
equal_predicate(&mut reader).unwrap(),
PredicateFuncValue::Equal {
source_info: SourceInfo::new(Pos::new(1, 3), Pos::new(1, 5)),
let mut reader = Reader::new("== 1.1");
value: PredicateValue::Number(Number::Float(Float {
value: 1.1,
encoded: "1.1".to_string(),
})),
source_info: SourceInfo::new(Pos::new(1, 3), Pos::new(1, 4)),
let mut reader = Reader::new("== 2");
value: PredicateValue::Number(Number::Integer(I64::new(2, "2".to_string()))),
let mut reader = Reader::new("== \"Bob\"");
value: PredicateValue::String(Template {
delimiter: Some('"'),
elements: vec![TemplateElement::String {
value: "Bob".to_string(),
encoded: "Bob".to_string(),
}],
source_info: SourceInfo::new(Pos::new(1, 4), Pos::new(1, 9)),
}),
fn test_equal_expression_predicate() {
let mut reader = Reader::new("== {{count}}");
value: PredicateValue::Placeholder(Placeholder {
source_info: SourceInfo::new(Pos::new(1, 6), Pos::new(1, 6)),
expr: Expr {
kind: ExprKind::Variable(Variable {
name: "count".to_string(),
source_info: SourceInfo::new(Pos::new(1, 6), Pos::new(1, 11)),
space1: Whitespace {
source_info: SourceInfo::new(Pos::new(1, 11), Pos::new(1, 11)),
fn test_start_with_predicate() {
let mut reader = Reader::new("startsWith 2");
let error = start_with_predicate(&mut reader).err().unwrap();
error.pos,
Pos {
line: 1,
column: 12,
assert_eq!(error.kind, ParseErrorKind::PredicateValue);
fn test_date_predicate() {
let mut reader = Reader::new("isDate");
let result = date_predicate(&mut reader);
assert_eq!(result.unwrap(), PredicateFuncValue::IsDate);