[Bf-blender-cvs] [6c78f41a2ff] functions: add StringMap tests

[Jacques Lucke]

Commit: 6c78f41a2ff2512294abf8008f7b264632539de9
Author: Jacques Lucke
Date:   Tue Aug 20 18:31:19 2019 +0200
Branches: functions
https://developer.blender.org/rB6c78f41a2ff2512294abf8008f7b264632539de9

add StringMap tests

===================================================================

A	tests/gtests/blenlib/BLI_string_map_test.cc
M	tests/gtests/blenlib/CMakeLists.txt

===================================================================

diff --git a/tests/gtests/blenlib/BLI_string_map_test.cc b/tests/gtests/blenlib/BLI_string_map_test.cc
new file mode 100644
index 00000000000..753ef4876e7
--- /dev/null
+++ b/tests/gtests/blenlib/BLI_string_map_test.cc
@@ -0,0 +1,158 @@
+#include "testing/testing.h"
+#include "BLI_string_map.hpp"
+#include "BLI_vector.hpp"
+
+using namespace BLI;
+
+TEST(string_map, DefaultConstructor)
+{
+  StringMap<int> map;
+  EXPECT_EQ(map.size(), 0);
+}
+
+TEST(string_map, AddNew)
+{
+  StringMap<int> map;
+  EXPECT_EQ(map.size(), 0);
+
+  map.add_new("Why", 5);
+  EXPECT_EQ(map.size(), 1);
+  EXPECT_EQ(map.lookup("Why"), 5);
+
+  map.add_new("Where", 6);
+  EXPECT_EQ(map.size(), 2);
+  EXPECT_EQ(map.lookup("Where"), 6);
+}
+
+TEST(string_map, AddNew_Many)
+{
+  StringMap<int> map;
+
+  for (uint i = 0; i < 100; i++) {
+    map.add_new(std::to_string(i), i);
+  }
+  EXPECT_EQ(map.size(), 100);
+}
+
+TEST(string_map, Contains)
+{
+  StringMap<int> map;
+  map.add_new("A", 0);
+  map.add_new("B", 0);
+  EXPECT_TRUE(map.contains("A"));
+  EXPECT_TRUE(map.contains("B"));
+  EXPECT_FALSE(map.contains("C"));
+}
+
+TEST(string_map, Contains_Many)
+{
+  StringMap<int> map;
+  for (uint i = 0; i < 50; i++) {
+    map.add_new(std::to_string(i), i);
+  }
+  for (uint i = 100; i < 200; i++) {
+    map.add_new(std::to_string(i), i);
+  }
+  EXPECT_EQ(map.size(), 150);
+  for (uint i = 0; i < 200; i++) {
+    if (i < 50 || i >= 100) {
+      EXPECT_TRUE(map.contains(std::to_string(i)));
+    }
+    else {
+      EXPECT_FALSE(map.contains(std::to_string(i)));
+    }
+  }
+}
+
+TEST(string_map, Lookup)
+{
+  StringMap<int> map;
+  map.add_new("A", 5);
+  map.add_new("B", 8);
+  map.add_new("C", 10);
+  EXPECT_EQ(map.lookup("A"), 5);
+  EXPECT_EQ(map.lookup("B"), 8);
+  EXPECT_EQ(map.lookup("C"), 10);
+}
+
+TEST(string_map, LookupPtr)
+{
+  StringMap<int> map;
+  map.add_new("test1", 13);
+  map.add_new("test2", 14);
+  map.add_new("test3", 15);
+  EXPECT_EQ(*map.lookup_ptr("test1"), 13);
+  EXPECT_EQ(*map.lookup_ptr("test2"), 14);
+  EXPECT_EQ(*map.lookup_ptr("test3"), 15);
+  EXPECT_EQ(map.lookup_ptr("test4"), nullptr);
+}
+
+TEST(string_map, LookupDefault)
+{
+  StringMap<int> map;
+  EXPECT_EQ(map.lookup_default("test", 42), 42);
+  map.add_new("test", 5);
+  EXPECT_EQ(map.lookup_default("test", 42), 5);
+}
+
+TEST(string_map, FindKeyForValue)
+{
+  StringMap<int> map;
+  map.add_new("A", 1);
+  map.add_new("B", 2);
+  map.add_new("C", 3);
+  EXPECT_EQ(map.find_key_for_value(1), "A");
+  EXPECT_EQ(map.find_key_for_value(2), "B");
+  EXPECT_EQ(map.find_key_for_value(3), "C");
+}
+
+TEST(string_map, ForeachValue)
+{
+  StringMap<int> map;
+  map.add_new("A", 4);
+  map.add_new("B", 5);
+  map.add_new("C", 1);
+
+  Vector<int> values;
+  map.foreach_value([&values](int &value) { values.append(value); });
+  EXPECT_EQ(values.size(), 3);
+  EXPECT_TRUE(values.contains(1));
+  EXPECT_TRUE(values.contains(4));
+  EXPECT_TRUE(values.contains(5));
+}
+
+TEST(string_map, ForeachKey)
+{
+  StringMap<int> map;
+  map.add_new("A", 4);
+  map.add_new("B", 5);
+  map.add_new("C", 1);
+
+  Vector<std::string> keys;
+  map.foreach_key([&keys](StringRefNull key) { keys.append(key.to_std_string()); });
+  EXPECT_EQ(keys.size(), 3);
+  EXPECT_TRUE(keys.contains("A"));
+  EXPECT_TRUE(keys.contains("B"));
+  EXPECT_TRUE(keys.contains("C"));
+}
+
+TEST(string_map, ForeachKeyValuePair)
+{
+  StringMap<int> map;
+  map.add_new("A", 4);
+  map.add_new("B", 5);
+  map.add_new("C", 1);
+
+  Vector<std::string> keys;
+  Vector<int> values;
+
+  map.foreach_key_value_pair([&keys, &values](StringRefNull key, int value) {
+    keys.append(key.to_std_string());
+    values.append(value);
+  });
+
+  EXPECT_EQ(keys.size(), 3);
+  EXPECT_EQ(values[keys.index("A")], 4);
+  EXPECT_EQ(values[keys.index("B")], 5);
+  EXPECT_EQ(values[keys.index("C")], 1);
+}
diff --git a/tests/gtests/blenlib/CMakeLists.txt b/tests/gtests/blenlib/CMakeLists.txt
index 6b7acb20bca..35ecc77ecf9 100644
--- a/tests/gtests/blenlib/CMakeLists.txt
+++ b/tests/gtests/blenlib/CMakeLists.txt
@@ -72,6 +72,7 @@ BLENDER_TEST(BLI_map "bf_blenlib")
 BLENDER_TEST(BLI_multi_map "bf_blenlib")
 BLENDER_TEST(BLI_stack "bf_blenlib")
 BLENDER_TEST(BLI_string "bf_blenlib")
+BLENDER_TEST(BLI_string_map "bf_blenlib")
 BLENDER_TEST(BLI_string_ref "bf_blenlib")
 BLENDER_TEST(BLI_string_utf8 "bf_blenlib")
 BLENDER_TEST(BLI_task "bf_blenlib;bf_intern_numaapi")