1 files changed, 34 insertions, 0 deletions
diff --git a/src/test/scala/xyz/driver/restquery/db/SearchFilterExprSuite.scala b/src/test/scala/xyz/driver/restquery/db/SearchFilterExprSuite.scala
new file mode 100644
index 0000000..35c8d30
--- /dev/null
+++ b/src/test/scala/xyz/driver/restquery/db/SearchFilterExprSuite.scala
@@ -0,0 +1,34 @@
+package xyz.driver.pdsuicommon.db
+
+import org.scalatest.{FreeSpecLike, MustMatchers}
+
+class SearchFilterExprSuite extends FreeSpecLike with MustMatchers {
+
+  "replace" - {
+    "all entities are changed" in {
+      val ast = SearchFilterExpr.Union(
+        Seq(
+          SearchFilterExpr.Intersection(
+            Seq(
+              SearchFilterExpr.Atom.Binary("foo", SearchFilterBinaryOperation.Gt, "10"),
+              SearchFilterExpr.Atom.Binary("foo", SearchFilterBinaryOperation.Lt, "20")
+            )),
+          SearchFilterExpr.Atom.NAry("bar", SearchFilterNAryOperation.In, Seq("x", "y", "z")),
+          SearchFilterExpr.Atom.Binary("foo", SearchFilterBinaryOperation.Eq, "40")
+        ))
+
+      val newAst = ast.replace {
+        case x: SearchFilterExpr.Atom.Binary if x.dimension.name == "foo" =>
+          x.copy(dimension = x.dimension.copy(name = "bar"))
+      }
+
+      val result = newAst.find {
+        case x: SearchFilterExpr.Atom.Binary => x.dimension.name == "foo"
+        case _                               => false
+      }
+
+      result mustBe empty
+    }
+  }
+
+}