1324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver/* 2324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * [The "BSD license"] 3324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * Copyright (c) 2010 Terence Parr 4324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * All rights reserved. 5324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 6324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * Redistribution and use in source and binary forms, with or without 7324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * modification, are permitted provided that the following conditions 8324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * are met: 9324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 1. Redistributions of source code must retain the above copyright 10324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * notice, this list of conditions and the following disclaimer. 11324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 2. Redistributions in binary form must reproduce the above copyright 12324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * notice, this list of conditions and the following disclaimer in the 13324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * documentation and/or other materials provided with the distribution. 14324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 3. The name of the author may not be used to endorse or promote products 15324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * derived from this software without specific prior written permission. 16324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 17324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 18324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 19324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 20324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 21324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 22324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver */ 28324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverpackage org.antlr.test; 29324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 30324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport org.antlr.runtime.tree.*; 31324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport org.junit.Test; 32324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 33324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport static org.junit.Assert.assertEquals; 34324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 35324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverpublic class TestTreeIterator { 36324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver static final String[] tokens = new String[] { 37324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver "<invalid>", "<EOR>", "<DOWN>", "<UP>", "A", "B", "C", "D", "E", "F", "G" 38324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver }; 39324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 40324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Test public void testNode() { 41324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeAdaptor adaptor = new CommonTreeAdaptor(); 42324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeWizard wiz = new TreeWizard(adaptor, tokens); 43324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver CommonTree t = (CommonTree)wiz.create("A"); 44324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeIterator it = new TreeIterator(t); 45324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver StringBuffer buf = toString(it); 46324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String expecting = "A EOF"; 47324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String found = buf.toString(); 48324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assertEquals(expecting, found); 49324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 50324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 51324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Test public void testFlatAB() { 52324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeAdaptor adaptor = new CommonTreeAdaptor(); 53324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeWizard wiz = new TreeWizard(adaptor, tokens); 54324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver CommonTree t = (CommonTree)wiz.create("(nil A B)"); 55324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeIterator it = new TreeIterator(t); 56324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver StringBuffer buf = toString(it); 57324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String expecting = "nil DOWN A B UP EOF"; 58324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String found = buf.toString(); 59324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assertEquals(expecting, found); 60324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 61324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 62324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Test public void testAB() { 63324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeAdaptor adaptor = new CommonTreeAdaptor(); 64324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeWizard wiz = new TreeWizard(adaptor, tokens); 65324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver CommonTree t = (CommonTree)wiz.create("(A B)"); 66324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeIterator it = new TreeIterator(t); 67324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver StringBuffer buf = toString(it); 68324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String expecting = "A DOWN B UP EOF"; 69324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String found = buf.toString(); 70324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assertEquals(expecting, found); 71324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 72324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 73324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Test public void testABC() { 74324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeAdaptor adaptor = new CommonTreeAdaptor(); 75324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeWizard wiz = new TreeWizard(adaptor, tokens); 76324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver CommonTree t = (CommonTree)wiz.create("(A B C)"); 77324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeIterator it = new TreeIterator(t); 78324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver StringBuffer buf = toString(it); 79324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String expecting = "A DOWN B C UP EOF"; 80324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String found = buf.toString(); 81324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assertEquals(expecting, found); 82324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 83324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 84324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Test public void testVerticalList() { 85324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeAdaptor adaptor = new CommonTreeAdaptor(); 86324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeWizard wiz = new TreeWizard(adaptor, tokens); 87324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver CommonTree t = (CommonTree)wiz.create("(A (B C))"); 88324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeIterator it = new TreeIterator(t); 89324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver StringBuffer buf = toString(it); 90324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String expecting = "A DOWN B DOWN C UP UP EOF"; 91324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String found = buf.toString(); 92324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assertEquals(expecting, found); 93324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 94324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 95324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Test public void testComplex() { 96324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeAdaptor adaptor = new CommonTreeAdaptor(); 97324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeWizard wiz = new TreeWizard(adaptor, tokens); 98324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver CommonTree t = (CommonTree)wiz.create("(A (B (C D E) F) G)"); 99324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeIterator it = new TreeIterator(t); 100324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver StringBuffer buf = toString(it); 101324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String expecting = "A DOWN B DOWN C DOWN D E UP F UP G UP EOF"; 102324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String found = buf.toString(); 103324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assertEquals(expecting, found); 104324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 105324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 106324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Test public void testReset() { 107324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeAdaptor adaptor = new CommonTreeAdaptor(); 108324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeWizard wiz = new TreeWizard(adaptor, tokens); 109324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver CommonTree t = (CommonTree)wiz.create("(A (B (C D E) F) G)"); 110324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver TreeIterator it = new TreeIterator(t); 111324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver StringBuffer buf = toString(it); 112324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String expecting = "A DOWN B DOWN C DOWN D E UP F UP G UP EOF"; 113324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String found = buf.toString(); 114324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assertEquals(expecting, found); 115324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 116324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver it.reset(); 117324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver buf = toString(it); 118324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver expecting = "A DOWN B DOWN C DOWN D E UP F UP G UP EOF"; 119324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver found = buf.toString(); 120324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assertEquals(expecting, found); 121324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 122324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 123324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver protected static StringBuffer toString(TreeIterator it) { 124324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver StringBuffer buf = new StringBuffer(); 125324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver while ( it.hasNext() ) { 126324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver CommonTree n = (CommonTree)it.next(); 127324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver buf.append(n); 128324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( it.hasNext() ) buf.append(" "); 129324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 130324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return buf; 131324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 132324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 133