SERVER-9515 Track indirect roles for roles in the RoleGraph

1 files changed, 24 insertions, 0 deletions

diff --git a/src/mongo/db/auth/role_graph_test.cpp b/src/mongo/db/auth/role_graph_test.cpp
index 6f907cc67d3..3abb14974a1 100644
--- a/src/mongo/db/auth/role_graph_test.cpp
+++ b/src/mongo/db/auth/role_graph_test.cpp
@@ -108,6 +108,30 @@ namespace {
         }
         ASSERT_FALSE(it.more());
 
+        it = graph.getIndirectSubordinates(roleA); // should have roleB, roleC and roleD
+        bool hasB = false;
+        bool hasC = false;
+        bool hasD = false;
+        int num = 0;
+        while (it.more()) {
+            ++num;
+            RoleName cur = it.next();
+            if (cur == roleB) {
+                hasB = true;
+            } else if (cur == roleC) {
+                hasC = true;
+            } else if (cur == roleD) {
+                hasD = true;
+            } else {
+                FAIL(mongoutils::str::stream() << "unexpected role returned: " <<
+                        cur.getFullName());
+            }
+        }
+        ASSERT_EQUALS(3, num);
+        ASSERT(hasB);
+        ASSERT(hasC);
+        ASSERT(hasD);
+
         it = graph.getDirectSubordinates(roleB); // should be roleC and roleD, order doesn't matter
         cur = it.next();
         if (cur == roleC) {