Function verify_pretty_statement

src/sql-pretty/tests/parser.rs:79–117

(stmt: &str)

}

fn verify_pretty_statement(stmt: &str) {
    let original = match parse_statements(stmt) {
        Ok(stmt) => match stmt.into_iter().next() {
            Some(stmt) => stmt,
            None => return,
        },
        Err(_) => return,
    };
    for width in &[1, 40, 1000000] {
        let width = *width;
        let pretty1 = to_pretty(
            &original.ast,
            PrettyConfig {
                width,
                format_mode: FormatMode::Simple,
            },
        );
        let prettied = parse_statements(&pretty1)
            .unwrap_or_else(|_| panic!("could not parse: {pretty1}, original: {stmt}"))
            .into_iter()
            .next()
            .unwrap();
        let pretty2 = to_pretty(
            &prettied.ast,
            PrettyConfig {
                width,
                format_mode: FormatMode::Simple,
            },
        );
        assert_eq!(pretty1, pretty2);
        assert_eq!(
            original.ast.to_ast_string_stable(),
            prettied.ast.to_ast_string_stable(),
            "\noriginal: {stmt}",
        );
        // It'd be nice to assert that this squashes to a single line at high Ns, but literals and
        // idents can contain newlines so that's not always possible.
    }
}