neale
/
irc-bot
mirror of https://github.com/nealey/irc-bot
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simultaneous connections test

This commit is contained in:
Neale Pickett 2008-03-07 13:52:13 -07:00
parent 8cefe61976
commit babb102cd4
2 changed files with 97 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
chat.ml
View File

@ -161,6 +161,11 @@ object (self)
end end
let chat_create ues script proc =
let a,b = Unix.socketpair Unix.PF_UNIX Unix.SOCK_STREAM 0 in
ignore (proc ues a);
ignore (new chat_handler script ues b)
(** Run a chat script (** Run a chat script
[chat script proc] will create a Unix domain socket pair, call [proc [chat script proc] will create a Unix domain socket pair, call [proc
@ -168,20 +173,16 @@ end
[script] through it. [script] through it.
*) *)
let chat script proc = let chat_run ues =
let ues = new unix_event_system () in try
let a,b = Unix.socketpair Unix.PF_UNIX Unix.SOCK_STREAM 0 in Unixqueue.run ues;
let _ = proc ues a in with
let _ = new chat_handler script ues b in | Chat_match (got, expected) ->
try raise (Failure ("Not matched: got \"" ^
Unixqueue.run ues; (String.escaped got) ^
with "\"\n expected " ^
| Chat_match (got, expected) -> (string_of_chat_event expected)))
raise (Failure ("Not matched: got \"" ^ | Chat_timeout evt ->
(String.escaped got) ^ raise (Failure ("Timeout waiting for " ^
"\"\n expected " ^ (string_of_chat_event evt)))
(string_of_chat_event expected)))
| Chat_timeout evt ->
raise (Failure ("Timeout waiting for " ^
(string_of_chat_event evt)))

117
tests.ml
View File

@ -3,13 +3,7 @@ open OUnit
open Chat open Chat
open Irc open Irc
let do_chat script () = let ues = Unixqueue.create_unix_event_system ()
let ircd_instance ues fd =
let g = Unixqueue.new_group ues in
Unixqueue.add_handler ues g Iobuf.handle_event;
Client.handle_connection ues g fd
in
chat script ircd_instance
let unit_tests = let unit_tests =
"Unit tests" >::: "Unit tests" >:::
@ -37,6 +31,7 @@ let unit_tests =
] ]
let do_login nick = let do_login nick =
[ [
Send ("USER " ^ nick ^ " +iw " ^ nick ^ " :gecos\r\n"); Send ("USER " ^ nick ^ " +iw " ^ nick ^ " :gecos\r\n");
@ -51,38 +46,86 @@ let regression_tests =
"Regression tests" >::: "Regression tests" >:::
[ [
"Simple connection" >:: "Simple connection" >::
(do_chat ((do_login "nick") @ (fun () ->
[ let script =
Send "BLARGH\r\n"; (do_login "nick") @
Recv ":testserver.test 421 nick BLARGH :Unknown or misconstructed command\r\n"; [
Send "MOTD\r\n"; Send "BLARGH\r\n";
Recv ":testserver.test 422 nick :MOTD File is missing\r\n"; Recv ":testserver.test 421 nick BLARGH :Unknown or misconstructed command\r\n";
Send "TIME\r\n"; Send "MOTD\r\n";
Regex ":testserver\\.test 391 nick testserver\\.test :[-0-9]+T[:0-9]+Z\r\n"; Recv ":testserver.test 422 nick :MOTD File is missing\r\n";
Send "VERSION\r\n"; Send "TIME\r\n";
Recv ":testserver.test 351 nick 0.1 testserver.test :\r\n"; Regex ":testserver\\.test 391 nick testserver\\.test :[-0-9]+T[:0-9]+Z\r\n";
Send "PING snot\r\n"; Send "VERSION\r\n";
Recv ":testserver.test PONG testserver.test :snot\r\n"; Recv ":testserver.test 351 nick 0.1 testserver.test :\r\n";
Send "PING :snot\r\n"; Send "PING snot\r\n";
Recv ":testserver.test PONG testserver.test :snot\r\n"; Recv ":testserver.test PONG testserver.test :snot\r\n";
Send "ISON nick otherguy\r\n"; Send "PING :snot\r\n";
Recv ":testserver.test 303 nick :nick\r\n"; Recv ":testserver.test PONG testserver.test :snot\r\n";
Send "ISON otherguy thirdguy\r\n"; Send "ISON nick otherguy\r\n";
Recv ":testserver.test 303 nick :\r\n"; Recv ":testserver.test 303 nick :nick\r\n";
Send "PRIVMSG nick :hello\r\n"; Send "ISON otherguy thirdguy\r\n";
Recv ":nick!nick@UDS PRIVMSG nick :hello\r\n"; Recv ":testserver.test 303 nick :\r\n";
Send "NOTICE nick :hello\r\n"; Send "PRIVMSG nick :hello\r\n";
Recv ":nick!nick@UDS NOTICE nick :hello\r\n"; Recv ":nick!nick@UDS PRIVMSG nick :hello\r\n";
Send "PRIVMSG otherguy :hello\r\n"; Send "NOTICE nick :hello\r\n";
Recv ":testserver.test 401 nick otherguy :No such nick/channel\r\n"; Recv ":nick!nick@UDS NOTICE nick :hello\r\n";
])); Send "PRIVMSG otherguy :hello\r\n";
Recv ":testserver.test 401 nick otherguy :No such nick/channel\r\n";
]
in
let g = Unixqueue.new_group ues in
let a,b = Unix.socketpair Unix.PF_UNIX Unix.SOCK_STREAM 0 in
Unixqueue.add_handler ues g Iobuf.handle_event;
Client.handle_connection ues g a;
ignore (new chat_handler script ues b);
chat_run ues);
"Second connection" >:: "Second connection" >::
(do_chat ((do_login "otherguy") @ (fun () ->
[ let script =
Send "ISON nick otherguy\r\n"; (do_login "otherguy") @
Recv ":testserver.test 303 otherguy :otherguy\r\n"; [
])); Send "ISON nick otherguy\r\n";
Recv ":testserver.test 303 otherguy :otherguy\r\n";
]
in
let g = Unixqueue.new_group ues in
let a,b = Unix.socketpair Unix.PF_UNIX Unix.SOCK_STREAM 0 in
Unixqueue.add_handler ues g Iobuf.handle_event;
Client.handle_connection ues g a;
ignore (new chat_handler script ues b);
chat_run ues);
"Simultaneous connections" >::
(fun () ->
let script1 =
(do_login "alice") @
[
Send "ISON bob\r\n";
Recv ":testserver.test 303 alice :bob\r\n";
Send "PRIVMSG bob :Hi Bob!\r\n";
Send "PING :foo\r\n"; (* Make sure we don't disconnect too soon *)
Recv ":testserver.test PONG testserver.test :foo\r\n";
]
in
let script2 =
(do_login "bob") @
[
Send "ISON alice\r\n";
Recv ":testserver.test 303 bob :alice\r\n";
Recv ":alice!alice@UDS PRIVMSG bob :Hi Bob!\r\n";
]
in
let g = Unixqueue.new_group ues in
let aa,ab = Unix.socketpair Unix.PF_UNIX Unix.SOCK_STREAM 0 in
let ba,bb = Unix.socketpair Unix.PF_UNIX Unix.SOCK_STREAM 0 in
Unixqueue.add_handler ues g Iobuf.handle_event;
Client.handle_connection ues g aa;
Client.handle_connection ues g ba;
ignore (new chat_handler script1 ues ab);
ignore (new chat_handler script2 ues bb);
chat_run ues);
] ]
let _ = let _ =