2 files changed, 70 insertions, 0 deletions
diff --git a/test/test_rnt.ml b/test/test_rnt.ml
index a2a8e84..f825c51 100644
--- a/test/test_rnt.ml
+++ b/test/test_rnt.ml
@@ -1,5 +1,6 @@
 let all_suites =
   Unit.Concepts.Serialization.suites ()
+  @ Unit.Kernel.Initialization.suites ()
   @ Unit.Backend.Storage.suites ()
   @ Unit.Backend.Lifecycle.suites ()
 
diff --git a/test/unit/kernel/Initialization.ml b/test/unit/kernel/Initialization.ml
new file mode 100644
index 0000000..c0e9c17
--- /dev/null
+++ b/test/unit/kernel/Initialization.ml
@@ -0,0 +1,69 @@
+open Rnt
+
+module Hash = Concepts.Hash
+module Initialization = Rnt.Kernel.Initialization
+
+module Memory_storage = struct
+  type connection = (Hash.hash, Concepts.Representation.blob) Hashtbl.t
+  type transaction = connection
+
+  let connect _ = Ok (Hashtbl.create 16)
+  let start conn = Ok conn
+  let commit _ = Ok ()
+  let abort _ = Ok ()
+  let get tx address = Ok (Hashtbl.find_opt tx address)
+  let put tx address blob = Hashtbl.replace tx address blob; Ok ()
+end
+
+module Store_initialization = Initialization.Make (Memory_storage)
+
+let fail_condition c = Alcotest.fail (Concepts.Condition.to_string_hum c)
+let hash text = Hash.hash_of_bytes (Bytes.of_string text)
+
+let check_hash label expected actual =
+  Alcotest.(check bool) label true (Hash.hash_equals expected actual)
+
+let test_initialization_round_trips () =
+  let expected =
+    Initialization.make ~default_branch:"master" ~branches:(hash "branch-index")
+  in
+  match Initialization.(expected |> to_bytes |> of_bytes) with
+  | Ok actual ->
+      Alcotest.(check string)
+        "default branch"
+        (Initialization.default_branch expected)
+        (Initialization.default_branch actual);
+      check_hash "branches" (Initialization.branches expected) (Initialization.branches actual)
+  | Error c -> fail_condition c
+
+let test_registers_under_fixed_label () =
+  let tx = Hashtbl.create 16 in
+  let expected =
+    Initialization.make ~default_branch:"master" ~branches:(hash "branch-index")
+  in
+  let before, registered, after =
+    begin
+      let open Utilities.Result in
+      let* before = Store_initialization.load tx in
+      let* registered = Store_initialization.register tx expected in
+      let* after = Store_initialization.load tx in
+      Ok (before, registered, after)
+    end
+    |> Helpers.condition_as_failure
+  in
+  Alcotest.(check (option bool))
+    "missing before registration" None (Option.map (Fun.const true) before);
+  check_hash "registered address" (Initialization.address expected) registered;
+  Alcotest.(check bool)
+    "fixed label exists" true
+    (Hashtbl.mem tx Initialization.fixed_label_address);
+  match after with
+  | Some actual -> Alcotest.(check bool) "loaded object" true (Initialization.equal expected actual)
+  | None -> Alcotest.fail "expected initialization object after registration"
+
+let suites () =
+  [ ( "kernel/initialization",
+      [ Alcotest.test_case "initialization round-trips" `Quick
+          test_initialization_round_trips;
+        Alcotest.test_case "registers under fixed label" `Quick
+          test_registers_under_fixed_label ] ) ]