NCBI C++ Toolkit Cross Reference

   // Copyright (c) 2005, Google Inc.
   // All rights reserved.
   //
   // Redistribution and use in source and binary forms, with or without
   // modification, are permitted provided that the following conditions are
   // met:
   //
   //     * Redistributions of source code must retain the above copyright
   // notice, this list of conditions and the following disclaimer.
  //     * Redistributions in binary form must reproduce the above
  // copyright notice, this list of conditions and the following disclaimer
  // in the documentation and/or other materials provided with the
  // distribution.
  //     * Neither the name of Google Inc. nor the names of its
  // contributors may be used to endorse or promote products derived from
  // this software without specific prior written permission.
  //
  // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  //
  // Author: Greg J. Badros
  //
  // Unittest for scanner, especially GetNextComments and GetComments()
  // functionality.
  
  #ifdef HAVE_CONFIG_H
  #include "config.h"
  #endif
  
  #include <stdio.h>
  #include <string>
  #include <vector>
  
  #include "pcrecpp.h"
  #include "pcre_stringpiece.h"
  #include "pcre_scanner.h"
  
  #define FLAGS_unittest_stack_size   49152
  
  // Dies with a fatal error if the two values are not equal.
  #define CHECK_EQ(a, b)  do {                                    \
    if ( (a) != (b) ) {                                           \
      fprintf(stderr, "%s:%d: Check failed because %s != %s\n",   \
              __FILE__, __LINE__, #a, #b);                        \
      exit(1);                                                    \
    }                                                             \
  } while (0)
  
  using std::vector;
  using pcrecpp::StringPiece;
  using pcrecpp::Scanner;
  
  static void TestScanner() {
    const char input[] = "\n"
                         "alpha = 1; // this sets alpha\n"
                         "bravo = 2; // bravo is set here\n"
                         "gamma = 33; /* and here is gamma */\n";
  
    const char *re = "(\\w+) = (\\d+);";
  
    Scanner s(input);
    string var;
    int number;
    s.SkipCXXComments();
    s.set_save_comments(true);
    vector<StringPiece> comments;
  
    s.Consume(re, &var, &number);
    CHECK_EQ(var, "alpha");
    CHECK_EQ(number, 1);
    CHECK_EQ(s.LineNumber(), 3);
    s.GetNextComments(&comments);
    CHECK_EQ(comments.size(), 1);
    CHECK_EQ(comments[0].as_string(), " // this sets alpha\n");
    comments.resize(0);
  
    s.Consume(re, &var, &number);
    CHECK_EQ(var, "bravo");
    CHECK_EQ(number, 2);
    s.GetNextComments(&comments);
    CHECK_EQ(comments.size(), 1);
    CHECK_EQ(comments[0].as_string(), " // bravo is set here\n");
    comments.resize(0);
  
    s.Consume(re, &var, &number);
    CHECK_EQ(var, "gamma");
    CHECK_EQ(number, 33);
    s.GetNextComments(&comments);
    CHECK_EQ(comments.size(), 1);
    CHECK_EQ(comments[0].as_string(), " /* and here is gamma */\n");
   comments.resize(0);
 
   s.GetComments(0, sizeof(input), &comments);
   CHECK_EQ(comments.size(), 3);
   CHECK_EQ(comments[0].as_string(), " // this sets alpha\n");
   CHECK_EQ(comments[1].as_string(), " // bravo is set here\n");
   CHECK_EQ(comments[2].as_string(), " /* and here is gamma */\n");
   comments.resize(0);
 
   s.GetComments(0, strchr(input, '/') - input, &comments);
   CHECK_EQ(comments.size(), 0);
   comments.resize(0);
 
   s.GetComments(strchr(input, '/') - input - 1, sizeof(input),
                 &comments);
   CHECK_EQ(comments.size(), 3);
   CHECK_EQ(comments[0].as_string(), " // this sets alpha\n");
   CHECK_EQ(comments[1].as_string(), " // bravo is set here\n");
   CHECK_EQ(comments[2].as_string(), " /* and here is gamma */\n");
   comments.resize(0);
 
   s.GetComments(strchr(input, '/') - input - 1,
                 strchr(input + 1, '\n') - input + 1, &comments);
   CHECK_EQ(comments.size(), 1);
   CHECK_EQ(comments[0].as_string(), " // this sets alpha\n");
   comments.resize(0);
 }
 
 static void TestBigComment() {
   string input;
   for (int i = 0; i < 1024; ++i) {
     char buf[1024];  // definitely big enough
     sprintf(buf, "    # Comment %d\n", i);
     input += buf;
   }
   input += "name = value;\n";
 
   Scanner s(input.c_str());
   s.SetSkipExpression("\\s+|#.*\n");
 
   string name;
   string value;
   s.Consume("(\\w+) = (\\w+);", &name, &value);
   CHECK_EQ(name, "name");
   CHECK_EQ(value, "value");
 }
 
 // TODO: also test scanner and big-comment in a thread with a
 //       small stack size
 
 int main(int argc, char** argv) {
   TestScanner();
   TestBigComment();
 
   // Done
   printf("OK\n");
 
   return 0;
 }