Support C++11 range-based for loops over BSONObj

1 files changed, 21 insertions, 0 deletions

diff --git a/src/mongo/bson/bson_obj_test.cpp b/src/mongo/bson/bson_obj_test.cpp
index 443700f037d..6befb751760 100644
--- a/src/mongo/bson/bson_obj_test.cpp
+++ b/src/mongo/bson/bson_obj_test.cpp
@@ -31,6 +31,7 @@
 #include "mongo/unittest/unittest.h"
 
 namespace {
+    using namespace mongo;
 
     TEST(BSONObjToString, EmptyArray) {
         const char text[] = "{ x: [] }";
@@ -182,4 +183,24 @@ namespace {
         }
     }
 
+    TEST(Looping, Cpp11Basic) {
+        int count = 0;
+        for (BSONElement e : BSON("a" << 1 << "a" << 2 << "a" << 3)) {
+            ASSERT_EQUALS( e.fieldNameStringData() , "a" );
+            count += e.Int();
+        }
+
+        ASSERT_EQUALS( count , 1 + 2 + 3 );
+    }
+
+    TEST(Looping, Cpp11Auto) {
+        int count = 0;
+        for (auto e : BSON("a" << 1 << "a" << 2 << "a" << 3)) {
+            ASSERT_EQUALS( e.fieldNameStringData() , "a" );
+            count += e.Int();
+        }
+
+        ASSERT_EQUALS( count , 1 + 2 + 3 );
+    }
+
 } // unnamed namespace