summaryrefslogtreecommitdiff
path: root/vendor/ruflin/elastica/test/lib/Elastica/Test/Query/HighlightTest.php
blob: c6850019ec4eba25f4efbaeab55562f43cd7683d (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
<?php
namespace Elastica\Test\Query;

use Elastica\Document;
use Elastica\Query;
use Elastica\Query\QueryString;
use Elastica\Test\Base as BaseTest;

class HighlightTest extends BaseTest
{
    /**
     * @group functional
     */
    public function testHightlightSearch()
    {
        $index = $this->_createIndex();
        $type = $index->getType('helloworld');

        $phrase = 'My name is ruflin';

        $type->addDocuments(array(
            new Document(1, array('id' => 1, 'phrase' => $phrase, 'username' => 'hanswurst', 'test' => array('2', '3', '5'))),
            new Document(2, array('id' => 2, 'phrase' => $phrase, 'username' => 'peter', 'test' => array('2', '3', '5'))),
        ));

        $queryString = new QueryString('rufl*');
        $query = new Query($queryString);
        $query->setHighlight(array(
            'pre_tags' => array('<em class="highlight">'),
            'post_tags' => array('</em>'),
            'fields' => array(
                'phrase' => array(
                    'fragment_size' => 200,
                    'number_of_fragments' => 1,
                ),
            ),
        ));

        $index->refresh();

        $resultSet = $type->search($query);
        foreach ($resultSet as $result) {
            $highlight = $result->getHighlights();
            $this->assertEquals(array('phrase' => array(0 => 'My name is <em class="highlight">ruflin</em>')), $highlight);
        }
        $this->assertEquals(2, $resultSet->count());
    }
}