tools/tests.ml

   1
   2 open Printf
   3
   4 let pr fmt = ksprintf print_endline fmt
   5
   6 let test ?n ?s f =
   7   try
   8     f ()
   9   with
  10     exn ->
  11       let msg = match s,n with
  12       | Some s, Some n -> sprintf " %d %S" n s
  13       | Some s, None -> sprintf " %S" s
  14       | None, Some n -> sprintf " %d" n
  15       | None, None -> ""
  16       in
  17       pr "Test%s failed: %s" msg (Printexc2.to_string exn)
  18
  19 let test1 ?n ?s f x = test ?n ?s (fun () -> f x)
  20
  21 let magnet s =
  22   let magnet = CommonTypes.parse_magnet_url s in
  23   pr "name: %S" magnet#name;
  24   begin match magnet#size with Some size -> pr "size: %Ld" size | None -> () end;
  25   pr "uids:";
  26   List.iter (fun x -> pr "  %s" (CommonTypes.string_of_uid x)) magnet#uids;
  27   pr ""
  28
  29 let test_magnet () =
  30   let t s = test1 ~s magnet s in
  31   t "magnet:?xt=urn:tree:tiger:UXNWMYERN37HJNXB7V6KDJKZXMFBIQAGMDMYDBY&dn=DCPlusPlus-0.4032.exe";
  32   t "magnet:?xt=urn:ed2k:354B15E68FB8F36D7CD88FF94116CDC1&xl=10826029&dn=mediawiki-1.15.1.tar.gz&xt=urn:tree:tiger:7N5OAMRNGMSSEUE3ORHOKWN4WWIQ5X4EBOOTLJY&xt=urn:btih:QHQXPYWMACKDWKP47RRVIV7VOURXFE5Q&tr=http%3A%2F%2Ftracker.example.org%2Fannounce.php%3Fuk%3D1111111111%26&as=http%3A%2F%2Fdownload.wikimedia.org%2Fmediawiki%2F1.15%2Fmediawiki-1.15.1.tar.gz&xs=http%3A%2F%2Fcache.example.org%2FXRX2PEFXOOEJFRVUCX6HMZMKS5TWG4K5&xs=dchub://example.org";
  33   t "magnet:?xt=urn:ed2k:31D6CFE0D16AE931B73C59D7E0C089C0&xl=0&dn=zero_len.fil&xt=urn:bitprint:3I42H3S6NNFQ2MSVX7XZKYAYSCX5QBYJ.LWPNACQDBZRYXW3VHJVCJ64QBZNGHOHHHZWCLNQ&xt=urn:md5:D41D8CD98F00B204E9800998ECF8427E";
  34   ()
  35
  36 let test_shorten () =
  37   let orig = "привет" in
  38   for i = 0 to 100 do
  39     test ~n:i ~s:"shorten" begin fun () ->
  40       let s = DcGlobals.shorten_string orig i in
  41       assert (s = String.sub orig 0 (min (String.length orig) (i*2)))
  42     end
  43   done;
  44   ()
  45
  46 let test_dc_parse () =
  47   let t x s =
  48     test ~s (fun () ->
  49       match DcProtocol.dc_parse false s with
  50       | DcProtocol.UnknownReq _ -> assert (not x)
  51       | _ -> assert x)
  52   in
  53   t true "$ADCGET list /shared1/ 0 -1";
  54   t true "$ADCGET file TTH/ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789012 1332982893 9194387";
  55   t false "$ADCGET tthl q 0 -1"
  56
  57 let test_hmac_md5 () =
  58   test ~s:"HMAC-MD5" begin fun () ->
  59     let t k c s = Mailer.hmac_md5 k c = Md4.Base16.of_string 16 s in
  60     assert (t (String.make 16 '\x0B') "Hi There" "9294727a3638bb1c13f48ef8158bfc9d");
  61     assert (t "Jefe" "what do ya want for nothing?" "750c783e6ab0b503eaa86e310a5db738");
  62     assert (t (String.make 16 '\xAA') (String.make 50 '\xDD') "56be34521d144c88dbb8c733f0e8b3f6");
  63   end
  64
  65 let () =
  66 (*   let _ = Ip.addr_of_string "dchub://83.102.255.226" in *)
  67 (*   let _ = Url.of_string "/submit?q=dcn+dchub://example.com+411" in *)
  68   test_magnet ();
  69   test_shorten ();
  70   test_dc_parse ();
  71   test_hmac_md5 ();
  72   pr "Tests finished";
  73   ()