src/stl_logging_unittest.cc

   1 // Copyright (c) 2003, 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 #include "config.h"
  31
  32 #ifdef HAVE_USING_OPERATOR
  33
  34 #include "glog/stl_logging.h"
  35
  36 #include <iostream>
  37 #include <map>
  38 #include <string>
  39 #include <strstream>
  40 #include <vector>
  41
  42 #ifdef __GNUC__
  43 # include <ext/hash_map>
  44 # include <ext/hash_set>
  45 #endif
  46
  47 #include "glog/logging.h"
  48 #include "googletest.h"
  49
  50 using namespace std;
  51 #ifdef __GNUC__
  52 using namespace __gnu_cxx;
  53 #endif
  54
  55 struct user_hash {
  56   size_t operator()(int x) const { return x; }
  57 };
  58
  59 void TestSTLLogging() {
  60   {
  61     // Test a sequence.
  62     vector<int> v;
  63     v.push_back(10);
  64     v.push_back(20);
  65     v.push_back(30);
  66     char ss_buf[1000];
  67     ostrstream ss(ss_buf, sizeof(ss_buf));
  68     // Just ostrstream s1; leaks heap.
  69     ss << v << ends;
  70     CHECK_STREQ(ss.str(), "10 20 30");
  71     vector<int> copied_v(v);
  72     CHECK_EQ(v, copied_v);  // This must compile.
  73   }
  74
  75   {
  76     // Test a sorted pair associative container.
  77     map< int, string > m;
  78     m[20] = "twenty";
  79     m[10] = "ten";
  80     m[30] = "thirty";
  81     char ss_buf[1000];
  82     ostrstream ss(ss_buf, sizeof(ss_buf));
  83     ss << m << ends;
  84     CHECK_STREQ(ss.str(), "(10, ten) (20, twenty) (30, thirty)");
  85     map< int, string > copied_m(m);
  86     CHECK_EQ(m, copied_m);  // This must compile.
  87   }
  88
  89 #ifdef __GNUC__
  90   {
  91     // Test a hashed simple associative container.
  92     hash_set<int> hs;
  93     hs.insert(10);
  94     hs.insert(20);
  95     hs.insert(30);
  96     char ss_buf[1000];
  97     ostrstream ss(ss_buf, sizeof(ss_buf));
  98     ss << hs << ends;
  99     CHECK_STREQ(ss.str(), "10 20 30");
 100     hash_set<int> copied_hs(hs);
 101     CHECK_EQ(hs, copied_hs);  // This must compile.
 102   }
 103 #endif
 104
 105 #ifdef __GNUC__
 106   {
 107     // Test a hashed pair associative container.
 108     hash_map<int, string> hm;
 109     hm[10] = "ten";
 110     hm[20] = "twenty";
 111     hm[30] = "thirty";
 112     char ss_buf[1000];
 113     ostrstream ss(ss_buf, sizeof(ss_buf));
 114     ss << hm << ends;
 115     CHECK_STREQ(ss.str(), "(10, ten) (20, twenty) (30, thirty)");
 116     hash_map<int, string> copied_hm(hm);
 117     CHECK_EQ(hm, copied_hm);  // this must compile
 118   }
 119 #endif
 120
 121   {
 122     // Test a long sequence.
 123     vector<int> v;
 124     string expected;
 125     for (int i = 0; i < 100; i++) {
 126       v.push_back(i);
 127       if (i > 0) expected += ' ';
 128       char buf[256];
 129       sprintf(buf, "%d", i);
 130       expected += buf;
 131     }
 132     v.push_back(100);
 133     expected += " ...";
 134     char ss_buf[1000];
 135     ostrstream ss(ss_buf, sizeof(ss_buf));
 136     ss << v << ends;
 137     CHECK_STREQ(ss.str(), expected.c_str());
 138   }
 139
 140   {
 141     // Test a sorted pair associative container.
 142     // Use a non-default comparison functor.
 143     map< int, string, greater<int> > m;
 144     m[20] = "twenty";
 145     m[10] = "ten";
 146     m[30] = "thirty";
 147     char ss_buf[1000];
 148     ostrstream ss(ss_buf, sizeof(ss_buf));
 149     ss << m << ends;
 150     CHECK_STREQ(ss.str(), "(30, thirty) (20, twenty) (10, ten)");
 151     map< int, string, greater<int> > copied_m(m);
 152     CHECK_EQ(m, copied_m);  // This must compile.
 153   }
 154
 155 #ifdef __GNUC__
 156   {
 157     // Test a hashed simple associative container.
 158     // Use a user defined hash function.
 159     hash_set<int, user_hash> hs;
 160     hs.insert(10);
 161     hs.insert(20);
 162     hs.insert(30);
 163     char ss_buf[1000];
 164     ostrstream ss(ss_buf, sizeof(ss_buf));
 165     ss << hs << ends;
 166     CHECK_STREQ(ss.str(), "10 20 30");
 167     hash_set<int, user_hash> copied_hs(hs);
 168     CHECK_EQ(hs, copied_hs);  // This must compile.
 169   }
 170 #endif
 171 }
 172
 173 int main(int argc, char** argv) {
 174   TestSTLLogging();
 175   std::cout << "PASS\n";
 176   return 0;
 177 }
 178
 179 #else
 180
 181 #include <iostream>
 182
 183 int main(int argc, char** argv) {
 184   std::cout << "We don't support stl_logging for this compiler.\n"
 185             << "(we need compiler support of 'using ::operator<<' "
 186             << "for this feature.)\n";
 187   return 0;
 188 }
 189
 190 #endif  // HAVE_USING_OPERATOR