MNT Fix test

silverstripe · Aug 11, 2022 · 52c68fb · 52c68fb
1 parent 2cd2dd1
commit 52c68fb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/php/ORM/ListDecoratorTest.php b/tests/php/ORM/ListDecoratorTest.php
@@ -2,6 +2,7 @@
 
 namespace SilverStripe\ORM\Tests;
 
+use ArrayIterator;
 use LogicException;
 use PHPUnit\Framework\MockObject\MockObject;
 use SilverStripe\Dev\SapphireTest;
@@ -34,8 +35,9 @@ protected function setUp(): void
 
     public function testGetIterator()
     {
-        $this->list->expects($this->once())->method('getIterator')->willReturn('mock');
-        $this->assertSame('mock', $this->decorator->getIterator());
+        $iterator = new ArrayIterator();
+        $this->list->expects($this->once())->method('getIterator')->willReturn($iterator);
+        $this->assertSame($iterator, $this->decorator->getIterator());
     }
 
     public function testCanSortBy()