From 9c65ac73112f1a0f837cb50a2a446129acf19081 Mon Sep 17 00:00:00 2001
From: Jacques Lucke <jacques@blender.org>
Date: Tue, 28 Apr 2020 13:55:36 +0200
Subject: BLI: add Map.lookup_or_add_default method

---
 tests/gtests/blenlib/BLI_map_test.cc | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'tests/gtests')

diff --git a/tests/gtests/blenlib/BLI_map_test.cc b/tests/gtests/blenlib/BLI_map_test.cc
index e9e4b1895ab..5a19216fa7c 100644
--- a/tests/gtests/blenlib/BLI_map_test.cc
+++ b/tests/gtests/blenlib/BLI_map_test.cc
@@ -213,6 +213,17 @@ TEST(map, AddOverride)
   EXPECT_EQ(map.lookup(3), 7.0f);
 }
 
+TEST(map, LookupOrAddDefault)
+{
+  IntFloatMap map;
+  map.lookup_or_add_default(3) = 6;
+  EXPECT_EQ(map.lookup(3), 6);
+  map.lookup_or_add_default(5) = 2;
+  EXPECT_EQ(map.lookup(5), 2);
+  map.lookup_or_add_default(3) += 4;
+  EXPECT_EQ(map.lookup(3), 10);
+}
+
 TEST(map, MoveConstructorSmall)
 {
   IntFloatMap map1;
-- 
cgit v1.2.3