fix position context on single character number literal

zigtools · Nov 20, 2024 · 28489a2 · 28489a2
1 parent 36982fd
commit 28489a2
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/src/analysis.zig b/src/analysis.zig
@@ -3488,7 +3488,10 @@ pub fn getPositionContext(
             // `tok` is the latter of the two.
             if (!should_do_lookahead) break;
             switch (tok.tag) {
-                .identifier, .builtin => should_do_lookahead = false,
+                .identifier,
+                .builtin,
+                .number_literal,
+                => should_do_lookahead = false,
                 else => break,
             }
         }

diff --git a/tests/utility/position_context.zig b/tests/utility/position_context.zig
@@ -354,6 +354,24 @@ test "global error set" {
     // , .global_error_set, .{});
 }
 
+test "number literal" {
+    try testContext(
+        \\var foo = <loc>5<cursor></loc>;
+    , .number_literal, .{});
+    try testContext(
+        \\var foo = <loc><cursor>5</loc>;
+    , .number_literal, .{ .lookahead = true });
+}
+
+test "char literal" {
+    try testContext(
+        \\var foo = <loc>'5<cursor>'</loc>;
+    , .char_literal, .{ .lookahead = true });
+    try testContext(
+        \\var foo = <loc>'<cursor>5'</loc>;
+    , .char_literal, .{ .lookahead = true });
+}
+
 test "enum literal" {
     try testContext(
         \\var foo = <loc>.<cursor>tag</loc>;