chore: add tests for multi-line fetches

Author: gnikit

test/test_parser.py

@@ -1,3 +1,4 @@
+import pytest
 from setup_tests import test_dir
 
 from fortls.parsers.internal.parser import FortranFile
@@ -57,3 +58,41 @@ def test_weird_parser_bug():
     ast = file.parse()
     assert err_str is None
     assert not ast.end_errors
+
+
+@pytest.mark.parametrize(
+    "ln_no, pp_defs, reference",
+    [
+        (6, {}, 6),
+        (7, {}, 6),
+        (8, {}, 6),
+        (11, {"TEST": True}, 60),  # not entirely correct ref vals
+        (23, {"MULT": True}, 90),  # not entirely correct ref vals
+        (32, {"TEST": True, "MULT": True}, 130),  # not entirely correct ref vals
+        (39, {"TEST": True, "MULT": True}, 2400),  # not entirely correct ref vals
+    ],
+)
+def test_get_code_line_multilines(ln_no: int, pp_defs: dict, reference: int):
+    """Tests how the get_code_line performs with multi-line and preprocessor
+
+    Not all the results are correct, since get_code_line is not aware of the
+    preprocessor skips. Instead what it does is it evaluates all the line
+    continuations and appends them in post.
+    """
+
+    def calc_result(res: tuple):
+        pre, cur, post = res
+        res = "".join(pre + [cur] + post).replace(" ", "")
+        assert "result" in res, "Fortran variable `result` not found in results"
+        loc = {}
+        exec(res, None, loc)
+        return loc["result"]
+
+    file_path = test_dir / "parse" / "mixed" / "multilines.F90"
+    file = FortranFile(str(file_path))
+    file.load_from_disk()
+    file.preprocess(pp_defs=pp_defs)
+    pp = bool(pp_defs)
+    res = file.get_code_line(line_no=ln_no, pp_content=pp)
+    result = calc_result(res)
+    assert result == reference

test/test_source/parse/mixed/multilines.F90

@@ -0,0 +1,49 @@
+program multiline_tests
+  implicit none
+  integer :: result
+  character(len=100) :: str
+
+  ! Test: Simple multi-line continuation
+  result = 1 + &
+    2 + &
+    3
+
+  ! Test: Multi-line continuation with a preprocessor directive
+  result = 10 + &
+#ifdef TEST
+    20 + &
+#endif
+    30
+
+  ! Test: Multi-line continuation with string concatenation
+  str = 'Hello' // &
+  & ' ' // &
+  &  'World'
+
+  ! Test: Multi-line continuation with mixed preprocessor and arithmetic operations
+  result = &
+#ifdef MULT
+    (10*2) + &
+#else
+    (10 * 3) + &
+#endif
+  & 10 * 4
+
+  ! Test: Multi-line continuation with C preprocessor && sequence
+  result = 100 + &
+#if defined(TEST) && defined(MULT)
+  &(20) + &
+#endif
+  &10
+
+  ! Test: multiplee Multi-line continuation with C preprocessor and comments
+  result = 1000 + & ! Comment 0
+#if defined( TEST ) && defined( MULT )
+  &100 + &  ! Comment 1
+  &200+&    !! Comment 2
+#else
+    500 + & !!! Comment 3
+#endif
+  &600
+
+end program multiline_tests