Add from, range, filter functions to QueryBuilder

Signed-off-by: davidarendsen <davidarendsen@hey.com>
This commit is contained in:
davidarendsen 2022-08-08 14:42:10 +00:00
commit 34845d68a1
5 changed files with 164 additions and 23 deletions

View file

@ -9,12 +9,11 @@ final class FilterFunctionTest extends TestCase {
public function testSimpleFilter()
{
$expression = new Filter([
KeyValue::set('_measurement', 'test_measurement')
->and('_field', 'user')
->or('_field', 'field2')
->and('user', 'my_username'),
]);
$expression = new Filter(KeyValue::set('_measurement', 'test_measurement')
->and('_field', 'user')
->or('_field', 'field2')
->and('user', 'my_username')
);
$query = '|> filter(fn: (r) => r._measurement == "test_measurement" and r._field == "user" or ' .
'r._field == "field2" and r.user == "my_username") ';

View file

@ -7,31 +7,65 @@ use Arendsen\FluxQueryBuilder\QueryBuilder;
final class QueryBuilderTest extends TestCase {
/**
* @dataProvider somethingProvider
* @dataProvider simpleQueryProvider
*/
public function testSomething($bucket, $measurement, $range, $expectedQuery)
public function testSimpleQuery($bucket, $measurement, $range, $expectedQuery)
{
$queryBuilder = new QueryBuilder();
$queryBuilder->from($bucket)
->fromMeasurement($measurement)
->addRangeStart($range);
$this->assertEquals($queryBuilder->build(), $expectedQuery);
$this->assertEquals($expectedQuery, $queryBuilder->build());
}
public function somethingProvider(): array
public function simpleQueryProvider(): array
{
return [
'case 1' => [
[
'bucket' => 'example-bucket',
'host' => 'host',
'org' => 'example-org',
'token' => 'token'
'bucket' => 'example_bucket',
],
'test_measurement',
'-360h',
'from(bucket: "test_bucket", host: "host", org: "example-org", token: "token") |> '
'from(bucket: "example_bucket") |> range(start: "-360h") |> filter(fn: (r) => r._measurement == "test_measurement") '
],
];
}
/**
* @dataProvider throwsExceptionWithoutRequiredDataProvider
*/
public function testThrowsExceptionWithoutRequiredData($from, $measurement, $range)
{
$this->expectException(Exception::class);
$queryBuilder = new QueryBuilder();
if($from) {
$queryBuilder->from($from);
}
if($measurement) {
$queryBuilder->fromMeasurement($measurement);
}
if($range) {
$queryBuilder->addRange($range);
}
$queryBuilder->build();
}
public function throwsExceptionWithoutRequiredDataProvider(): array
{
return [
'without from data' => [
null, 'test_measurement', ['start' => '-360h'],
],
'without measurement data' => [
['from' => 'test_bucket'], null, ['start' => '-360h'],
],
'without range data' => [
['from' => 'test_bucket'], 'test_measurement', null,
],
];
}