use llmchain::escape_sql_string;
#[test]
fn test_escape_sql_string() {
let input = "Hello, World!";
let output = escape_sql_string(input);
assert_eq!(output, "Hello, World!");
let input = "Hello, 'World'!";
let output = escape_sql_string(input);
assert_eq!(output, "Hello, ''World''!");
let input = "Hello, 'World'! \n";
let output = escape_sql_string(input);
assert_eq!(output, "Hello, ''World''! ");
let input = "Hello, 'World'! \r";
let output = escape_sql_string(input);
assert_eq!(output, "Hello, ''World''! \\r");
}