progress

Author: psteinroe

crates/pgt_workspace/src/features/completions.rs

@@ -76,7 +76,6 @@ pub(crate) fn get_statement_for_completions(
 
 #[cfg(test)]
 mod tests {
-    use pgt_fs::PgTPath;
     use pgt_text_size::TextSize;
 
     use crate::workspace::Document;
@@ -93,11 +92,7 @@ mod tests {
         let pos: u32 = pos.try_into().unwrap();
 
         (
-            Document::new(
-                PgTPath::new("test.sql"),
-                sql.replace(CURSOR_POSITION, ""),
-                5,
-            ),
+            Document::new(sql.replace(CURSOR_POSITION, ""), 5),
             TextSize::new(pos),
         )
     }

crates/pgt_workspace/src/workspace/server/annotation.rs

@@ -76,8 +76,8 @@ mod tests {
             ("SELECT * FROM foo\n", false),
         ];
 
-        for (idx, (content, expected)) in test_cases.iter().enumerate() {
-            let statement_id = StatementId::Root(idx.into());
+        for (content, expected) in test_cases.iter() {
+            let statement_id = StatementId::new(content);
 
             let annotations = store.get_annotations(&statement_id, content);