Add compilation exception if trying to finish a non-existing loop

Zomis · Dec 10, 2016 · e29867a · e29867a
1 parent f74ebb4
commit e29867a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/main/groovy/net/zomis/brainf/model/ast/tree/Parser.java b/src/main/groovy/net/zomis/brainf/model/ast/tree/Parser.java
@@ -55,6 +55,9 @@ public SyntaxTree parse(List<Token> tokens) {
                     }
                     if (bft.command == BrainFCommand.END_WHILE) {
                         SyntaxTree loopSyntax = depth.pop();
+                        if (depth.isEmpty()) {
+                            throw new BrainfuckCompilationException("No more loops to finish.");
+                        }
                         SyntaxTree current = depth.peek();
                         current.getTokens().addAll(inner.getTokens());
                         current.syntax.add(loopSyntax);

diff --git a/src/test/groovy/net/zomis/brainf/BrainTest.groovy b/src/test/groovy/net/zomis/brainf/BrainTest.groovy
@@ -149,11 +149,9 @@ public class BrainTest extends BrainfuckTest {
         assert output.toString() == "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
     }
 
-    @Test
+    @Test(expected = BrainfuckCompilationException)
     public void unbalanced1() {
         useCode("++[->+<] ]")
-        analyze(new WhileLoopAnalysis())
-        assert !analyze.get(WhileLoopAnalysis).bracketsMatching
     }
 
     @Test