chiark / gitweb /
Apply upstream revision 1631 (Closes: #815921)
[pcre3.git] / pcre_scanner_unittest.cc
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
59 using std::vector;
60 using pcrecpp::StringPiece;
61 using pcrecpp::Scanner;
62
63 static 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
129 static 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
151 int 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 }