1 // [The "BSD licence"] 2 // Copyright (c) 2007 Kay Roepke 3 // All rights reserved. 4 // 5 // Redistribution and use in source and binary forms, with or without 6 // modification, are permitted provided that the following conditions 7 // are met: 8 // 1. Redistributions of source code must retain the above copyright 9 // notice, this list of conditions and the following disclaimer. 10 // 2. Redistributions in binary form must reproduce the above copyright 11 // notice, this list of conditions and the following disclaimer in the 12 // documentation and/or other materials provided with the distribution. 13 // 3. The name of the author may not be used to endorse or promote products 14 // derived from this software without specific prior written permission. 15 // 16 // THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 17 // IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 18 // OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 19 // IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 20 // INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 21 // NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 25 // THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 27 #import "TestRewriteRuleTokenStream.h" 28 #import "ANTLRRewriteRuleTokenStream.h" 29 #import "ANTLRCommonTreeAdaptor.h" 30 #import "ANTLRCommonToken.h" 31 32 @implementation TestRewriteRuleTokenStream 33 34 - (void) setUp 35 { 36 treeAdaptor = [ANTLRCommonTreeAdaptor newTreeAdaptor]; 37 stream = [ANTLRRewriteRuleTokenStream newANTLRRewriteRuleTokenStream:treeAdaptor 38 description:@"rewrite rule token stream"]; 39 token1 = [ANTLRCommonToken newToken:5]; 40 token2 = [ANTLRCommonToken newToken:6]; 41 token3 = [ANTLRCommonToken newToken:7]; 42 token4 = [ANTLRCommonToken newToken:8]; 43 [token1 setText:@"token 1"]; 44 [token2 setText:@"token 2"]; 45 [token3 setText:@"token 3"]; 46 [token4 setText:@"token 4"]; 47 } 48 49 - (void) tearDown 50 { 51 [token1 release]; token1 = nil; 52 [token2 release]; token2 = nil; 53 [token3 release]; token3 = nil; 54 [token4 release]; token4 = nil; 55 56 [treeAdaptor release]; treeAdaptor = nil; 57 [stream release]; stream = nil; 58 } 59 60 - (void) test01EmptyRewriteStream 61 { 62 treeAdaptor = [ANTLRCommonTreeAdaptor newTreeAdaptor]; 63 stream = [ANTLRRewriteRuleTokenStream newANTLRRewriteRuleTokenStream:treeAdaptor 64 description:@"rewrite rule token stream"]; 65 STAssertFalse([stream hasNext], @"-(BOOL)hasNext should be NO, but isn't"); 66 STAssertThrows([stream nextToken], @"-next on empty stream should throw exception, but doesn't"); 67 } 68 69 - (void) test02RewriteStreamCount 70 { 71 treeAdaptor = [ANTLRCommonTreeAdaptor newTreeAdaptor]; 72 stream = [ANTLRRewriteRuleTokenStream newANTLRRewriteRuleTokenStream:treeAdaptor 73 description:@"rewrite rule token stream"]; 74 token1 = [ANTLRCommonToken newToken:5]; 75 token2 = [ANTLRCommonToken newToken:6]; 76 [token1 setText:@"token 1"]; 77 [token2 setText:@"token 2"]; 78 STAssertTrue([stream size] == 0, 79 @"empty stream should have count==0"); 80 [stream addElement:token1]; 81 STAssertTrue([stream size] == 1, 82 @"single element stream should have count==1"); 83 [stream addElement:token2]; 84 STAssertTrue([stream size] == 2, 85 @"multiple stream should have count==2"); 86 87 } 88 89 - (void) test03SingleElement 90 { 91 treeAdaptor = [ANTLRCommonTreeAdaptor newTreeAdaptor]; 92 stream = [ANTLRRewriteRuleTokenStream newANTLRRewriteRuleTokenStream:treeAdaptor 93 description:@"rewrite rule token stream"]; 94 token1 = [ANTLRCommonToken newToken:5]; 95 token2 = [ANTLRCommonToken newToken:6]; 96 token3 = [ANTLRCommonToken newToken:7]; 97 token4 = [ANTLRCommonToken newToken:8]; 98 [token1 setText:@"token 1"]; 99 [token2 setText:@"token 2"]; 100 [token3 setText:@"token 3"]; 101 [token4 setText:@"token 4"]; 102 [stream addElement:token1]; 103 STAssertTrue([stream hasNext], @"-hasNext should be YES, but isn't"); 104 ANTLRCommonTree *tree = [stream nextNode]; 105 STAssertEqualObjects([tree getToken], token1, @"return token from stream should be token1, but isn't"); 106 } 107 108 - (void) test04SingleElementDup 109 { 110 [stream addElement:token1]; 111 ANTLRCommonTree *tree1, *tree2; 112 STAssertNoThrow(tree1 = [stream nextNode], 113 @"stream iteration should not throw exception" 114 ); 115 STAssertNoThrow(tree2 = [stream nextNode], 116 @"stream iteration past element count (single element) should not throw exception" 117 ); 118 STAssertEqualObjects([tree1 getToken], [tree2 getToken], 119 @"tokens should be the same"); 120 STAssertFalse(tree1 == tree2, 121 @"trees should be different, but aren't"); 122 } 123 124 - (void) test05MultipleElements 125 { 126 [stream addElement:token1]; 127 [stream addElement:token2]; 128 [stream addElement:token3]; 129 ANTLRCommonTree *tree1, *tree2, *tree3, *tree4; 130 STAssertNoThrow(tree1 = [stream nextNode], 131 @"stream iteration should not throw exception" 132 ); 133 STAssertEqualObjects([tree1 getToken], token1, 134 @"[tree1 token] should be equal to token1" 135 ); 136 STAssertNoThrow(tree2 = [stream nextNode], 137 @"stream iteration should not throw exception" 138 ); 139 STAssertEqualObjects([tree2 getToken], token2, 140 @"[tree2 token] should be equal to token2" 141 ); 142 STAssertNoThrow(tree3 = [stream nextNode], 143 @"stream iteration should not throw exception" 144 ); 145 STAssertEqualObjects([tree3 getToken], token3, 146 @"[tree3 token] should be equal to token3" 147 ); 148 STAssertThrows(tree4 = [stream nextNode], 149 @"iterating beyond end of stream should throw an exception" 150 ); 151 } 152 153 - (void) test06MultipleElementsAfterReset 154 { 155 [stream addElement:token1]; 156 [stream addElement:token2]; 157 [stream addElement:token3]; 158 ANTLRCommonTree *tree1, *tree2, *tree3; 159 160 // consume the stream completely 161 STAssertNoThrow(tree1 = [stream nextNode], 162 @"stream iteration should not throw exception" 163 ); 164 STAssertEqualObjects([tree1 getToken], token1, 165 @"[tree1 token] should be equal to token1" 166 ); 167 STAssertNoThrow(tree2 = [stream nextNode], 168 @"stream iteration should not throw exception" 169 ); 170 STAssertEqualObjects([tree2 getToken], token2, 171 @"[tree2 token] should be equal to token2" 172 ); 173 STAssertNoThrow(tree3 = [stream nextNode], 174 @"stream iteration should not throw exception" 175 ); 176 177 [stream reset]; // after resetting the stream it should dup 178 179 ANTLRCommonTree *tree1Dup, *tree2Dup, *tree3Dup; 180 181 STAssertNoThrow(tree1Dup = [stream nextNode], 182 @"stream iteration should not throw exception" 183 ); 184 STAssertTrue(tree1 != tree1Dup, 185 @"[tree1 token] should be equal to token1" 186 ); 187 STAssertNoThrow(tree2Dup = [stream nextNode], 188 @"stream iteration should not throw exception" 189 ); 190 STAssertTrue(tree2 != tree2Dup, 191 @"[tree2 token] should be equal to token2" 192 ); 193 STAssertNoThrow(tree3Dup = [stream nextNode], 194 @"stream iteration should not throw exception" 195 ); 196 STAssertTrue(tree3 != tree3Dup, 197 @"[tree3 token] should be equal to token3" 198 ); 199 } 200 201 @end 202