ONGR\ElasticsearchBundle\Test\AbstractElasticsearchTestCase

Complexity

Total Complexity

17

Size/Duplication

Total Lines	102
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	4

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	setUp()	0	3	1
A	getDataArray()	0	4	1
A	populateElastic()	0	7	2
A	tearDown()	0	12	3
A	getContainer()	0	10	3
B	getIndex()	0	27	7

<?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\ElasticsearchBundle\Test;

use Elasticsearch\Common\Exceptions\BadRequest400Exception;
use ONGR\ElasticsearchBundle\Service\IndexService;
use Symfony\Bundle\FrameworkBundle\Test\WebTestCase;
use Symfony\Component\DependencyInjection\ContainerInterface;

/**
 * Base test which creates unique connection to test with.
 */
abstract class AbstractElasticsearchTestCase extends WebTestCase
{
    protected static $cachedContainer;

    /**
     * @var IndexService[]
     */
    private $indexes = [];

    //You may use setUp() for your personal needs.
    protected function setUp()
    {
    }

    /**
     * Can be overwritten in child class to populate elasticsearch index with the data.
     *
     * Example:
     *      "/This/Should/Be/Index/Document/Namespace" =>
     *      [
     *          '_doc' => [
     *              [
     *                  '_id' => 1,
     *                  'title' => 'foo',
     *              ],
     *              [
     *                  '_id' => 2,
     *                  'title' => 'bar',
     *              ]
     *          ]
     *      ]
     *
     * @return array
     */
    protected function getDataArray()
    {
        return [];
    }

    private function populateElastic(IndexService $indexService, array $documents = [])
    {
        foreach ($documents as $document) {
            $indexService->bulk('index', $document);
        }
        $indexService->commit();
    }

    /**
     * {@inheritdoc}
     */
    protected function tearDown()
    {
        parent::tearDown();

        foreach ($this->indexes as $name => $index) {
            try {
                $index->dropIndex();
            } catch (\Exception $e) {
                // Do nothing.
            }
        }
    }

    protected function getContainer($reinitialize = false, $kernelOptions = []): ContainerInterface
    {
        if (!self::$cachedContainer && !$reinitialize) {
            static::bootKernel($kernelOptions);

            self::$cachedContainer = static::createClient(['environment' => 'test'])->getContainer();
        }

        return self::$cachedContainer;
    }

    protected function getIndex($namespace, $createIndex = true): IndexService
    {
        try {
            if (!array_key_exists($namespace, $this->indexes)) {
                $this->indexes[$namespace] = $this->getContainer()->get($namespace);
            }

            if (!$this->indexes[$namespace]->indexExists() && $createIndex) {
                $this->indexes[$namespace]->dropAndCreateIndex();

                // Populates elasticsearch index with the data
                $data = $this->getDataArray();
                if (!empty($data[$namespace])) {
                    $this->populateElastic($this->indexes[$namespace], $data[$namespace]);
                }
                $this->indexes[$namespace]->refresh();
            }

            return $this->indexes[$namespace];
        } catch (BadRequest400Exception $e) {
            throw new \LogicException($e->getMessage());
        } catch (\Exception $e) {
            throw new \LogicException(
                sprintf("There is no Elastic index defined in the '%s' namespace", $namespace)
            );
        }
    }
}