let () = Mirage_crypto_rng_unix.use_default ()
let get_ok = function Error e -> failwith e | Ok v -> v

let () =
  let priv = Mirage_crypto_pk.Rsa.generate ~bits:2048 () in
  let pub = Mirage_crypto_pk.Rsa.pub_of_priv priv in
  let () =
    let open Crypto.RsaPrivateKey in
    let priv' = priv |> to_octets |> of_octets |> get_ok in
    assert (to_octets priv = to_octets priv')
  in
  let () =
    let open Crypto.RsaPublicKey in
    let pub' = pub |> to_octets |> of_octets |> get_ok in
    assert (to_octets pub = to_octets pub')
  in
  ()

let () =
  let open Api in
  let check jsont s =
    let encode v = encode jsont v |> get_ok in
    let decode v = decode jsont v |> get_ok in
    let ts = decode s in
    let s' = encode ts in
    let ts' = decode s' in
    let s'' = encode ts' in
    assert (String.equal s' s'')
  in
  let check_bad jsont s =
    let decode = decode jsont in
    assert (Result.is_error (decode s))
  in
  check Timestamp.jsont {|{"t_s": 123456780}|};
  check Timestamp.jsont {|{"t_s": "never"}|};
  check_bad Timestamp.jsont {|{"t_s": "123456780"}|};
  check_bad Timestamp.jsont {|{"t_s": "agagou"}|};

  (* CS not implemented *)
  check_bad DenominationKey.jsont
    {|{"cipher": "CS", "age_mask": 18, "cs_pub": "ouhagag"}|};

  (* TODO test with a valid rsa_pub value *)
  (* TODO handle "rsa pub_of_octets failure" correctly *)
  (* check_bad DenominationKey.jsont
    {|{"cipher": "RSA", "age_mask": 18, "rsa_pub": "agagouh"}|}; *)
  ()

let () =
  let open Amount in
  let v =
    make ~sign:(Some Sign_plus) ~currency:"EUR" ~value:(Int64.of_int 25)
      ~fraction:(Int32.of_int 678)
    |> Result.get_ok
  in
  let s = to_string v in
  let v' = of_string s |> Result.get_ok in
  assert (v = v');
  let r = of_string {|~EUR:4.9|} in
  assert (Result.is_error r);
  let r = of_string {|+EUR:4.99999999999999999999999|} in
  assert (Result.is_error r);
  ()

let () =
  let open Headers_lib.Etag in
  let check input = assert (Result.is_ok (parse input)) in
  let check_bad input = assert (Result.is_error (parse input)) in

  check "*";
  check "\"foo\"";
  check "W/\"foo\"";
  check "\"foo\", \"bar\"";
  check " W/\"x\" , W/\"y\" , \"z\" ";
  check "  \"one\" ,  \"two\"  ,  \"three\"  ";
  check "W/\"a\"";
  check "  W/\"a\" , \"b\"";

  check_bad "";
  check_bad "foo";
  check_bad "W/foo";
  check_bad "W/\"unterminated";
  check_bad "\"foo\", W/";
  check_bad "* , \"bar\"";
  check_bad "\"a\" \"b\"";
  check_bad "W/\"a\" W/\"b\"";
  check_bad "\"fo\x7Fo\"";

  (* TODO trailing comma are valid actually, I think *)
  check_bad "\"foo\" ,";
  check_bad ", \"foo\"";
  ()

(*

let () =
  let open Binary_formats.WithdrawConfirmationPS in
  let str64 = String.init 64 (fun i -> Char.unsafe_chr (i + 1)) in
  let dummy_t = { h_planchets= { hash= str64 }; noreveal_index= 0_l } in
  let size = Bin.size_of_value bin dummy_t |> Option.get in
  assert (size = 76);

  (* for cmp test with signatures.c output *)
  (*
  let raw_str = Bin.to_string bin dummy_t in
  Printf.printf "%s" raw_str;
  *)
  ()
  *)

let () =
  let v =
    "04000274VF2F65WWABKGZYHREQWAFTVPMWEK9EQYMCPP1E6CMC224XDDDBVM7NRMNDX3TETHTK5G7GPPPYGN08FEMSY6B7NGBE8AYMG5HWT2BXRSAHAWDWW5Y16YVGXJG0BCKW3X17EWEF8MNGFGJBKVWQV5WPT0R2QK82QXFHFVGP3HR1D8HF69JXBC22B62M4SFETRJNXRG2GXX03BETSNQW0X7BYTMFJX5JQH4B2KJMABB02RFSNNMND7CT7MSAX5SCZY5ZB9XS4ZDKA9AVTS11V790KXXBSFZSD3NNEHKB6RS99RJD3MDJE04G1Q83HR50N5NVBDW3HPJZKQS5SSG7R2R2F7K8P06HJ4YSQFF52GF3RZW22N4FRQ28KQ2JZMZ1GJS1KBNHNNMS1RW9DZQ3882Z5V0000000004002"
  in
  let decode s = B32.decode s |> get_ok in
  let v' = B32.encode (decode v) in
  let v'' = B32.encode (decode v') in
  Fmt.pr "v  : %s@." v;
  Fmt.pr "v' : %s@." v';
  Fmt.pr "v'': %s@." v'';
  Fmt.pr "len v  : %d@." (String.length v);
  Fmt.pr "len v' : %d@." (String.length v');
  Fmt.pr "len v'': %d@." (String.length v'');
  assert (v = v');
  assert (v' = v'');
  ()