root/lang/ruby/net-irc/trunk/spec/channel_manager_spec.rb @ 15296

#!/usr/bin/env ruby

$LOAD_PATH << "lib"
$LOAD_PATH << "../lib"

require "net/irc"
require "net/irc/client/channel_manager"
include Net::IRC
include Constants

class ChannelManagerTestServerSession < Net::IRC::Server::Session
        @@testq = SizedQueue.new(1)
        @@instance = nil

        def self.testq
                @@testq
        end

        def self.instance
                @@instance
        end

        def initialize(*args)
                super
                @@instance = self
        end

        def on_message(m)
                @@testq << m
        end
end

class ChannelManagerTestClient < Net::IRC::Client
        include Net::IRC::Client::ChannelManager
        @@testq = SizedQueue.new(1)

        def self.testq
                @@testq
        end

        def on_message(m)
                @@testq << m
        end
end

describe Net::IRC, "server and client" do
        before :all do
                @port = nil
                @server, @client = nil, nil

                Thread.abort_on_exception = true
                Thread.start do
                        @server = Net::IRC::Server.new("localhost", @port, ChannelManagerTestServerSession, {
                                :logger => Logger.new(nil),
                        })
                        @server.start
                end

                true until @server.instance_variable_get(:@serv)

                @port = @server.instance_variable_get(:@serv).addr[1]

                Thread.start do
                        @client = ChannelManagerTestClient.new("localhost", @port, {
                                :nick   => "foonick",
                                :user   => "foouser",
                                :real   => "foo real name",
                                :pass   => "foopass",
                                :logger => Logger.new(nil),
                        })
                        @client.start
                end
        end

        server_q = ChannelManagerTestServerSession.testq
        client_q = ChannelManagerTestClient.testq

        it "client should manage channel mode/users correctly" do
                client = @client
                client.instance_variable_set(:@prefix, Prefix.new("foonick!foouser@localhost"))

                ChannelManagerTestServerSession.instance.instance_eval do
                        Thread.exclusive do
                                post client.prefix,          JOIN,   "#test"
                                post nil,                    NOTICE, "#test", "sep1"
                        end
                end

                true until client_q.pop.to_s == "NOTICE #test sep1\r\n"

                c = @client.instance_variable_get(:@channels)
                c.synchronize do
                        c.should                       be_a_kind_of(Hash)
                        c["#test"].should              be_a_kind_of(Hash)
                        c["#test"][:modes].should      be_a_kind_of(Array)
                        c["#test"][:users].should      be_a_kind_of(Array)
                        c["#test"][:users].should      == ["foonick"]
                end

                ChannelManagerTestServerSession.instance.instance_eval do
                        Thread.exclusive do
                                post "test1!test@localhost", JOIN,   "#test"
                                post "test2!test@localhost", JOIN,   "#test"
                                post nil,                    NOTICE, "#test", "sep2"
                        end
                end

                true until client_q.pop.to_s == "NOTICE #test sep2\r\n"
                c.synchronize do
                        c["#test"][:users].should      == ["foonick", "test1", "test2"]
                end

                ChannelManagerTestServerSession.instance.instance_eval do
                        Thread.exclusive do
                                post nil,                    RPL_NAMREPLY, client.prefix.nick, "@", "#test", "foo1 foo2 foo3 @foo4 +foo5"
                                post nil,                    NOTICE, "#test", "sep3"
                        end
                end

                true until client_q.pop.to_s == "NOTICE #test sep3\r\n"
                c.synchronize do
                        c["#test"][:users].should      == ["foonick", "test1", "test2", "foo1", "foo2", "foo3", "foo4", "foo5"]
                        c["#test"][:modes].should      include([:s, nil])
                        c["#test"][:modes].should      include([:o, "foo4"])
                        c["#test"][:modes].should      include([:v, "foo5"])
                end

                ChannelManagerTestServerSession.instance.instance_eval do
                        Thread.exclusive do
                                post nil,                    RPL_NAMREPLY, client.prefix.nick, "@", "#test1", "foo1 foo2 foo3 @foo4 +foo5"
                                post "foo4!foo@localhost",   QUIT,   "message"
                                post "foo5!foo@localhost",   PART,   "#test1", "message"
                                post client.prefix,          KICK,   "#test", "foo1", "message"
                                post client.prefix,          MODE,   "#test", "+o", "foo2"
                                post nil,                    NOTICE, "#test", "sep4"
                        end
                end

                true until client_q.pop.to_s == "NOTICE #test sep4\r\n"
                c.synchronize do
                        c["#test"][:users].should      == ["foonick", "test1", "test2", "foo2", "foo3", "foo5"]
                        c["#test1"][:users].should     == ["foo1", "foo2", "foo3"]
                        c["#test"][:modes].should_not  include([:o, "foo4"])
                        c["#test"][:modes].should      include([:v, "foo5"])
                        c["#test1"][:modes].should_not include([:v, "foo5"])
                        c["#test"][:modes].should      include([:o, "foo2"])
                end

                ChannelManagerTestServerSession.instance.instance_eval do
                        Thread.exclusive do
                                post "foonick!test@localhost",  NICK, "foonick2"
                                post "foonick2!test@localhost", NICK, "foonick"
                                post "foo2!test@localhost",     NICK, "bar2"
                                post "foo3!test@localhost",     NICK, "bar3"
                                post nil,                     NOTICE, "#test", "sep5"
                        end
                end

                true until client_q.pop.to_s == "NOTICE #test sep5\r\n"
                c.synchronize do
                        c["#test"][:users].should      == ["foonick", "test1", "test2", "bar2", "bar3", "foo5"]
                        c["#test1"][:users].should     == ["foo1", "bar2", "bar3"]
                        c["#test"][:modes].should_not  include([:o, "foo4"])
                        c["#test"][:modes].should      include([:v, "foo5"])
                        c["#test1"][:modes].should_not include([:v, "foo5"])
                        c["#test"][:modes].should_not  include([:o, "foo2"])
                        c["#test"][:modes].should      include([:o, "bar2"])
                end
        end

        after :all do
                @server.finish
                @client.finish
        end
end