More test cases were added

Author: ThakeeNathees

jac/jaclang/compiler/parser.py

@@ -189,10 +189,15 @@ def feed_current_token(iparser: jl.InteractiveParser, tok: jl.Token) -> bool:
                 return feed_current_token(iparser, e.token)
 
             if Tok.KW_WITH.name in iparser.accepts():
-                self.log_error(
-                    "Arbitary statements must be declared inside an entry block",
-                    self.error_to_token(e),
-                )
+                # Copy and feed the `with` token cause we don't want to actually
+                # feed the `with` token in the main parser state.
+                ipcopy = iparser.copy(deepcopy_values=True)
+                ipcopy.feed_token(jl.Token(Tok.KW_WITH.name, "with"))
+                if Tok.KW_ENTRY.name in ipcopy.accepts():
+                    self.log_error(
+                        "Arbitary statements must be declared inside an entry block",
+                        self.error_to_token(e),
+                    )
                 return False
 
             # We're calling try_feed_missing_token twice here because the first missing

jac/jaclang/langserve/tests/server_test/test_lang_serve.py

@@ -62,7 +62,8 @@ def test_open_with_syntax_error():
     try:
         helper.open_document()
         helper.assert_has_diagnostics(
-            count=2, message_contains="Unexpected token 'error'"
+            count=2,
+            message_contains="Arbitary statements must be declared inside an entry block",
         )
 
         diagnostics = helper.get_diagnostics()

jac/jaclang/tests/test_language.py

@@ -194,15 +194,42 @@ def test_assignment_list_no_infinite_loop():
 
 def test_free_code_outside_entry_reports_error(capsys):
     """Free python code outside an entry should report an error."""
+
+    # Case 1: Free python code not wrapped in `with entry { ... }` is invalid
     prog = JacProgram()
-    # Free python code not wrapped in `with entry { ... }` is invalid
     code = 'print("hello world");'
     # Compile and capture stderr via pytest capsys
     prog.compile(use_str=code, file_path="test.jac")
     captured = capsys.readouterr()
+
     # The parser change logs this specific message when encountering free code
     assert "Arbitary statements must be declared inside an entry block" in captured.err
 
+    # Case 2: This is unexpected token as well but not the same error
+    prog = JacProgram()
+    code = """
+    with entry {
+        *;
+    }
+    """
+    prog.compile(use_str=code, file_path="test.jac")
+    captured = capsys.readouterr()
+    assert (
+        "Arbitary statements must be declared inside an entry block" not in captured.err
+    )
+
+    # Case 3: `with` context manager block
+    prog = JacProgram()
+    code = """
+    with entry {
+        with open('some/file.txt', 'r') as f {
+            print(f.read());
+        }
+    """
+    prog.compile(use_str=code, file_path="test.jac")
+    captured = capsys.readouterr()
+    assert len(prog.errors_had) == 0
+
 
 def test_need_import(fixture_path, capture_stdout):
     """Test importing python."""