Index: third_party/re2/util/sparse_array_test.cc |
diff --git a/third_party/re2/util/sparse_array_test.cc b/third_party/re2/util/sparse_array_test.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..bc7a19f8103d1ad10c21d1871a38aa0f1d54850d |
--- /dev/null |
+++ b/third_party/re2/util/sparse_array_test.cc |
@@ -0,0 +1,150 @@ |
+// Copyright 2006 The RE2 Authors. All Rights Reserved. |
+// Use of this source code is governed by a BSD-style |
+// license that can be found in the LICENSE file. |
+ |
+// Simple tests that SparseArray behaves. |
+ |
+#include "util/util.h" |
+#include "utest/utest.h" |
+ |
+namespace re2 { |
+ |
+static const string kNotFound = "NOT FOUND"; |
+ |
+TEST(SparseArray, BasicOperations) { |
+ static const int n = 50; |
+ SparseArray<int> set(n); |
+ |
+ int order[n]; |
+ int value[n]; |
+ for (int i = 0; i < n; i++) |
+ order[i] = i; |
+ for (int i = 0; i < n; i++) |
+ value[i] = rand()%1000 + 1; |
+ for (int i = 1; i < n; i++) { |
+ int j = rand()%i; |
+ int t = order[i]; |
+ order[i] = order[j]; |
+ order[j] = t; |
+ } |
+ |
+ for (int i = 0;; i++) { |
+ for (int j = 0; j < i; j++) { |
+ ASSERT_TRUE(set.has_index(order[j])); |
+ ASSERT_EQ(value[order[j]], set.get(order[j], -1)); |
+ } |
+ if (i >= n) |
+ break; |
+ for (int j = i; j < n; j++) |
+ ASSERT_FALSE(set.has_index(order[j])); |
+ set.set(order[i], value[order[i]]); |
+ } |
+ |
+ int nn = 0; |
+ for (SparseArray<int>::iterator i = set.begin(); i != set.end(); ++i) { |
+ ASSERT_EQ(order[nn++], i->index()); |
+ ASSERT_EQ(value[i->index()], i->value()); |
+ } |
+ ASSERT_EQ(nn, n); |
+ |
+ set.clear(); |
+ for (int i = 0; i < n; i++) |
+ ASSERT_FALSE(set.has_index(i)); |
+ |
+ ASSERT_EQ(0, set.size()); |
+ ASSERT_EQ(0, distance(set.begin(), set.end())); |
+} |
+ |
+class SparseArrayStringTest : public testing::Test { |
+ protected: |
+ SparseArrayStringTest() |
+ : str_map_(10) { |
+ InsertOrUpdate(&str_map_, 1, "a"); |
+ InsertOrUpdate(&str_map_, 5, "b"); |
+ InsertOrUpdate(&str_map_, 2, "c"); |
+ InsertOrUpdate(&str_map_, 7, "d"); |
+ } |
+ |
+ SparseArray<string> str_map_; |
+ typedef SparseArray<string>::iterator iterator; |
+}; |
+ |
+TEST_F(SparseArrayStringTest, FindGetsPresentElement) { |
+ iterator it = str_map_.find(2); |
+ ASSERT_TRUE(str_map_.end() != it); |
+ EXPECT_EQ("c", it->second); |
+} |
+ |
+TEST_F(SparseArrayStringTest, FindDoesNotFindAbsentElement) { |
+ iterator it = str_map_.find(3); |
+ ASSERT_TRUE(str_map_.end() == it); |
+} |
+ |
+TEST_F(SparseArrayStringTest, ContainsKey) { |
+ EXPECT_TRUE(ContainsKey(str_map_, 1)); |
+ EXPECT_TRUE(ContainsKey(str_map_, 2)); |
+ EXPECT_FALSE(ContainsKey(str_map_, 3)); |
+} |
+ |
+TEST_F(SparseArrayStringTest, InsertIfNotPresent) { |
+ EXPECT_FALSE(ContainsKey(str_map_, 3)); |
+ EXPECT_TRUE(InsertIfNotPresent(&str_map_, 3, "r")); |
+ EXPECT_EQ("r", FindWithDefault(str_map_, 3, kNotFound)); |
+ EXPECT_FALSE(InsertIfNotPresent(&str_map_, 3, "other value")); |
+ EXPECT_EQ("r", FindWithDefault(str_map_, 3, kNotFound)); |
+} |
+ |
+TEST(SparseArrayTest, Erase) { |
+ SparseArray<string> str_map(5); |
+ str_map.set(1, "a"); |
+ str_map.set(2, "b"); |
+ EXPECT_EQ("a", FindWithDefault(str_map, 1, kNotFound)); |
+ EXPECT_EQ("b", FindWithDefault(str_map, 2, kNotFound)); |
+ str_map.erase(1); |
+ EXPECT_EQ("NOT FOUND", FindWithDefault(str_map, 1, kNotFound)); |
+ EXPECT_EQ("b", FindWithDefault(str_map, 2, kNotFound)); |
+} |
+ |
+typedef SparseArrayStringTest SparseArrayStringSurvivesInvalidIndexTest; |
+// TODO(jyasskin): Cover invalid arguments to every method. |
+ |
+TEST_F(SparseArrayStringSurvivesInvalidIndexTest, SetNegative) { |
+ EXPECT_DEBUG_DEATH(str_map_.set(-123456789, "hi"), |
+ "\\(jyasskin\\) Illegal index -123456789 passed to" |
+ " SparseArray\\(10\\).set\\(\\)."); |
+ EXPECT_EQ(4, str_map_.size()); |
+} |
+ |
+TEST_F(SparseArrayStringSurvivesInvalidIndexTest, SetTooBig) { |
+ EXPECT_DEBUG_DEATH(str_map_.set(12345678, "hi"), |
+ "\\(jyasskin\\) Illegal index 12345678 passed to" |
+ " SparseArray\\(10\\).set\\(\\)."); |
+ EXPECT_EQ(4, str_map_.size()); |
+} |
+ |
+TEST_F(SparseArrayStringSurvivesInvalidIndexTest, SetNew_Negative) { |
+ EXPECT_DEBUG_DEATH(str_map_.set_new(-123456789, "hi"), |
+ "\\(jyasskin\\) Illegal index -123456789 passed to" |
+ " SparseArray\\(10\\).set_new\\(\\)."); |
+ EXPECT_EQ(4, str_map_.size()); |
+} |
+ |
+TEST_F(SparseArrayStringSurvivesInvalidIndexTest, SetNew_Existing) { |
+ EXPECT_DEBUG_DEATH({ |
+ str_map_.set_new(2, "hi"); |
+ EXPECT_EQ("hi", FindWithDefault(str_map_, 2, kNotFound)); |
+ |
+ // The old value for 2 is still present, but can never be removed. |
+ // This risks crashing later, if the map fills up. |
+ EXPECT_EQ(5, str_map_.size()); |
+ }, "Check failed: !has_index\\(i\\)"); |
+} |
+ |
+TEST_F(SparseArrayStringSurvivesInvalidIndexTest, SetNew_TooBig) { |
+ EXPECT_DEBUG_DEATH(str_map_.set_new(12345678, "hi"), |
+ "\\(jyasskin\\) Illegal index 12345678 passed to" |
+ " SparseArray\\(10\\).set_new\\(\\)."); |
+ EXPECT_EQ(4, str_map_.size()); |
+} |
+ |
+} // namespace re2 |