<?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;
    }
}