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.3 2009-11-26 09:10:37 maas Exp $ *)
   9
  10 open OUnit
  11 open Libevent
  12
  13 (* Tests the creation of new events *)
  14 let test_create_event () =
  15   let e1 = create () in
  16   let e2 = create () in
  17   "Events should be different" @? (e1 <> e2)
  18
  19 let test_pending () =
  20   let base = init () in
  21   let e = create () in
  22   let all = [READ;WRITE;SIGNAL;TIMEOUT] in
  23   "Fresh event is not pending" @? (false = pending e all);
  24   set_timer base e ~persist:true (fun () -> assert false);
  25   "Still not pending" @? (false = pending e all);
  26   add e (Some 120.);
  27   "Pending on some type" @? (pending e all);
  28   "Pending on TIMEOUT" @? (pending e [TIMEOUT]);
  29   "Not pending on READ" @? (false = pending e [READ]);
  30   "Not pending on empty flags" @? (false = pending e []);
  31   del e;
  32   free base;
  33   "Not pending on any type" @? (false = pending e all);
  34   ()
  35
  36 (* Test eof on a read callback *)
  37 let test_read_eof () =
  38   let test_string = "This is a test string\n\n\n" in
  39   let buflen = 512 in
  40   let buf = String.create buflen in
  41   let read_count = ref 0 in
  42   let evt = create () in
  43   let read_cb fd event_type =
  44     (* read data from the fd *)
  45     let len = Unix.read fd buf 0 buflen in
  46     (* when 0 bytes are read this is the EOF, and we are done. *)
  47     if len <> 0 then
  48     begin
  49       read_count := !read_count + len;
  50       add evt None
  51     end
  52   in
  53
  54   (* Create a socket pair for testing *)
  55   let s1, s2 = Unix.socketpair Unix.PF_UNIX Unix.SOCK_STREAM 0 in
  56   let _ = Unix.write s1 test_string 0 (String.length test_string) in
  57
  58   (* A shutdown_send will cause an EOF on the reading end *)
  59   Unix.shutdown s1 Unix.SHUTDOWN_SEND;
  60
  61   (* Setup the event *)
  62   Global.set evt s2 [READ] false read_cb;
  63   add evt None;
  64   Global.dispatch ();
  65
  66   (* Now its time to check some things *)
  67   assert_equal (String.length test_string) ! read_count
  68
  69 (* This is not really a test (yet) *)
  70 let call_set () =
  71   let do_nothing _ _ =
  72     ()
  73   in
  74   let e1 = create () in
  75   Global.set e1 Unix.stderr [WRITE] false do_nothing;
  76   Global.set e1 Unix.stdout [WRITE] false do_nothing;
  77   Global.set e1 Unix.stdin [READ] false do_nothing;
  78   add e1 (Some 0.1);
  79   Global.loop ONCE
  80
  81 let test_free () =
  82   let base = init () in
  83   let ev = create () in
  84   let called = ref 0 in
  85   set_timer base ev ~persist:true (fun () -> incr called);
  86   add ev (Some 1.);
  87   loop base ONCE;
  88   free base;
  89   Gc.compact ();
  90   "reached end with callback called once as expected" @? (!called = 1)
  91
  92 (* Construct the test suite *)
  93 let suite = "event" >:::
  94   ["create_event" >:: test_create_event;
  95    "test_pending" >:: test_pending;
  96    "test_read_eof" >:: test_read_eof;
  97    "call_set" >:: call_set;
  98    "event_base_free" >:: test_free;
  99  ]
 100
 101 (* Run the tests in the test suite *)
 102 let _ =
 103   run_test_tt_main suite