Rewrite the way to handle filters

1 files changed, 26 insertions, 1 deletions

diff --git a/tests/test_migration.ml b/tests/test_migration.ml
index 35480d1..e26f354 100644
--- a/tests/test_migration.ml
+++ b/tests/test_migration.ml
@@ -45,7 +45,7 @@ let int_container_testable = make_test (module ImportContainers.IntSet)
 let expression_testable =
   make_test
     (module struct
-      type t = ImportConf.Path.t ImportExpression.T.t [@@deriving show, eq]
+      type t = ImportDataTypes.Path.t ImportExpression.T.t [@@deriving show, eq]
     end)
 
 let dep_key_testable =
@@ -53,3 +53,28 @@ let dep_key_testable =
     (module struct
       type t = ImportAnalyser.Dependency.key [@@deriving show, eq]
     end)
+
+let chunk =
+  make_test
+    (module struct
+      type t = ImportAnalyser.Chunk.t
+
+      let pp formater t =
+        Format.fprintf formater "%s" (Buffer.contents t.ImportAnalyser.Chunk.b)
+
+      let equal t1 t2 =
+        let to_string t = Buffer.contents t.ImportAnalyser.Chunk.b in
+        String.equal (to_string t1) (to_string t2)
+    end)
+
+let syntax =
+  make_test
+    (module struct
+      type t = ImportConf.Syntax.t
+
+      let pp format t =
+        Format.fprintf format "%s"
+          (Otoml.Printer.to_string (ImportConf.Syntax.repr t))
+
+      let equal t1 t2 = t1 = t2
+    end)