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.h>      /* for strchr */
41#include <string>
42#include <vector>
43
44#include "pcrecpp.h"
45#include "pcre_stringpiece.h"
46#include "pcre_scanner.h"
47
48#define FLAGS_unittest_stack_size   49152
49
50// Dies with a fatal error if the two values are not equal.
51#define CHECK_EQ(a, b)  do {                                    \
52  if ( (a) != (b) ) {                                           \
53    fprintf(stderr, "%s:%d: Check failed because %s != %s\n",   \
54            __FILE__, __LINE__, #a, #b);                        \
55    exit(1);                                                    \
56  }                                                             \
57} while (0)
58
59using std::vector;
60using pcrecpp::StringPiece;
61using pcrecpp::Scanner;
62
63static void TestScanner() {
64  const char input[] = "\n"
65                       "alpha = 1; // this sets alpha\n"
66                       "bravo = 2; // bravo is set here\n"
67                       "gamma = 33; /* and here is gamma */\n";
68
69  const char *re = "(\\w+) = (\\d+);";
70
71  Scanner s(input);
72  string var;
73  int number;
74  s.SkipCXXComments();
75  s.set_save_comments(true);
76  vector<StringPiece> comments;
77
78  s.Consume(re, &var, &number);
79  CHECK_EQ(var, "alpha");
80  CHECK_EQ(number, 1);
81  CHECK_EQ(s.LineNumber(), 3);
82  s.GetNextComments(&comments);
83  CHECK_EQ(comments.size(), 1);
84  CHECK_EQ(comments[0].as_string(), " // this sets alpha\n");
85  comments.resize(0);
86
87  s.Consume(re, &var, &number);
88  CHECK_EQ(var, "bravo");
89  CHECK_EQ(number, 2);
90  s.GetNextComments(&comments);
91  CHECK_EQ(comments.size(), 1);
92  CHECK_EQ(comments[0].as_string(), " // bravo is set here\n");
93  comments.resize(0);
94
95  s.Consume(re, &var, &number);
96  CHECK_EQ(var, "gamma");
97  CHECK_EQ(number, 33);
98  s.GetNextComments(&comments);
99  CHECK_EQ(comments.size(), 1);
100  CHECK_EQ(comments[0].as_string(), " /* and here is gamma */\n");
101  comments.resize(0);
102
103  s.GetComments(0, sizeof(input), &comments);
104  CHECK_EQ(comments.size(), 3);
105  CHECK_EQ(comments[0].as_string(), " // this sets alpha\n");
106  CHECK_EQ(comments[1].as_string(), " // bravo is set here\n");
107  CHECK_EQ(comments[2].as_string(), " /* and here is gamma */\n");
108  comments.resize(0);
109
110  s.GetComments(0, (int)(strchr(input, '/') - input), &comments);
111  CHECK_EQ(comments.size(), 0);
112  comments.resize(0);
113
114  s.GetComments((int)(strchr(input, '/') - input - 1), sizeof(input),
115                &comments);
116  CHECK_EQ(comments.size(), 3);
117  CHECK_EQ(comments[0].as_string(), " // this sets alpha\n");
118  CHECK_EQ(comments[1].as_string(), " // bravo is set here\n");
119  CHECK_EQ(comments[2].as_string(), " /* and here is gamma */\n");
120  comments.resize(0);
121
122  s.GetComments((int)(strchr(input, '/') - input - 1),
123                (int)(strchr(input + 1, '\n') - input + 1), &comments);
124  CHECK_EQ(comments.size(), 1);
125  CHECK_EQ(comments[0].as_string(), " // this sets alpha\n");
126  comments.resize(0);
127}
128
129static void TestBigComment() {
130  string input;
131  for (int i = 0; i < 1024; ++i) {
132    char buf[1024];  // definitely big enough
133    sprintf(buf, "    # Comment %d\n", i);
134    input += buf;
135  }
136  input += "name = value;\n";
137
138  Scanner s(input.c_str());
139  s.SetSkipExpression("\\s+|#.*\n");
140
141  string name;
142  string value;
143  s.Consume("(\\w+) = (\\w+);", &name, &value);
144  CHECK_EQ(name, "name");
145  CHECK_EQ(value, "value");
146}
147
148// TODO: also test scanner and big-comment in a thread with a
149//       small stack size
150
151int main(int argc, char** argv) {
152  (void)argc;
153  (void)argv;
154  TestScanner();
155  TestBigComment();
156
157  // Done
158  printf("OK\n");
159
160  return 0;
161}
162