open Unixqueue
open OUnit
open Chat
open Irc

let do_chat script () =
  let ircd_instance ues fd =
    let srv = Server.create () in
    let handle_event = Client.create_event_handler srv in
    let g = Unixqueue.new_group ues in
    let cli = Client.create ues g fd in
      Hashtbl.replace srv.Irc.clients_by_file_descr fd cli;
      Unixqueue.add_handler ues g 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:string_of_command
	     {sender = None;
	      command = "NICK";
	      args = ["name"];
	      text = None}
	     (command_of_string "NICK name");
	   assert_equal
	     ~printer:string_of_command
	     {sender = Some "foo";
	      command = "NICK";
	      args = ["name"];
	      text = None}
	     (command_of_string ":foo NICK name");
	   assert_equal
	     ~printer:string_of_command
	     {sender = Some "foo.bar";
	      command = "PART";
	      args = ["#foo"; "#bar"];
	      text = Some "ta ta"}
	     (command_of_string ":foo.bar PART #foo #bar :ta ta");
	)
    ]
      

let regression_tests =
  let login_script =
    [
      Send "USER nick +iw nick :gecos\n";
      Send "NICK nick\n";
      Recv ":testserver.test NOTICE nick :*** Hi there.\n";
      Recv "PING :12345\n";
      Send "PONG :12345\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\n";
		  Recv ":testserver.test 002 nick :Your host is testserver.test\n";
		  Recv ":testserver.test 003 nick :This server is timeless\n";
		  Recv ":testserver.test 004 nick testserver.test testscript DGabcdfg bilmnopst bkloveI\n";
		  Recv ":testserver.test 005 nick CALLERID CASEMAPPING=rfc1459 KICKLEN=160 MODES=4 WHATEVER=4 WHO=1 CARES=3 :are supported by this server\n";
		  Recv ":testserver.test 043 00XAAAAL6 :your unique ID\n";
		  Recv ":testserver.test 251 nick :There are 14 users and 4 invisible on 1 servers\n";
		  Recv ":testserver.test 252 nick 1 :IRC Operators online\n";
		  Recv ":testserver.test 254 4 :channels formed\n";
		  Recv ":testserver.test 255 nick :I have 17 clients and 0 servers\n";
		  Recv ":testserver.test 265 nick :Current local users: 17  Max: 25\n";
		  Recv ":testserver.test 266 nick :Current global users: 17  Max: 25\n";
		  Recv ":testserver.test 250 nick :Highest connection count: 25 (25 clients) (430 connections received)\n";
		  Recv ":testserver.test 375 nick :- xirc.lanl.gov Message of the Day -\n";
		  Recv ":testserver.test 372 nick :- This is ircd-hybrid MOTD replace it with something better\n";
		  Recv ":testserver.test 376 nick :End of /MOTD command.\n";
		]
	     ));
      ]

let _ =
  run_test_tt_main (TestList [unit_tests; regression_tests])