germanubis/rust-postgres: Native PostgreSQL driver for the Rust programming language

postgres-protocol/src/escape/test.rs

17 строк677 B

Jeff Davis

Add escape_identifier() and escape_literal().

29 ноя 2020, 09:35

29 ноя 2020, 09:35aadf942

use crate::escape::{escape_identifier, escape_literal};

#[test]
fn test_escape_idenifier() {
    assert_eq!(escape_identifier("foo"), String::from("\"foo\""));
    assert_eq!(escape_identifier("f\\oo"), String::from("\"f\\oo\""));
    assert_eq!(escape_identifier("f'oo"), String::from("\"f'oo\""));
    assert_eq!(escape_identifier("f\"oo"), String::from("\"f\"\"oo\""));
}

#[test]
fn test_escape_literal() {
    assert_eq!(escape_literal("foo"), String::from("'foo'"));
    assert_eq!(escape_literal("f\\oo"), String::from(" E'f\\\\oo'"));
    assert_eq!(escape_literal("f'oo"), String::from("'f''oo'"));
    assert_eq!(escape_literal("f\"oo"), String::from("'f\"oo'"));
}