summaryrefslogtreecommitdiff
path: root/vendor/ruflin/elastica/test/lib/Elastica/Test/Filter/IndicesTest.php
blob: bc78aa6bb7565df2f4d92934685a714838397409 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
<?php
namespace Elastica\Test\Filter;

use Elastica\Document;
use Elastica\Filter\BoolNot;
use Elastica\Filter\Indices;
use Elastica\Filter\Term;
use Elastica\Index;
use Elastica\Query;
use Elastica\Test\Base as BaseTest;

class IndicesTest extends BaseTest
{
    /**
     * @group unit
     */
    public function testToArray()
    {
        $expected = array(
            'indices' => array(
                'indices' => array('index1', 'index2'),
                'filter' => array(
                    'term' => array('tag' => 'wow'),
                ),
                'no_match_filter' => array(
                    'term' => array('tag' => 'such filter'),
                ),
            ),
        );
        $filter = new Indices(new Term(array('tag' => 'wow')), array('index1', 'index2'));
        $filter->setNoMatchFilter(new Term(array('tag' => 'such filter')));
        $this->assertEquals($expected, $filter->toArray());
    }

    /**
     * @group functional
     */
    public function testIndicesFilter()
    {
        $docs = array(
            new Document(1, array('color' => 'blue')),
            new Document(2, array('color' => 'green')),
            new Document(3, array('color' => 'blue')),
            new Document(4, array('color' => 'yellow')),
        );

        $index1 = $this->_createIndex();
        $index1->addAlias('indices_filter');
        $index1->getType('test')->addDocuments($docs);
        $index1->refresh();

        $index2 = $this->_createIndex();
        $index2->addAlias('indices_filter');
        $index2->getType('test')->addDocuments($docs);
        $index2->refresh();

        $filter = new Indices(new BoolNot(new Term(array('color' => 'blue'))), array($index1->getName()));
        $filter->setNoMatchFilter(new BoolNot(new Term(array('color' => 'yellow'))));
        $query = new Query();
        $query->setPostFilter($filter);

        // search over the alias
        $index = $this->_getClient()->getIndex('indices_filter');
        $results = $index->search($query);

        // ensure that the proper docs have been filtered out for each index
        $this->assertEquals(5, $results->count());
        foreach ($results->getResults() as $result) {
            $data = $result->getData();
            $color = $data['color'];
            if ($result->getIndex() === $index1->getName()) {
                $this->assertNotEquals('blue', $color);
            } else {
                $this->assertNotEquals('yellow', $color);
            }
        }
    }

    /**
     * @group unit
     */
    public function testSetIndices()
    {
        $client = $this->_getClient();
        $index1 = $client->getIndex('index1');
        $index2 = $client->getIndex('index2');

        $indices = array('one', 'two');
        $filter = new Indices(new Term(array('color' => 'blue')), $indices);
        $this->assertEquals($indices, $filter->getParam('indices'));

        $indices[] = 'three';
        $filter->setIndices($indices);
        $this->assertEquals($indices, $filter->getParam('indices'));

        $filter->setIndices(array($index1, $index2));
        $expected = array($index1->getName(), $index2->getName());
        $this->assertEquals($expected, $filter->getParam('indices'));

        $returnValue = $filter->setIndices($indices);
        $this->assertInstanceOf('Elastica\Filter\Indices', $returnValue);
    }

    /**
     * @group unit
     */
    public function testAddIndex()
    {
        $client = $this->_getClient();
        $index = $client->getIndex('someindex');

        $filter = new Indices(new Term(array('color' => 'blue')), array());

        $filter->addIndex($index);
        $expected = array($index->getName());
        $this->assertEquals($expected, $filter->getParam('indices'));

        $filter->addIndex('foo');
        $expected = array($index->getName(), 'foo');
        $this->assertEquals($expected, $filter->getParam('indices'));

        $returnValue = $filter->addIndex('bar');
        $this->assertInstanceOf('Elastica\Filter\Indices', $returnValue);
    }
}