1 files changed, 198 insertions, 0 deletions
diff --git a/tests/sql_db.ml b/tests/sql_db.ml
new file mode 100644
index 0000000..c966f4e
--- /dev/null
+++ b/tests/sql_db.ml
@@ -0,0 +1,198 @@
+(** Test the behavior of the sqlite with a in-memory database *)
+
+open OUnit2
+open StdLabels
+
+let ( let* ) res cont =
+  match res with
+  | Ok value -> cont value
+  | Error e -> raise e
+
+(** Test a process with a simple configuration in-memory *)
+let run_test ~configuration ~input ~expected name =
+  name >:: fun _ ->
+  (* We expect a valid configuration *)
+  let conf =
+    ImportConf.t_of_toml (Otoml.Parser.from_string configuration)
+    |> Result.get_ok
+  in
+
+  let exec db =
+    let table = List.hd @@ ImportAnalyser.Dependency.get_process_order conf in
+    let* () = ImportSQL.Db.create_table db table in
+
+    (* Prepare the statement in order to import data *)
+    let* stmt = ImportSQL.Db.prepare_insert db table in
+
+    (* Inject some data into the table *)
+    let result, _ =
+      List.fold_left ~init:(Ok (), 0) input ~f:(fun (_, i) data ->
+          let result =
+            let* () = ImportSQL.Db.insert ~id:i db stmt data in
+            let* () = ImportSQL.Db.reset stmt in
+            Ok ()
+          in
+          (result, i + 1))
+    in
+    let* () = result in
+
+    let* () = ImportSQL.Db.finalize stmt in
+
+    let expected = ref expected in
+
+    (* Collect the data *)
+    let* () =
+      ImportSQL.Db.query db conf ~f:(fun rows ->
+          match !expected with
+          | [] -> ()
+          | hd :: tl ->
+              expected := tl;
+              let () =
+                Array.iter2 rows hd ~f:(fun (_, value) expected ->
+                    assert_equal ~printer:ImportCSV.DataType.to_string value
+                      expected)
+              in
+              ())
+    in
+
+    Ok ()
+  in
+
+  (* Use a magic keyword for in-memory database *)
+  ignore @@ ImportSQL.Db.with_db ":memory:" exec
+
+(** Simple test used to check the process *)
+let simple_extraction =
+  run_test "simple_extraction"
+    ~configuration:
+      {|version = 1
+
+[source]
+name = "source_name"
+file = "source_file"
+
+[sheet]
+columns = [
+    ":A ^ '_'", 
+    ":B",
+    ":E"]|}
+    ~input:
+      [
+        [
+          (0, ImportCSV.DataType.Integer 123);
+          (1, ImportCSV.DataType.Integer 2);
+          (4, ImportCSV.DataType.Integer 5);
+        ];
+      ]
+    ~expected:
+      [
+        [|
+          ImportCSV.DataType.Content "123_";
+          ImportCSV.DataType.Integer 2;
+          ImportCSV.DataType.Integer 5;
+        |];
+      ]
+
+(** Ensure the behavior of the sum function when a filter is given. It is
+    expected to accumulate the values over each line *)
+let sum_sort =
+  run_test "sum_sort"
+    ~configuration:
+      {|version = 1
+
+[source]
+name = "source_name"
+file = "source_file"
+
+[sheet]
+columns = [
+    ":A", 
+    "sum(:C, [:B], [:A])",
+]|}
+    ~input:
+      [
+        [
+          (0, ImportCSV.DataType.Integer 1);
+          (1, ImportCSV.DataType.Content "A");
+          (2, ImportCSV.DataType.Integer 100);
+        ];
+        [
+          (0, ImportCSV.DataType.Integer 2);
+          (1, ImportCSV.DataType.Content "A");
+          (2, ImportCSV.DataType.Integer 100);
+        ];
+        [
+          (0, ImportCSV.DataType.Integer 3);
+          (1, ImportCSV.DataType.Content "A");
+          (2, ImportCSV.DataType.Integer 100);
+        ];
+      ]
+    ~expected:
+      [
+        [| ImportCSV.DataType.Integer 1; ImportCSV.DataType.Integer 100 |];
+        [| ImportCSV.DataType.Integer 2; ImportCSV.DataType.Integer 200 |];
+        [| ImportCSV.DataType.Integer 3; ImportCSV.DataType.Integer 300 |];
+      ]
+
+let sum_total =
+  run_test "sum_total"
+    ~configuration:
+      {|version = 1
+
+[source]
+name = "source_name"
+file = "source_file"
+
+[sheet]
+columns = [
+    ":A", 
+    "sum(:C, [], [])",
+]|}
+    ~input:
+      [
+        [
+          (0, ImportCSV.DataType.Integer 1); (2, ImportCSV.DataType.Integer 100);
+        ];
+        [
+          (0, ImportCSV.DataType.Integer 2); (2, ImportCSV.DataType.Integer 100);
+        ];
+      ]
+    ~expected:
+      [
+        [| ImportCSV.DataType.Integer 1; ImportCSV.DataType.Integer 200 |];
+        [| ImportCSV.DataType.Integer 2; ImportCSV.DataType.Integer 200 |];
+      ]
+
+(** Ensure the behavior of the sum function when no filter is given. It is
+    expected to get the total sum for each line *)
+let sum_unfiltered =
+  run_test "sum_unfiltered"
+    ~configuration:
+      {|version = 1
+
+[source]
+name = "source_name"
+file = "source_file"
+
+[sheet]
+columns = [
+    ":A", 
+    "sum(:C, [], [:A])",
+]|}
+    ~input:
+      [
+        [
+          (0, ImportCSV.DataType.Integer 1); (2, ImportCSV.DataType.Integer 100);
+        ];
+        [
+          (0, ImportCSV.DataType.Integer 2); (2, ImportCSV.DataType.Integer 100);
+        ];
+      ]
+    ~expected:
+      [
+        [| ImportCSV.DataType.Integer 1; ImportCSV.DataType.Integer 200 |];
+        [| ImportCSV.DataType.Integer 2; ImportCSV.DataType.Integer 200 |];
+      ]
+
+let test_suit = [ simple_extraction; sum_sort; sum_total; sum_unfiltered ]
+let tests = "sql_db" >::: test_suit