Serialize runtime objects: Multigroup, Branch & Relation

2 files changed, 50 insertions, 1 deletions

diff --git a/test/test_rnt.ml b/test/test_rnt.ml
index 7173837..a2a8e84 100644
--- a/test/test_rnt.ml
+++ b/test/test_rnt.ml
@@ -1,2 +1,6 @@
-let all_suites = Unit.Backend.Storage.suites () @ Unit.Backend.Lifecycle.suites ()
+let all_suites =
+  Unit.Concepts.Serialization.suites ()
+  @ Unit.Backend.Storage.suites ()
+  @ Unit.Backend.Lifecycle.suites ()
+
 let () = Alcotest.run "RNT" all_suites
diff --git a/test/unit/concepts/Serialization.ml b/test/unit/concepts/Serialization.ml
new file mode 100644
index 0000000..9df8d26
--- /dev/null
+++ b/test/unit/concepts/Serialization.ml
@@ -0,0 +1,45 @@
+module Branch = Rnt.Concepts.Branch
+module Hash = Rnt.Concepts.Hash
+module Multigroup = Rnt.Concepts.Multigroup
+module Relation = Rnt.Concepts.Relation
+
+let fail_condition c = Alcotest.fail (Rnt.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_branch_round_trips () =
+  let expected = Branch.make ~name:"master" ~target:(hash "master-target") in
+  match Branch.(expected |> to_bytes |> of_bytes) with
+  | Ok actual ->
+      Alcotest.(check string) "name" (Branch.name expected) (Branch.name actual);
+      check_hash "target" (Branch.target expected) (Branch.target actual)
+  | Error c -> fail_condition c
+
+let test_multigroup_round_trips () =
+  let expected = Multigroup.make ~name:"public" ~relations:(hash "public-relations") in
+  match Multigroup.(expected |> to_bytes |> of_bytes) with
+  | Ok actual ->
+      Alcotest.(check string) "name" (Multigroup.name expected) (Multigroup.name actual);
+      check_hash "relations" (Multigroup.relations expected) (Multigroup.relations actual)
+  | Error c -> fail_condition c
+
+let test_relation_round_trips () =
+  let schema = Relation.schema_of_list ["tag", "string"; "name", "string"] in
+  let expected = Relation.make ~name:"users" ~tuples:(hash "users-tuples") ~schema in
+  match Relation.(expected |> to_bytes |> of_bytes) with
+  | Ok actual ->
+      Alcotest.(check string) "name" (Relation.name expected) (Relation.name actual);
+      check_hash "tuples" (Relation.tuples expected) (Relation.tuples actual);
+      Alcotest.(check (list (pair string string)))
+        "schema"
+        (Relation.schema expected |> Relation.schema_to_list)
+        (Relation.schema actual |> Relation.schema_to_list)
+  | Error c -> fail_condition c
+
+let suites () =
+  [ ( "concepts/serialization",
+      [ Alcotest.test_case "branch round-trips" `Quick test_branch_round_trips;
+        Alcotest.test_case "multigroup round-trips" `Quick test_multigroup_round_trips;
+        Alcotest.test_case "relation round-trips" `Quick test_relation_round_trips ] ) ]