<?php

/*
 * This file is part of the ONGR package.
 *
 * (c) NFQ Technologies UAB <info@nfq.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace ONGR\ElasticsearchDSL\Tests\Unit\DSL\Query\Span;

use ONGR\ElasticsearchDSL\Query\Span\SpanNearQuery;
use ONGR\ElasticsearchDSL\Query\Span\SpanQueryInterface;

/**
 * Unit test for SpanNearQuery.
 */
class SpanNearQueryTest extends \PHPUnit_Framework_TestCase
{
    /**
     * @var SpanQueryInterface|\PHPUnit_Framework_MockObject_MockObject
     */
    protected $mock;

    /**
     * Create mock object.
     */
    protected function setUp()
    {
        $this->mock = $this->getMockBuilder('ONGR\ElasticsearchDSL\Query\Span\SpanQueryInterface')->getMock();
        $this->mock->expects($this->atMost(1))
            ->method('getType')
            ->will($this->returnValue('span_or'));
        $this->mock->expects($this->atMost(1))
            ->method('toArray')
            ->will($this->returnValue(['key' => 'value']));
    }

    /**
     * Reset mock object.
     */
    public function tearDown()
    {
        unset($this->mock);
    }

    /**
     * Tests toArray method.
     */
    public function testSpanMultiTermQueryToArray()
    {
        $query = new SpanNearQuery(['name']);
        $query->setSlop(5);
        $query->addQuery($this->mock);
        $result = [
            'clauses' => [
                0 => [
                    'span_or' => [
                        'key' => 'value',
                    ],
                ],
            ],
            'slop' => 5,
            0 => 'name',
        ];
        $this->assertEquals($result, $query->toArray());
    }

    /**
     * Tests get Type method.
     */
    public function testSpanNearQueryGetType()
    {
        $query = new SpanNearQuery(['name']);
        $result = $query->getType();
        $this->assertEquals('span_near', $result);
    }
}