1// Copyright (c) 2005, Google Inc. 2// All rights reserved. 3// 4// Redistribution and use in source and binary forms, with or without 5// modification, are permitted provided that the following conditions are 6// met: 7// 8// * Redistributions of source code must retain the above copyright 9// notice, this list of conditions and the following disclaimer. 10// * Redistributions in binary form must reproduce the above 11// copyright notice, this list of conditions and the following disclaimer 12// in the documentation and/or other materials provided with the 13// distribution. 14// * Neither the name of Google Inc. nor the names of its 15// contributors may be used to endorse or promote products derived from 16// this software without specific prior written permission. 17// 18// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 19// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 20// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 21// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 22// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 23// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 24// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 25// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 26// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 28// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 29// 30// Author: Greg J. Badros 31// 32// Unittest for scanner, especially GetNextComments and GetComments() 33// functionality. 34 35#ifdef HAVE_CONFIG_H 36#include "config.h" 37#endif 38 39#include <stdio.h> 40#include <string> 41#include <vector> 42 43#include "pcrecpp.h" 44#include "pcre_stringpiece.h" 45#include "pcre_scanner.h" 46 47#define FLAGS_unittest_stack_size 49152 48 49// Dies with a fatal error if the two values are not equal. 50#define CHECK_EQ(a, b) do { \ 51 if ( (a) != (b) ) { \ 52 fprintf(stderr, "%s:%d: Check failed because %s != %s\n", \ 53 __FILE__, __LINE__, #a, #b); \ 54 exit(1); \ 55 } \ 56} while (0) 57 58using std::vector; 59using pcrecpp::StringPiece; 60using pcrecpp::Scanner; 61 62static void TestScanner() { 63 const char input[] = "\n" 64 "alpha = 1; // this sets alpha\n" 65 "bravo = 2; // bravo is set here\n" 66 "gamma = 33; /* and here is gamma */\n"; 67 68 const char *re = "(\\w+) = (\\d+);"; 69 70 Scanner s(input); 71 string var; 72 int number; 73 s.SkipCXXComments(); 74 s.set_save_comments(true); 75 vector<StringPiece> comments; 76 77 s.Consume(re, &var, &number); 78 CHECK_EQ(var, "alpha"); 79 CHECK_EQ(number, 1); 80 CHECK_EQ(s.LineNumber(), 3); 81 s.GetNextComments(&comments); 82 CHECK_EQ(comments.size(), 1); 83 CHECK_EQ(comments[0].as_string(), " // this sets alpha\n"); 84 comments.resize(0); 85 86 s.Consume(re, &var, &number); 87 CHECK_EQ(var, "bravo"); 88 CHECK_EQ(number, 2); 89 s.GetNextComments(&comments); 90 CHECK_EQ(comments.size(), 1); 91 CHECK_EQ(comments[0].as_string(), " // bravo is set here\n"); 92 comments.resize(0); 93 94 s.Consume(re, &var, &number); 95 CHECK_EQ(var, "gamma"); 96 CHECK_EQ(number, 33); 97 s.GetNextComments(&comments); 98 CHECK_EQ(comments.size(), 1); 99 CHECK_EQ(comments[0].as_string(), " /* and here is gamma */\n"); 100 comments.resize(0); 101 102 s.GetComments(0, sizeof(input), &comments); 103 CHECK_EQ(comments.size(), 3); 104 CHECK_EQ(comments[0].as_string(), " // this sets alpha\n"); 105 CHECK_EQ(comments[1].as_string(), " // bravo is set here\n"); 106 CHECK_EQ(comments[2].as_string(), " /* and here is gamma */\n"); 107 comments.resize(0); 108 109 s.GetComments(0, strchr(input, '/') - input, &comments); 110 CHECK_EQ(comments.size(), 0); 111 comments.resize(0); 112 113 s.GetComments(strchr(input, '/') - input - 1, sizeof(input), 114 &comments); 115 CHECK_EQ(comments.size(), 3); 116 CHECK_EQ(comments[0].as_string(), " // this sets alpha\n"); 117 CHECK_EQ(comments[1].as_string(), " // bravo is set here\n"); 118 CHECK_EQ(comments[2].as_string(), " /* and here is gamma */\n"); 119 comments.resize(0); 120 121 s.GetComments(strchr(input, '/') - input - 1, 122 strchr(input + 1, '\n') - input + 1, &comments); 123 CHECK_EQ(comments.size(), 1); 124 CHECK_EQ(comments[0].as_string(), " // this sets alpha\n"); 125 comments.resize(0); 126} 127 128static void TestBigComment() { 129 string input; 130 for (int i = 0; i < 1024; ++i) { 131 char buf[1024]; // definitely big enough 132 sprintf(buf, " # Comment %d\n", i); 133 input += buf; 134 } 135 input += "name = value;\n"; 136 137 Scanner s(input.c_str()); 138 s.SetSkipExpression("\\s+|#.*\n"); 139 140 string name; 141 string value; 142 s.Consume("(\\w+) = (\\w+);", &name, &value); 143 CHECK_EQ(name, "name"); 144 CHECK_EQ(value, "value"); 145} 146 147// TODO: also test scanner and big-comment in a thread with a 148// small stack size 149 150int main(int argc, char** argv) { 151 TestScanner(); 152 TestBigComment(); 153 154 // Done 155 printf("OK\n"); 156 157 return 0; 158} 159