1 files changed, 115 insertions, 0 deletions
diff --git a/vendor/ruflin/elastica/test/lib/Elastica/Test/ScanAndScrollTest.php b/vendor/ruflin/elastica/test/lib/Elastica/Test/ScanAndScrollTest.php
new file mode 100644
index 00000000..a64670bb
--- /dev/null
+++ b/vendor/ruflin/elastica/test/lib/Elastica/Test/ScanAndScrollTest.php
@@ -0,0 +1,115 @@
+<?php
+
+namespace Elastica\Test;
+
+use Elastica\Document;
+use Elastica\Query;
+use Elastica\ResultSet;
+use Elastica\ScanAndScroll;
+use Elastica\Search;
+use Elastica\Test\Base as BaseTest;
+
+class ScanAndScrollTest extends BaseTest {
+
+    public function testConstruct() {
+        $scanAndScroll = $this->_prepareScanAndScroll();
+
+        $this->assertInstanceOf('Elastica\ScanAndScroll', $scanAndScroll);
+    }
+
+    public function testDefaultProperties() {
+        $scanAndScroll = $this->_prepareScanAndScroll();
+
+        $this->assertEquals('1m', $scanAndScroll->expiryTime);
+        $this->assertEquals(1000, $scanAndScroll->sizePerShard);
+    }
+
+    public function testQuerySizeOverride() {
+        $query = new Query();
+        $query->setSize(100);
+
+        $index = $this->_createIndex('test_1');
+        $index->refresh();  // Waits for the index to be fully created.
+        $type = $index->getType('scanAndScrollTest');
+
+        $search = new Search($this->_getClient());
+        $search->addIndex($index)->addType($type);
+        $search->setQuery($query);
+
+        $scanAndScroll = new ScanAndScroll($search);
+        $scanAndScroll->sizePerShard = 10;
+        $scanAndScroll->rewind();
+
+        $this->assertEquals(10, $query->getParam('size'));
+    }
+
+    public function testSizePerShard() {
+        $search = $this->_prepareSearch('test_2', 2, 20);
+
+        $scanAndScroll = new ScanAndScroll($search);
+        $scanAndScroll->sizePerShard = 5;
+        $scanAndScroll->rewind();
+
+        $this->assertEquals(10, $scanAndScroll->current()->count());
+    }
+
+    public function testScrollId() {
+        $search = $this->_prepareSearch('test_3', 1, 2);
+
+        $scanAndScroll = new ScanAndScroll($search);
+        $scanAndScroll->sizePerShard = 1;
+
+        $scanAndScroll->rewind();
+        $this->assertEquals(
+            $scanAndScroll->current()->getResponse()->getScrollId(),
+            $scanAndScroll->key()
+        );
+    }
+
+    public function testForeach() {
+        $search = $this->_prepareSearch('test_4', 2, 11);
+
+        $scanAndScroll = new ScanAndScroll($search);
+        $scanAndScroll->sizePerShard = 5;
+
+        // We expect 2 scrolls:
+        // 1. with 10 hits,
+        // 2. with 1 hit
+        // Note: there is a 3. scroll with 0 hits
+
+        $count = 0;
+        foreach($scanAndScroll as $resultSet) {
+            /** @var ResultSet $resultSet */
+            $count++;
+
+            switch(true) {
+                case $count == 1: $this->assertEquals(10, $resultSet->count()); break;
+                case $count == 2: $this->assertEquals(1, $resultSet->count()); break;
+            }
+        }
+
+        $this->assertEquals(2, $count);
+    }
+
+    private function _prepareScanAndScroll() {
+        return new ScanAndScroll(new Search($this->_getClient()));
+    }
+
+    private function _prepareSearch($indexName, $indexShards, $docs) {
+        $index = $this->_createIndex($indexName, true, $indexShards);
+        $type = $index->getType('scanAndScrollTest');
+
+        $insert = array();
+        for ($x = 1; $x <= $docs; $x++) {
+            $insert[] = new Document($x, array('id' => $x, 'key' => 'value'));
+        }
+
+        $type->addDocuments($insert);
+        $index->refresh();
+
+        $search = new Search($this->_getClient());
+        $search->addIndex($index)->addType($type);
+
+        return $search;
+    }
+} 
+\ No newline at end of file