irc-bot/tests.ml

open Unixqueue
open OUnit
open Chat
open Irc

let do_chat script () =
  let ircd_instance ues fd =
    let g = Unixqueue.new_group ues in
      ignore (Client.create ues g fd);
      Unixqueue.add_handler ues g Client.handle_event;
      Unixqueue.add_resource ues g (Unixqueue.Wait_in fd, -.1.0)
  in
    chat script ircd_instance

let unit_tests =
  "Unit tests" >:::
    [
      "command_of_string" >:: 
	(fun () ->
	   assert_equal
	     ~printer:Command.as_string
	     (Command.create "NICK" ["name"])
	     (Command.from_string "NICK name");
	   assert_equal
	     ~printer:Command.as_string
	     (Command.create ~sender:(Some "foo") "NICK" ["name"])
	     (Command.from_string ":foo NICK name");
	   assert_equal
	     ~printer:Command.as_string
	     (Command.create 
		~sender:(Some "foo.bar") 
		~text:(Some "ta ta") 
		"PART" ["#foo"; "#bar"])
	     (Command.from_string ":foo.bar PART #foo #bar :ta ta");
	)
    ]
      

let regression_tests =
  let login_script =
    [
      Send "USER nick +iw nick :gecos\r\n";
      Send "NICK nick\r\n";
      Recv ":testserver.test NOTICE nick :*** Hi there.\r\n";
    ]
  in
    "Regression tests" >:::
      [
	"Simple connection" >::
	  (do_chat
	     login_script);

	"Full connection" >::
	  (do_chat
	     (login_script @
		[
		  Recv ":testserver.test 001 nick :Welcome to the test script\r\n";
		  Recv ":testserver.test 002 nick :Your host is testserver.test\r\n";
		  Recv ":testserver.test 003 nick :This server is timeless\r\n";
		  Recv ":testserver.test 004 nick testserver.test testscript DGabcdfg bilmnopst bkloveI\r\n";
		  Recv ":testserver.test 005 nick CALLERID CASEMAPPING=rfc1459 KICKLEN=160 MODES=4 WHATEVER=4 WHO=1 CARES=3 :are supported by this server\r\n";
		  Recv ":testserver.test 043 00XAAAAL6 :your unique ID\r\n";
		  Recv ":testserver.test 251 nick :There are 14 users and 4 invisible on 1 servers\r\n";
		  Recv ":testserver.test 252 nick 1 :IRC Operators online\r\n";
		  Recv ":testserver.test 254 4 :channels formed\r\n";
		  Recv ":testserver.test 255 nick :I have 17 clients and 0 servers\r\n";
		  Recv ":testserver.test 265 nick :Current local users: 17  Max: 25\r\n";
		  Recv ":testserver.test 266 nick :Current global users: 17  Max: 25\r\n";
		  Recv ":testserver.test 250 nick :Highest connection count: 25 (25 clients) (430 connections received)\r\n";
		  Recv ":testserver.test 375 nick :- xirc.lanl.gov Message of the Day -\r\n";
		  Recv ":testserver.test 372 nick :- This is ircd-hybrid MOTD replace it with something better\r\n";
		  Recv ":testserver.test 376 nick :End of /MOTD command.\r\n";
		]
	     ));
      ]

let _ =
  Irc.name := "testserver.test";
  run_test_tt_main (TestList [unit_tests; regression_tests])