From bbff5bd20a62884c21f3e941464ff76b0b7a9a7e Mon Sep 17 00:00:00 2001
From: Sam Harwell <sam@tunnelvisionlabs.com>
Date: Mon, 12 Mar 2012 07:26:39 -0500
Subject: [PATCH] New tests related to if/if/else constructs

---
 .../org/antlr/v4/test/TestParserExec.java     | 51 +++++++++++++++++++
 1 file changed, 51 insertions(+)

diff --git a/tool/test/org/antlr/v4/test/TestParserExec.java b/tool/test/org/antlr/v4/test/TestParserExec.java
index 9f3c6622b..feb62df3e 100644
--- a/tool/test/org/antlr/v4/test/TestParserExec.java
+++ b/tool/test/org/antlr/v4/test/TestParserExec.java
@@ -162,4 +162,55 @@ public class TestParserExec extends BaseTest {
 		assertEquals("a34c\n", found);
 	}
 
+	/**
+	 * This test is meant to detect regressions of bug antlr/antlr4#41.
+	 * https://github.com/antlr/antlr4/issues/41
+	 */
+	@Test
+	public void testOptional() throws Exception {
+		String grammar =
+			"grammar T;\n" +
+			"s : a | 'x';\n" +
+			"a : 'a' s ('b' s)?;\n"
+			;
+
+		String found = execParser("T.g", grammar, "TParser", "TLexer", "s", "x", false);
+		assertEquals("", found);
+		assertNull(this.stderrDuringParse);
+
+		found = execParser("T.g", grammar, "TParser", "TLexer", "s", "axbx", false);
+		assertEquals("", found);
+		assertNull(this.stderrDuringParse);
+
+		found = execParser("T.g", grammar, "TParser", "TLexer", "s", "ax", false);
+		assertEquals("", found);
+		assertNull(this.stderrDuringParse);
+
+		found = execParser("T.g", grammar, "TParser", "TLexer", "s", "aaxbx", false);
+		assertEquals("", found);
+		assertNull(this.stderrDuringParse);
+	}
+
+	/**
+	 * This test is meant to test the expected solution to antlr/antlr4#42.
+	 * https://github.com/antlr/antlr4/issues/42
+	 */
+	@Test
+	public void testIfIfElse() throws Exception {
+		String grammar =
+			"grammar T;\n" +
+			"stmt : ifStmt | ID;\n" +
+			"ifStmt : 'if' ID stmt ('else' stmt | {_input.LA(1) != ELSE}?);\n" +
+			"ELSE : 'else';\n" +
+			"ID : [a-zA-Z]+;\n" +
+			"WS : (' ' | '\\t')+ -> skip;\n"
+			;
+
+		String found = execParser("T.g", grammar, "TParser", "TLexer", "stmt",
+								  "if x if x a else b", true);
+		String expecting = "";
+		assertEquals(expecting, found);
+		assertNull(this.stderrDuringParse);
+	}
+
 }