unittest.ml

   1 (***********************************************************************)
   2 (* The OcamlEvent library                                              *)
   3 (*                                                                     *)
   4 (* Copyright 2002, 2003 Maas-Maarten Zeeman. All rights reserved. See  *)
   5 (* LICENCE for details.                                                *)
   6 (***********************************************************************)
   7
   8 (* $Id: unittest.ml,v 1.9 2004/12/21 19:15:14 maas Exp $ *)
   9
  10 open OUnit
  11
  12 let create_event = Libevent.create
  13
  14 (* Tests the creation of new events *)
  15 let test_create_event _ =
  16   let e1 = create_event () in
  17   let e2 = create_event () in
  18   "Events should be different" @? (e1 <> e2)
  19
  20 (* Tests if pending can be called with and without the optional float *)
  21 let test_pending _ =
  22   let e = create_event () in
  23   ignore(Libevent.pending e [])
  24
  25 (* Test eof on a read callback *)
  26 let test_read_eof _ =
  27   let test_string = "This is a test string\n\n\n" in
  28   let buflen = 512 in
  29   let buf = String.create buflen in
  30   let read_count = ref 0 in
  31   let evt = create_event () in
  32   let read_cb fd event_type =
  33     (* read data from the fd *)
  34     let len = Unix.read fd buf 0 buflen in
  35     (* when 0 bytes are read this is the EOF, and we are done. *)
  36     if len <> 0 then
  37       begin
  38         read_count := !read_count + len;
  39         Libevent.add evt None
  40       end
  41   in
  42
  43   (* Create a socket pair for testing *)
  44   let s1, s2 = Unix.socketpair Unix.PF_UNIX Unix.SOCK_STREAM 0 in
  45   let l = Unix.write s1 test_string 0 (String.length test_string) in
  46
  47   (* A shutdown_send will cause an EOF on the reading end *)
  48   Unix.shutdown s1 Unix.SHUTDOWN_SEND;
  49
  50   (* Setup the event *)
  51   Libevent.set evt s2 [Libevent.READ] false read_cb;
  52   Libevent.add evt None;
  53   Libevent.dispatch ();
  54
  55   (* Now its time to check some things *)
  56   assert_equal (String.length test_string) !read_count
  57
  58 (* This is not really a test (yet) *)
  59 let call_set _ =
  60   let do_nothing _ _ =
  61     ()
  62   in
  63   let e1 = Libevent.create () in
  64   Libevent.set e1 Unix.stderr [Libevent.WRITE] false do_nothing;
  65   Libevent.set e1 Unix.stdout [Libevent.WRITE] false do_nothing;
  66   Libevent.set e1 Unix.stdin [Libevent.READ] false do_nothing;
  67   Libevent.add e1 (Some 0.1);
  68   Libevent.loop(Libevent.ONCE)
  69
  70 (* Construct the test suite *)
  71 let suite = "event" >:::
  72   ["create_event" >:: test_create_event;
  73    "test_pending" >:: test_pending;
  74    "test_read_eof" >:: test_read_eof;
  75    "call_set" >:: call_set;
  76  ]
  77
  78 (* Run the tests in the test suite *)
  79 let _ =
  80   run_test_tt_main suite