icingaweb2/test/php/library/Icinga/Protocol/Statusdat/Query/GroupTest.php

<?php
// {{{ICINGA_LICENSE_HEADER}}}
// {{{ICINGA_LICENSE_HEADER}}}

namespace Tests\Icinga\Protocol\Statusdat\Query;

use Icinga\Test\BaseTestCase;
use Icinga\Protocol\Statusdat\Query\Group;
use Icinga\Protocol\Statusdat\Query\IQueryPart;

class QueryExpressionMock implements IQueryPart
{
    public $rawExpression;
    public $value;
    public $filter = array();

    public function __construct($expression = null, &$value = array())
    {
        $this->value = array_shift($value);
        $this->rawExpression = $expression;
    }

    public function filter(array &$base, &$idx = null)
    {
        return array_intersect(array_values($idx), array_values($this->filter));
    }

    /**
     * Add additional information about the query this filter belongs to
     *
     * @param $query
     * @return mixed
     */
    public function setQuery($query)
    {
        // TODO: Implement setQuery() method.
    }
}

class GroupTest extends BaseTestCase
{
    public function testParsingSingleCondition()
    {
        $testQuery = new Group();
        $value = array(4);
        $testQuery->fromString(
            "numeric_val >= ?",
            $value,
            "Tests\Icinga\Protocol\Statusdat\Query\QueryExpressionMock"
        );
        $this->assertCount(1, $testQuery->getItems());
        $this->assertCount(0, $value);

        $expression = $testQuery->getItems();
        $expression = $expression[0];

        $this->assertEquals("numeric_val >= ?", $expression->rawExpression);
        $this->assertEquals(4, $expression->value);
    }

    public function testParsingSimpleAndCondition()
    {
        $testQuery = new Group();
        $value = array(4, 'hosta');
        $testQuery->fromString(
            "numeric_val >= ? AND host_name = ?",
            $value,
            "Tests\Icinga\Protocol\Statusdat\Query\QueryExpressionMock"
        );
        $this->assertCount(2, $testQuery->getItems());
        $this->assertCount(0, $value);
        $this->assertEquals("AND", $testQuery->getType());
        $items = $testQuery->getItems();

        $expression0 = $items[0];
        $this->assertEquals("numeric_val >= ?", $expression0->rawExpression);
        $this->assertEquals(4, $expression0->value);

        $expression1 = $items[1];
        $this->assertEquals("host_name = ?", $expression1->rawExpression);
        $this->assertEquals("hosta", $expression1->value);
    }

    public function testParsingSimpleORCondition()
    {
        $testQuery = new Group();
        $value = array(4, 'hosta');
        $testQuery->fromString(
            "numeric_val >= ? OR host_name = ?",
            $value,
            "Tests\Icinga\Protocol\Statusdat\Query\QueryExpressionMock"
        );
        $this->assertCount(2, $testQuery->getItems());
        $this->assertCount(0, $value);
        $this->assertEquals("OR", $testQuery->getType());
        $items = $testQuery->getItems();

        $expression0 = $items[0];
        $this->assertEquals("numeric_val >= ?", $expression0->rawExpression);
        $this->assertEquals(4, $expression0->value);

        $expression1 = $items[1];
        $this->assertEquals("host_name = ?", $expression1->rawExpression);
        $this->assertEquals("hosta", $expression1->value);
    }

    public function testParsingExplicitSubgroup()
    {
        $testQuery = new Group();
        $value = array(4, 'service1', 'hosta');
        $testQuery->fromString(
            "numeric_val >= ? AND (service_description = ? OR host_name = ?)",
            $value,
            "Tests\Icinga\Protocol\Statusdat\Query\QueryExpressionMock"
        );
        $this->assertCount(2, $testQuery->getItems());
        $this->assertCount(0, $value);
        $this->assertEquals("AND", $testQuery->getType());
        $items = $testQuery->getItems();

        $expression0 = $items[0];
        $this->assertEquals("numeric_val >= ?", $expression0->rawExpression);
        $this->assertEquals(4, $expression0->value);

        $subgroup = $items[1];
        $this->assertInstanceOf("Icinga\Protocol\Statusdat\Query\Group", $subgroup);
        $this->assertEquals("OR", $subgroup->getType());
        $orItems = $subgroup->getItems();

        $expression1 = $orItems[0];
        $this->assertEquals("service_description = ?", $expression1->rawExpression);
        $this->assertEquals("service1", $expression1->value);

        $expression2 = $orItems[1];
        $this->assertEquals("host_name = ?", $expression2->rawExpression);
        $this->assertEquals("hosta", $expression2->value);
    }

    public function testParsingImplicitSubgroup()
    {
        $testQuery = new Group();
        $value = array(4, 'service1', 'hosta');
        $testQuery->fromString(
            "numeric_val >= ? AND service_description = ? OR host_name = ?",
            $value,
            "Tests\Icinga\Protocol\Statusdat\Query\QueryExpressionMock"
        );
        $this->assertCount(2, $testQuery->getItems());
        $this->assertCount(0, $value);
        $this->assertEquals("AND", $testQuery->getType());
        $items = $testQuery->getItems();

        $expression0 = $items[0];
        $this->assertEquals("numeric_val >= ?", $expression0->rawExpression);
        $this->assertEquals(4, $expression0->value);

        $subgroup = $items[1];
        $this->assertInstanceOf("Icinga\Protocol\Statusdat\Query\Group", $subgroup);
        $this->assertEquals("OR", $subgroup->getType());
        $orItems = $subgroup->getItems();

        $expression1 = $orItems[0];
        $this->assertEquals("service_description = ?", $expression1->rawExpression);
        $this->assertEquals("service1", $expression1->value);

        $expression2 = $orItems[1];
        $this->assertEquals("host_name = ?", $expression2->rawExpression);
        $this->assertEquals("hosta", $expression2->value);
    }

    public function testAndFilter()
    {
        $testQuery = new Group();
        $testQuery->setType(Group::TYPE_AND);
        $exp1 = new QueryExpressionMock();
        $exp1->filter = array(1, 2, 3, 4, 5, 6, 8);
        $exp2 = new QueryExpressionMock();
        $exp2->filter = array(3, 4, 8);
        $base = array(0, 1, 2, 3, 4, 5, 6, 7, 8);

        $this->assertEquals(
            array(3, 4, 8),
            array_values($testQuery->addItem($exp1)->addItem($exp2)->filter($base))
        );
    }

    public function testOrFilter()
    {
        $testQuery = new Group();
        $testQuery->setType(Group::TYPE_OR);
        $exp1 = new QueryExpressionMock();
        $exp1->filter = array(1, 2, 3);
        $exp2 = new QueryExpressionMock();
        $exp2->filter = array(3, 4, 6, 8);
        $base = array(0, 1, 2, 3, 4, 5, 6, 7, 8);
        $this->assertEquals(
            array(1, 2, 3, 4, 6, 8),
            array_values($testQuery->addItem($exp1)->addItem($exp2)->filter($base))
        );
    }

    public function testCombinedFilter()
    {
        $testQuery_and = new Group();
        $testQuery_and->setType(Group::TYPE_AND);
        $testQuery_or = new Group();
        $testQuery_or->setType(Group::TYPE_OR);
        $base = array(0, 1, 2, 3, 4, 5, 6, 7, 8);

        $and_exp1 = new QueryExpressionMock();
        $and_exp1->filter = array(1, 2, 3, 4, 5, 6, 8);
        $and_exp2 = new QueryExpressionMock();
        $and_exp2->filter = array(3, 4, 8);

        $or_exp1 = new QueryExpressionMock();
        $or_exp1->filter = array(1, 2, 3);
        $or_exp2 = new QueryExpressionMock();
        $or_exp2->filter = array(3, 4, 6, 8);
        $this->assertEquals(array(3, 4, 8), array_values(
                $testQuery_and
                    ->addItem($and_exp1)
                    ->addItem($and_exp2)
                    ->addItem($testQuery_or->addItem($or_exp1)->addItem($or_exp2))
                    ->filter($base))
        );
    }
}