1 files changed, 42 insertions, 0 deletions
diff --git a/vendor/guzzle/guzzle/tests/Guzzle/Tests/Service/Mock/Model/MockCommandIterator.php b/vendor/guzzle/guzzle/tests/Guzzle/Tests/Service/Mock/Model/MockCommandIterator.php
new file mode 100644
index 0000000..8faf412
--- /dev/null
+++ b/vendor/guzzle/guzzle/tests/Guzzle/Tests/Service/Mock/Model/MockCommandIterator.php
@@ -0,0 +1,42 @@
+<?php
+
+namespace Guzzle\Tests\Service\Mock\Model;
+
+use Guzzle\Service\Resource\ResourceIterator;
+
+class MockCommandIterator extends ResourceIterator
+{
+    public $calledNext = 0;
+
+    protected function sendRequest()
+    {
+        if ($this->nextToken) {
+            $this->command->set('next_token', $this->nextToken);
+        }
+
+        $this->command->set('page_size', (int) $this->calculatePageSize());
+        $this->command->execute();
+
+        $data = json_decode($this->command->getResponse()->getBody(true), true);
+
+        $this->nextToken = $data['next_token'];
+
+        return $data['resources'];
+    }
+
+    public function next()
+    {
+        $this->calledNext++;
+        parent::next();
+    }
+
+    public function getResources()
+    {
+        return $this->resources;
+    }
+
+    public function getIteratedCount()
+    {
+        return $this->iteratedCount;
+    }
+}