1 files changed, 164 insertions, 0 deletions
diff --git a/tests/lib/Files/Search/QueryOptimizer/MergeDistributiveOperationsTest.php b/tests/lib/Files/Search/QueryOptimizer/MergeDistributiveOperationsTest.php
new file mode 100644
index 00000000000..9aaa7030aac
--- /dev/null
+++ b/tests/lib/Files/Search/QueryOptimizer/MergeDistributiveOperationsTest.php
@@ -0,0 +1,164 @@
+<?php
+
+/**
+ * SPDX-FileCopyrightText: 2023 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+namespace Test\Files\Search\QueryOptimizer;
+
+use OC\Files\Search\QueryOptimizer\FlattenSingleArgumentBinaryOperation;
+use OC\Files\Search\QueryOptimizer\MergeDistributiveOperations;
+use OC\Files\Search\SearchBinaryOperator;
+use OC\Files\Search\SearchComparison;
+use OCP\Files\Search\ISearchBinaryOperator;
+use OCP\Files\Search\ISearchComparison;
+use Test\TestCase;
+
+class MergeDistributiveOperationsTest extends TestCase {
+	private $optimizer;
+	private $simplifier;
+
+	protected function setUp(): void {
+		parent::setUp();
+
+		$this->optimizer = new MergeDistributiveOperations();
+		$this->simplifier = new FlattenSingleArgumentBinaryOperation();
+	}
+
+	public function testBasicOrOfAnds(): void {
+		$operator = new SearchBinaryOperator(
+			ISearchBinaryOperator::OPERATOR_OR,
+			[
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'foo'),
+				]),
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'bar'),
+				]),
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'asd'),
+				])
+			]
+		);
+		$this->assertEquals('((storage eq 1 and path eq "foo") or (storage eq 1 and path eq "bar") or (storage eq 1 and path eq "asd"))', $operator->__toString());
+
+		$this->optimizer->processOperator($operator);
+		$this->simplifier->processOperator($operator);
+
+		$this->assertEquals('(storage eq 1 and (path eq "foo" or path eq "bar" or path eq "asd"))', $operator->__toString());
+	}
+
+	public function testDontTouchIfNotSame(): void {
+		$operator = new SearchBinaryOperator(
+			ISearchBinaryOperator::OPERATOR_OR,
+			[
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'foo'),
+				]),
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 2),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'bar'),
+				]),
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 3),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'asd'),
+				])
+			]
+		);
+		$this->assertEquals('((storage eq 1 and path eq "foo") or (storage eq 2 and path eq "bar") or (storage eq 3 and path eq "asd"))', $operator->__toString());
+
+		$this->optimizer->processOperator($operator);
+		$this->simplifier->processOperator($operator);
+
+		$this->assertEquals('((storage eq 1 and path eq "foo") or (storage eq 2 and path eq "bar") or (storage eq 3 and path eq "asd"))', $operator->__toString());
+	}
+
+	public function testMergePartial(): void {
+		$operator = new SearchBinaryOperator(
+			ISearchBinaryOperator::OPERATOR_OR,
+			[
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'foo'),
+				]),
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'bar'),
+				]),
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 2),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'asd'),
+				])
+			]
+		);
+		$this->assertEquals('((storage eq 1 and path eq "foo") or (storage eq 1 and path eq "bar") or (storage eq 2 and path eq "asd"))', $operator->__toString());
+
+		$this->optimizer->processOperator($operator);
+		$this->simplifier->processOperator($operator);
+
+		$this->assertEquals('((storage eq 1 and (path eq "foo" or path eq "bar")) or (storage eq 2 and path eq "asd"))', $operator->__toString());
+	}
+
+	public function testOptimizeInside(): void {
+		$operator = new SearchBinaryOperator(
+			ISearchBinaryOperator::OPERATOR_AND,
+			[
+				new SearchBinaryOperator(
+					ISearchBinaryOperator::OPERATOR_OR,
+					[
+						new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+							new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+							new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'foo'),
+						]),
+						new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+							new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+							new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'bar'),
+						]),
+						new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+							new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+							new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'asd'),
+						])
+					]
+				),
+				new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'mimetype', 'text')
+			]
+		);
+		$this->assertEquals('(((storage eq 1 and path eq "foo") or (storage eq 1 and path eq "bar") or (storage eq 1 and path eq "asd")) and mimetype eq "text")', $operator->__toString());
+
+		$this->optimizer->processOperator($operator);
+		$this->simplifier->processOperator($operator);
+
+		$this->assertEquals('((storage eq 1 and (path eq "foo" or path eq "bar" or path eq "asd")) and mimetype eq "text")', $operator->__toString());
+	}
+
+	public function testMoveInnerOperations(): void {
+		$operator = new SearchBinaryOperator(
+			ISearchBinaryOperator::OPERATOR_OR,
+			[
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'foo'),
+				]),
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'bar'),
+				]),
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'storage', 1),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'path', 'asd'),
+					new SearchComparison(ISearchComparison::COMPARE_GREATER_THAN, 'size', '100'),
+				])
+			]
+		);
+		$this->assertEquals('((storage eq 1 and path eq "foo") or (storage eq 1 and path eq "bar") or (storage eq 1 and path eq "asd" and size gt "100"))', $operator->__toString());
+
+		$this->optimizer->processOperator($operator);
+		$this->simplifier->processOperator($operator);
+
+		$this->assertEquals('(storage eq 1 and (path eq "foo" or path eq "bar" or (path eq "asd" and size gt "100")))', $operator->__toString());
+	}
+}