Updated the assertion to an array comparison

cakephp · Apr 28, 2015 · ac8ccae · ac8ccae
1 parent 6fb3b4d
commit ac8ccae
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/tests/TestCase/ORM/AssociationCollectionTest.php b/tests/TestCase/ORM/AssociationCollectionTest.php
@@ -380,12 +380,8 @@ public function testAssociationsCanBeIterated()
         $belongsToMany = new BelongsToMany('');
         $this->associations->add('Cart', $belongsToMany);
 
-        foreach ($this->associations as $name => $association) {
-            if ($name === 'users') {
-                $this->assertInstanceOf('Cake\ORM\Association\BelongsTo', $association);
-            } else {
-                $this->assertInstanceOf('Cake\ORM\Association\BelongsToMany', $association);
-            }
-        }
+        $expected = ['users' => $belongsTo, 'cart' => $belongsToMany];
+        $result = iterator_to_array($this->associations, true);
+        $this->assertSame($expected, $result);
     }
 }