rust-postgres/postgres-protocol/src/escape/test.rs at master · ITCSsDeveloper/rust-postgres · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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'"));
}