jsonpath/tests/selector.rs

extern crate jsonpath_lib as jsonpath;
extern crate serde;
#[macro_use]
extern crate serde_json;

use serde::{Deserialize, Serialize};
use serde_json::Value;

use jsonpath::Selector;

#[derive(Serialize, Deserialize, PartialEq, Debug)]
struct Person {
    name: String,
    age: u8,
    phone: String,
}

fn input_str() -> &'static str {
    r#"[
        {
            "name": "이름1",
            "age": 40,
            "phone": "+33 12341234"
        },
        {
            "name": "이름2",
            "age": 42,
            "phone": "++44 12341234"
        },
        {
            "name": "이름3",
            "age": 50,
            "phone": "++55 111111"
        },
        {
            "name": "이름4",
            "age": 51,
            "phone": "++55 12341234"
        }
    ]"#
}

fn input_json() -> Value {
    serde_json::from_str(input_str()).unwrap()
}

fn input_person() -> Vec<Person> {
    serde_json::from_str(input_str()).unwrap()
}

#[test]
fn selector_value_from() {
    let result = Selector::new()
        .path("$..[?(@.age > 40)]").unwrap()
        .value_from(&input_person()).unwrap()
        .select_as::<Vec<Person>>().unwrap();
    assert_eq!(input_person()[1], result[0]);
}

#[test]
fn selector_value() {
    let result = Selector::new()
        .path("$..[?(@.age > 40)]").unwrap()
        .value((&input_json()).into()).unwrap()
        .select_as_value().unwrap();
    assert_eq!(input_json()[1], result[0]);
}

#[test]
fn selector_value_from_str() {
    let result = Selector::new()
        .path("$..[?(@.age > 40)]").unwrap()
        .value_from_str(input_str()).unwrap()
        .select_as_value().unwrap();
    assert_eq!(input_json()[1], result[0]);
}

#[test]
fn selector_select_to() {
    let mut selector = Selector::new();

    let result = selector
        .path("$..[?(@.age > 40)]").unwrap()
        .value_from_str(input_str()).unwrap()
        .select_as_value().unwrap();
    assert_eq!(input_json()[1], result[0]);

    let result = selector.select_as_str().unwrap();
    let value: Value = serde_json::from_str(&result).unwrap();
    assert_eq!(input_json()[1], value[0]);

    let result = selector.select_as::<Vec<Person>>().unwrap();
    assert_eq!(input_person()[1], result[0]);

    let _ = selector.path("$..[?(@.age == 40)]");

    let result = selector.select_as_value().unwrap();
    assert_eq!(input_json()[0], result[0]);

    let result = selector.select_as_str().unwrap();
    assert_eq!(serde_json::to_string(&vec![&input_json()[0].clone()]).unwrap(), result);

    let result = selector.select_as::<Vec<Person>>().unwrap();
    assert_eq!(input_person()[0], result[0]);
}

fn _remove_name(v: Value) -> Option<Value> {
    let r = match v {
        Value::Array(mut vec) => {
            for v in &mut vec {
                v.as_object_mut().unwrap().remove("name");
            }
            Value::Array(vec)
        }
        _ => Value::Null
    };
    Some(r)
}

fn _change_phone_number(v: Value) -> Option<Value> {
    let r = match v {
        Value::Array(mut vec) => {
            let mut v = vec.pop().unwrap();
            v.as_object_mut().unwrap()
                .insert("phone".to_string(), Value::String("1234".to_string()));
            v
        }
        _ => Value::Null
    };
    Some(r)
}

fn _rejuvenate(mut vec: Vec<Person>) -> Option<Vec<Person>> {
    for p in &mut vec {
        p.age = p.age - 10;
    }
    Some(vec)
}

#[test]
fn selector_map_basic() {
    let mut selector = Selector::new();

    let result = selector
        .path("$..[?(@.age > 40)]").unwrap()
        .value_from_str(input_str()).unwrap()
        .map(_remove_name).unwrap()
        .get().unwrap();

    assert_eq!(result, json!([
        {"phone": "++44 12341234", "age": 42},
        {"phone": "++55 111111", "age": 50},
        {"phone": "++55 12341234", "age": 51},
    ]));
}

#[test]
fn selector_map() {
    let mut selector = Selector::new();

    let result = selector
        .path("$..[?(@.age > 40)]").unwrap()
        .value_from_str(input_str()).unwrap()
        .map(_remove_name).unwrap()
        .path("$..[?(@.age == 50)]").unwrap()
        .map(_change_phone_number).unwrap()
        .get().unwrap();

    assert_eq!(result, json!({
        "phone": "1234",
        "age": 50,
    }));
}

#[test]
fn selector_map_as_basic() {
    let mut selector = Selector::new();

    let result = selector
        .path("$..[?(@.age > 40)]").unwrap()
        .value_from_str(input_str()).unwrap()
        .map_as(_rejuvenate).unwrap()
        .get().unwrap();

    assert_eq!(result, json!([
        {"name": "이름2", "phone": "++44 12341234", "age": 32},
        {"name": "이름3", "phone": "++55 111111", "age": 40},
        {"name": "이름4", "phone": "++55 12341234", "age": 41},
    ]));
}

#[test]
fn selector_map_as() {
    let mut selector = Selector::new();

    let result = selector
        .path("$..[?(@.age > 40)]").unwrap()
        .value_from_str(input_str()).unwrap()
        .map_as(_rejuvenate).unwrap()
        .path("$..[?(@.age == 40)]").unwrap()
        .map(_change_phone_number).unwrap()
        .get().unwrap();

    assert_eq!(result, json!({
        "name": "이름3",
        "phone": "1234",
        "age": 40,
    }));
}