Add test for chunk's early return fix

baopham · Nov 16, 2018 · 7e316c8 · 7e316c8
1 parent af66f94
commit 7e316c8
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/DynamoDbNonCompositeModelTest.php b/tests/DynamoDbNonCompositeModelTest.php
@@ -687,6 +687,22 @@ public function testChunkScanCondition()
         $this->assertEquals(3, $total_results);
     }
 
+    public function testChunkEarlyReturn()
+    {
+        $this->seed(['name' => ['S' => 'Foo']]);
+        $this->seed(['name' => ['S' => 'Foo2']]);
+        $this->seed(['name' => ['S' => 'Foo3']]);
+
+        $iteration = 1;
+
+        $this->testModel->chunk(2, function () use (&$iteration) {
+            $iteration++;
+            return false;
+        });
+
+        $this->assertEquals(2, $iteration);
+    }
+
     public function testStaticMethods()
     {
         $item = $this->seed(['name' => ['S' => 'Foo'], 'description' => ['S' => 'Bar']]);