use indoc, add setup

Author: juleswritescode

Cargo.lock

@@ -32,7 +32,7 @@ tokio = { version = "1.41.1", features = ["full"] }
 
 [dev-dependencies]
 criterion.workspace       = true
-indoc                     = "2.0.7"
+unindent                  = "0.2.4"
 insta.workspace           = true
 pgls_test_utils.workspace = true
 regex                     = "1.12.2"

crates/pgls_completions/Cargo.toml

@@ -50,8 +50,6 @@ fn get_completion_text(ctx: &TreesitterContext, col: &Column) -> CompletionText
 
 #[cfg(test)]
 mod tests {
-    use indoc::indoc;
-
     use sqlx::PgPool;
 
     use crate::test_helper::{TestCompletionsCase, TestCompletionsSuite};
@@ -79,13 +77,13 @@ mod tests {
 
         TestCompletionsSuite::new(&pool, Some(setup)).with_case(
         TestCompletionsCase::new()
-        .inside_static_statement(indoc! {r#"
+        .inside_static_statement(r#"
             select * from (
                 <sql>
             ) as subquery
             join public.users u
             on u.id = subquery.id;
-            "#})
+            "#)
             .type_sql("select id, narrator_id<1> from private.audio_books")
             .comment("Should prefer the one from private.audio_audiobooks, since the other tables are out of scope.")
         )

crates/pgls_completions/src/providers/columns.rs

@@ -2,6 +2,27 @@
 source: crates/pgls_completions/src/test_helper.rs
 expression: final_snapshot
 ---
+***Setup***
+
+create schema auth;
+
+create table auth.users (
+    uid serial primary key,
+    name text not null,
+    email text unique not null
+);
+
+create table auth.posts (
+    pid serial primary key,
+    user_id int not null references auth.users(uid),
+    title text not null,
+    content text,
+    created_at timestamp default now()
+);
+
+
+--------------
+
 select u.id, auth.posts.content from auth.users u join auth.posts p on |
 **Should prioritize primary keys here.**
 

crates/pgls_completions/src/snapshots/pgls_completions__test_helper__completes_in_join_on_clause.snap

@@ -2,6 +2,20 @@
 source: crates/pgls_completions/src/test_helper.rs
 expression: final_snapshot
 ---
+***Setup***
+
+create schema if not exists private;
+
+create table private.users (
+    id serial primary key,
+    email text unique not null,
+    name text not null,
+    "quoted_column" text
+);
+
+
+--------------
+
 s|
 select |
 

crates/pgls_completions/src/snapshots/pgls_completions__test_helper__completes_quoted_columns.snap

@@ -2,6 +2,25 @@
 source: crates/pgls_completions/src/test_helper.rs
 expression: final_snapshot
 ---
+***Setup***
+
+create schema if not exists private;
+
+create table private.users (
+    id serial primary key,
+    email text unique not null,
+    name text not null,
+    "quoted_column" text
+);
+
+create table public.names (
+    uid serial references private.users(id),
+    name text
+);
+
+
+--------------
+
 ***Case 1:***
 
 s|

crates/pgls_completions/src/snapshots/pgls_completions__test_helper__completes_quoted_columns_with_aliases.snap

@@ -2,6 +2,27 @@
 source: crates/pgls_completions/src/test_helper.rs
 expression: final_snapshot
 ---
+***Setup***
+
+create schema auth;
+
+create table auth.users (
+    uid serial primary key,
+    name text not null,
+    email text unique not null
+);
+
+create table auth.posts (
+    pid serial primary key,
+    user_id int not null references auth.users(uid),
+    title text not null,
+    content text,
+    created_at timestamp default now()
+);
+
+
+--------------
+
 s|
 select |
 

crates/pgls_completions/src/snapshots/pgls_completions__test_helper__does_not_complete_cols_in_join_clauses.snap

@@ -2,6 +2,27 @@
 source: crates/pgls_completions/src/test_helper.rs
 expression: final_snapshot
 ---
+***Setup***
+
+create schema auth;
+
+create table auth.users (
+    uid serial primary key,
+    name text not null,
+    email text unique not null
+);
+
+create table auth.posts (
+    pid serial primary key,
+    user_id int not null references auth.users(uid),
+    title text not null,
+    content text,
+    created_at timestamp default now()
+);
+
+
+--------------
+
 select u.id, p.content from auth.users u join auth.posts p |
 select u.id, p.content from auth.users u join auth.posts p o|
 select u.id, p.content from auth.users u join auth.posts p on |

crates/pgls_completions/src/snapshots/pgls_completions__test_helper__filters_out_by_aliases_in_join_on.snap

@@ -2,6 +2,27 @@
 source: crates/pgls_completions/src/test_helper.rs
 expression: final_snapshot
 ---
+***Setup***
+
+create schema auth;
+
+create table auth.users (
+    uid serial primary key,
+    name text not null,
+    email text unique not null
+);
+
+create table auth.posts (
+    pid serial primary key,
+    user_id int not null references auth.users(uid),
+    title text not null,
+    content text,
+    created_at timestamp default now()
+);
+
+
+--------------
+
 s| from auth.users u join auth.posts p on u.id = p.user_id;
 select | from auth.users u join auth.posts p on u.id = p.user_id;
 

crates/pgls_completions/src/snapshots/pgls_completions__test_helper__filters_out_by_aliases_in_select.snap

@@ -2,6 +2,28 @@
 source: crates/pgls_completions/src/test_helper.rs
 expression: final_snapshot
 ---
+***Setup***
+
+create schema private;
+
+create table public.users (
+    id serial primary key,
+    name text
+);
+
+create table public.audio_books (
+    id serial primary key,
+    narrator text
+);
+
+create table private.audio_books (
+    id serial primary key,
+    narrator_id text
+);
+
+
+--------------
+
 select * from (
     |
 ) as subquery