Line data Source code
1 : #[cfg(test)]
2 : mod pg_helpers_tests {
3 : use std::fs::File;
4 :
5 : use compute_api::spec::{ComputeSpec, GenericOption, GenericOptions, PgIdent};
6 : use compute_tools::pg_helpers::*;
7 :
8 : #[test]
9 1 : fn params_serialize() {
10 1 : let file = File::open("../libs/compute_api/tests/cluster_spec.json").unwrap();
11 1 : let spec: ComputeSpec = serde_json::from_reader(file).unwrap();
12 1 :
13 1 : assert_eq!(
14 1 : spec.cluster.databases.first().unwrap().to_pg_options(),
15 1 : "LC_COLLATE 'C' LC_CTYPE 'C' TEMPLATE template0 OWNER \"alexk\""
16 1 : );
17 1 : assert_eq!(
18 1 : spec.cluster.roles.first().unwrap().to_pg_options(),
19 1 : " LOGIN PASSWORD 'md56b1d16b78004bbd51fa06af9eda75972'"
20 1 : );
21 1 : }
22 :
23 : #[test]
24 1 : fn settings_serialize() {
25 1 : let file = File::open("../libs/compute_api/tests/cluster_spec.json").unwrap();
26 1 : let spec: ComputeSpec = serde_json::from_reader(file).unwrap();
27 1 :
28 1 : assert_eq!(
29 1 : spec.cluster.settings.as_pg_settings(),
30 1 : r#"fsync = off
31 1 : wal_level = logical
32 1 : hot_standby = on
33 1 : neon.safekeepers = '127.0.0.1:6502,127.0.0.1:6503,127.0.0.1:6501'
34 1 : wal_log_hints = on
35 1 : log_connections = on
36 1 : shared_buffers = 32768
37 1 : port = 55432
38 1 : max_connections = 100
39 1 : max_wal_senders = 10
40 1 : listen_addresses = '0.0.0.0'
41 1 : wal_sender_timeout = 0
42 1 : password_encryption = md5
43 1 : maintenance_work_mem = 65536
44 1 : max_parallel_workers = 8
45 1 : max_worker_processes = 8
46 1 : neon.tenant_id = 'b0554b632bd4d547a63b86c3630317e8'
47 1 : max_replication_slots = 10
48 1 : neon.timeline_id = '2414a61ffc94e428f14b5758fe308e13'
49 1 : shared_preload_libraries = 'neon'
50 1 : synchronous_standby_names = 'walproposer'
51 1 : neon.pageserver_connstring = 'host=127.0.0.1 port=6400'
52 1 : test.escaping = 'here''s a backslash \\ and a quote '' and a double-quote " hooray'
53 1 : "#
54 1 : );
55 1 : }
56 :
57 : #[test]
58 1 : fn ident_pg_quote() {
59 1 : let ident: PgIdent = PgIdent::from("\"name\";\\n select 1;");
60 1 :
61 1 : assert_eq!(ident.pg_quote(), "\"\"\"name\"\";\\n select 1;\"");
62 1 : }
63 :
64 : #[test]
65 1 : fn ident_pg_quote_dollar() {
66 1 : let test_cases = vec![
67 1 : ("name", ("$$name$$", "x")),
68 1 : ("name$$", ("$x$name$$$x$", "xx")),
69 1 : ("name$$$", ("$x$name$$$$x$", "xx")),
70 1 : ("name$$$$", ("$x$name$$$$$x$", "xx")),
71 1 : ("name$x$", ("$xx$name$x$$xx$", "xxx")),
72 1 : ];
73 :
74 6 : for (input, expected) in test_cases {
75 5 : let (escaped, tag) = PgIdent::from(input).pg_quote_dollar();
76 5 : assert_eq!(escaped, expected.0);
77 5 : assert_eq!(tag, expected.1);
78 : }
79 1 : }
80 :
81 : #[test]
82 1 : fn generic_options_search() {
83 1 : let generic_options: GenericOptions = Some(vec![
84 1 : GenericOption {
85 1 : name: "present_value".into(),
86 1 : value: Some("value".into()),
87 1 : vartype: "string".into(),
88 1 : },
89 1 : GenericOption {
90 1 : name: "missed_value".into(),
91 1 : value: None,
92 1 : vartype: "int".into(),
93 1 : },
94 1 : ]);
95 1 : assert_eq!(generic_options.find("present_value"), Some("value".into()));
96 1 : assert_eq!(generic_options.find("missed_value"), None);
97 1 : assert_eq!(generic_options.find("invalid_value"), None);
98 :
99 1 : let empty_generic_options: GenericOptions = Some(vec![]);
100 1 : assert_eq!(empty_generic_options.find("present_value"), None);
101 1 : assert_eq!(empty_generic_options.find("missed_value"), None);
102 1 : assert_eq!(empty_generic_options.find("invalid_value"), None);
103 :
104 1 : let none_generic_options: GenericOptions = None;
105 1 : assert_eq!(none_generic_options.find("present_value"), None);
106 1 : assert_eq!(none_generic_options.find("missed_value"), None);
107 1 : assert_eq!(none_generic_options.find("invalid_value"), None);
108 1 : }
109 :
110 : #[test]
111 1 : fn test_escape_literal() {
112 1 : assert_eq!(escape_literal("test"), "'test'");
113 1 : assert_eq!(escape_literal("test'"), "'test'''");
114 1 : assert_eq!(escape_literal("test\\'"), "E'test\\\\'''");
115 1 : assert_eq!(escape_literal("test\\'\\'"), "E'test\\\\''\\\\'''");
116 1 : }
117 : }
|