diff --git a/crates/pgls_pretty_print/src/nodes/alter_policy_stmt.rs b/crates/pgls_pretty_print/src/nodes/alter_policy_stmt.rs index d60919b88..d22e69cc7 100644 --- a/crates/pgls_pretty_print/src/nodes/alter_policy_stmt.rs +++ b/crates/pgls_pretty_print/src/nodes/alter_policy_stmt.rs @@ -13,9 +13,7 @@ pub(super) fn emit_alter_policy_stmt(e: &mut EventEmitter, n: &AlterPolicyStmt) e.space(); // Policy name - if !n.policy_name.is_empty() { - e.token(TokenKind::IDENT(n.policy_name.clone())); - } + super::emit_identifier_maybe_quoted(e, &n.policy_name); // Table name if let Some(ref table) = n.table { diff --git a/crates/pgls_pretty_print/src/nodes/create_policy_stmt.rs b/crates/pgls_pretty_print/src/nodes/create_policy_stmt.rs index a8c874fd6..8d17e64fb 100644 --- a/crates/pgls_pretty_print/src/nodes/create_policy_stmt.rs +++ b/crates/pgls_pretty_print/src/nodes/create_policy_stmt.rs @@ -12,7 +12,7 @@ pub(super) fn emit_create_policy_stmt(e: &mut EventEmitter, n: &CreatePolicyStmt e.space(); e.token(TokenKind::POLICY_KW); e.space(); - e.token(TokenKind::IDENT(n.policy_name.clone())); + super::emit_identifier_maybe_quoted(e, &n.policy_name); e.line(LineType::SoftOrSpace); e.token(TokenKind::ON_KW); diff --git a/crates/pgls_pretty_print/tests/data/single/alter_policy_quoted_name_0.sql b/crates/pgls_pretty_print/tests/data/single/alter_policy_quoted_name_0.sql new file mode 100644 index 000000000..af4f5ea24 --- /dev/null +++ b/crates/pgls_pretty_print/tests/data/single/alter_policy_quoted_name_0.sql @@ -0,0 +1 @@ +ALTER POLICY "Public can read stats" ON my_table USING (true); diff --git a/crates/pgls_pretty_print/tests/data/single/create_policy_quoted_name_0.sql b/crates/pgls_pretty_print/tests/data/single/create_policy_quoted_name_0.sql new file mode 100644 index 000000000..ecdd883e8 --- /dev/null +++ b/crates/pgls_pretty_print/tests/data/single/create_policy_quoted_name_0.sql @@ -0,0 +1 @@ +CREATE POLICY "Public can read stats" ON my_table FOR SELECT USING (true); diff --git a/crates/pgls_pretty_print/tests/snapshots/single/tests__alter_policy_quoted_name_0_100.snap b/crates/pgls_pretty_print/tests/snapshots/single/tests__alter_policy_quoted_name_0_100.snap new file mode 100644 index 000000000..d601e1025 --- /dev/null +++ b/crates/pgls_pretty_print/tests/snapshots/single/tests__alter_policy_quoted_name_0_100.snap @@ -0,0 +1,5 @@ +--- +source: crates/pgls_pretty_print/tests/tests.rs +input_file: crates/pgls_pretty_print/tests/data/single/alter_policy_quoted_name_0.sql +--- +alter policy "Public can read stats" on my_table using (true); diff --git a/crates/pgls_pretty_print/tests/snapshots/single/tests__alter_policy_quoted_name_0_80.snap b/crates/pgls_pretty_print/tests/snapshots/single/tests__alter_policy_quoted_name_0_80.snap new file mode 100644 index 000000000..d601e1025 --- /dev/null +++ b/crates/pgls_pretty_print/tests/snapshots/single/tests__alter_policy_quoted_name_0_80.snap @@ -0,0 +1,5 @@ +--- +source: crates/pgls_pretty_print/tests/tests.rs +input_file: crates/pgls_pretty_print/tests/data/single/alter_policy_quoted_name_0.sql +--- +alter policy "Public can read stats" on my_table using (true); diff --git a/crates/pgls_pretty_print/tests/snapshots/single/tests__create_policy_quoted_name_0_100.snap b/crates/pgls_pretty_print/tests/snapshots/single/tests__create_policy_quoted_name_0_100.snap new file mode 100644 index 000000000..a79c5335a --- /dev/null +++ b/crates/pgls_pretty_print/tests/snapshots/single/tests__create_policy_quoted_name_0_100.snap @@ -0,0 +1,5 @@ +--- +source: crates/pgls_pretty_print/tests/tests.rs +input_file: crates/pgls_pretty_print/tests/data/single/create_policy_quoted_name_0.sql +--- +create policy "Public can read stats" on my_table as permissive for select to public using (true); diff --git a/crates/pgls_pretty_print/tests/snapshots/single/tests__create_policy_quoted_name_0_80.snap b/crates/pgls_pretty_print/tests/snapshots/single/tests__create_policy_quoted_name_0_80.snap new file mode 100644 index 000000000..fac133f0e --- /dev/null +++ b/crates/pgls_pretty_print/tests/snapshots/single/tests__create_policy_quoted_name_0_80.snap @@ -0,0 +1,10 @@ +--- +source: crates/pgls_pretty_print/tests/tests.rs +input_file: crates/pgls_pretty_print/tests/data/single/create_policy_quoted_name_0.sql +--- +create policy "Public can read stats" +on my_table +as permissive +for select +to public +using (true);