<?php

declare(strict_types=1);

namespace Tests\Functions;

use Arendsen\FluxQueryBuilder\Expression\KeyValue;
use Arendsen\FluxQueryBuilder\Expression\KeyFilter;
use Arendsen\FluxQueryBuilder\Functions\Filter;
use PHPUnit\Framework\TestCase;

final class FilterFunctionTest extends TestCase
{
    public function testSimpleFilter()
    {
        $expression = new Filter(KeyValue::setEqualTo('_measurement', 'test_measurement')
            ->andEqualTo('_field', 'user')
            ->orEqualTo('_field', 'field2')
            ->andEqualTo('user', 'my_username'));

        $query = '|> filter(fn: (r) => r._measurement == "test_measurement" and r._field == "user" or ' .
            'r._field == "field2" and r.user == "my_username") ';

        $this->assertEquals($expression->__toString(), $query);
    }

    public function testFieldFilter()
    {
        $expression = new Filter(['user', 'field2', 'field3']);

        $query = '|> filter(fn: (r) => r._field == "user" or r._field == "field2" or r._field == "field3") ';

        $this->assertEquals($expression->__toString(), $query);
    }

    public function testKeyFilter()
    {
        $expression = new Filter(KeyFilter::setEqualTo('_measurement', 'test_measurement')
            ->andEqualTo('_field', 'user')
            ->orEqualTo('_field', 'field2')
            ->andEqualTo('user', 'my_username'));

        $query = '|> filter(fn: (r) => r._measurement == "test_measurement" and r._field == "user" or ' .
            'r._field == "field2" and r.user == "my_username") ';

        $this->assertEquals($expression->__toString(), $query);
    }
}