Fix ObjectRepositoryRoleProviderTest

Use RecursiveRoleIterator to traverse children
ZF-Commons · Jan 13, 2014 · 3934e41 · 3934e41
1 parent aafda31
commit 3934e41
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/ZfcRbacTest/Role/ObjectRepositoryRoleProviderTest.php b/tests/ZfcRbacTest/Role/ObjectRepositoryRoleProviderTest.php
@@ -19,6 +19,7 @@
 namespace ZfcRbacTest\Role;
 
 use Doctrine\ORM\Tools\SchemaTool;
+use Rbac\Traversal\RecursiveRoleIterator;
 use Zend\ServiceManager\ServiceManager;
 use ZfcRbac\Role\ObjectRepositoryRoleProvider;
 use ZfcRbacTest\Asset\FlatRole;
@@ -95,7 +96,11 @@ public function testObjectRepositoryProviderForHierarchicalRole()
         $this->assertInstanceOf('Rbac\Role\HierarchicalRoleInterface', $roles[0]);
         $this->assertEquals('admin', $roles[0]->getName());
 
-        $iteratorIterator = new \RecursiveIteratorIterator($roles[0], \RecursiveIteratorIterator::SELF_FIRST);
+        $iteratorIterator = new \RecursiveIteratorIterator(
+            new RecursiveRoleIterator($roles[0]->getChildren()),
+            \RecursiveIteratorIterator::SELF_FIRST
+        );
+
         $childRolesString = '';
 
         foreach ($iteratorIterator as $childRole) {