test(langserver): fix some regression tests

robotcodedev · Dec 1, 2023 · fec9542 · fec9542
1 parent 0016ef6
commit fec9542
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 32 deletions.
diff --git a/...egtest_outputs/rf41/test_foldingrange.test[False-foldingrange.robot-077-001-else_end].out b/...egtest_outputs/rf41/test_foldingrange.test[False-foldingrange.robot-077-001-else_end].out
@@ -5,22 +5,8 @@ data: !GeneratedTestData
 result:
 - !FoldingRange
   collapsed_text: null
-  end_character: 23
-  end_line: 76
-  kind: if
-  start_character: 0
-  start_line: 63
-- !FoldingRange
-  collapsed_text: null
-  end_character: 23
-  end_line: 76
-  kind: if
-  start_character: 0
-  start_line: 71
-- !FoldingRange
-  collapsed_text: null
-  end_character: 23
-  end_line: 76
+  end_character: 8
+  end_line: 77
   kind: if
   start_character: 0
   start_line: 74
diff --git a/...regtest_outputs/rf41/test_foldingrange.test[True-foldingrange.robot-077-001-else_end].out b/...regtest_outputs/rf41/test_foldingrange.test[True-foldingrange.robot-077-001-else_end].out
@@ -6,21 +6,7 @@ result:
 - !FoldingRange
   collapsed_text: null
   end_character: null
-  end_line: 76
-  kind: if
-  start_character: null
-  start_line: 63
-- !FoldingRange
-  collapsed_text: null
-  end_character: null
-  end_line: 76
-  kind: if
-  start_character: null
-  start_line: 71
-- !FoldingRange
-  collapsed_text: null
-  end_character: null
-  end_line: 76
+  end_line: 77
   kind: if
   start_character: null
   start_line: 74
diff --git a/tests/robotcode/language_server/robotframework/parts/test_foldingrange.py b/tests/robotcode/language_server/robotframework/parts/test_foldingrange.py
@@ -80,7 +80,9 @@ async def test(
             if "start" in data.name.lower()
             and r.start_line == data.line
             or "end" in data.name.lower()
-            and r.end_line == data.line - (0 if ("if" in data.name.lower() or "for" in data.name.lower()) else 1)
+            and r.end_line
+            == data.line
+            - (0 if ("else" in data.name.lower() or "if" in data.name.lower() or "for" in data.name.lower()) else 1)
         ]
     assert result
     regtest.write(yaml.dump({"data": data, "result": result}))