Format

mirage · Apr 5, 2024 · ce05bc2 · ce05bc2
1 parent e09f822
commit ce05bc2
Showing 1 changed file with 18 additions and 16 deletions.
diff --git a/test/test.ml b/test/test.ml
@@ -3,7 +3,9 @@ type _ s = Bytes : Bytes.t s | String : String.t s | Bigstring : bigstring s
 and bigstring =
   (char, Bigarray.int8_unsigned_elt, Bigarray.c_layout) Bigarray.Array1.t
 
-let title : type a k. [ `HMAC | `HMAC_feed | `Digest ] -> k Digestif.hash -> a s -> string =
+let title :
+    type a k.
+    [ `HMAC | `HMAC_feed | `Digest ] -> k Digestif.hash -> a s -> string =
  fun computation hash input ->
   let pp_computation ppf = function
     | `HMAC -> Fmt.string ppf "hmac"
@@ -54,28 +56,28 @@ let test_hmac :
       Alcotest.(check test_hash) title expect result
 
 let test_hmac_feed :
-  type k a. a s -> k Digestif.hash -> string -> a -> k -> unit =
-  fun kind hash key input expect ->
+    type k a. a s -> k Digestif.hash -> string -> a -> k -> unit =
+ fun kind hash key input expect ->
   let title = title `HMAC_feed hash kind in
   let module H = (val Digestif.module_of hash) in
   let test_hash = Alcotest.testable H.pp H.equal in
   let hmac_ctx = H.hmac_init ~key in
-  let total_len = match kind with
+  let total_len =
+    match kind with
     | Bytes -> Bytes.length input
     | String -> String.length input
-    | Bigstring -> Bigarray.Array1.dim input
-  in
+    | Bigstring -> Bigarray.Array1.dim input in
   let rec loop hmac_ctx off =
-    if off = total_len then hmac_ctx else
+    if off = total_len
+    then hmac_ctx
+    else
       let len = min (total_len - off) 16 in
       let hmac_ctx =
         match kind with
         | Bytes -> H.hmac_feed_bytes hmac_ctx ~off ~len input
         | String -> H.hmac_feed_string hmac_ctx ~off ~len input
-        | Bigstring -> H.hmac_feed_bigstring hmac_ctx ~off ~len input
-      in
-      loop hmac_ctx (off + len)
-  in
+        | Bigstring -> H.hmac_feed_bigstring hmac_ctx ~off ~len input in
+      loop hmac_ctx (off + len) in
   Alcotest.check test_hash title expect (H.hmac_get (loop hmac_ctx 0))
 
 let test_digest : type k a. a s -> k Digestif.hash -> a -> k Digestif.t -> unit
@@ -144,7 +146,8 @@ let makes ~name kind hash keys inputs expects =
 
 let makes' ~name kind hash keys inputs expects =
   List.map
-    (fun (key, input, expect) -> make_hmac_feed ~name kind hash key input expect)
+    (fun (key, input, expect) ->
+      make_hmac_feed ~name kind hash key input expect)
     (combine keys inputs expects)
 
 let to_bigstring s =
@@ -224,10 +227,9 @@ let results_sha256, results_sha256' =
       "aa36cd61caddefe26b07ba1d3d07ea978ed575c9d1f921837dff9f73e019713e";
       "a7c8b53d68678a8e6e4d403c6b97cf0f82c4ef7b835c41039c0a73aa4d627d05";
       "b2a83b628f7e0da71c3879b81075775072d0d35935c62cc6c5a79b337ccccca1";
-    ]
-  in
-  List.map (Digestif.of_hex Digestif.sha256) raw_results_sha256,
-  List.map Digestif.SHA256.of_hex raw_results_sha256
+    ] in
+  ( List.map (Digestif.of_hex Digestif.sha256) raw_results_sha256,
+    List.map Digestif.SHA256.of_hex raw_results_sha256 )
 
 let results_sha384 =
   [