room_test.go

   1 /*
   2 goircd -- minimalistic simple Internet Relay Chat (IRC) server
   3 Copyright (C) 2014-2015 Sergey Matveev <stargrave@stargrave.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, either version 3 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 package main
  20
  21 import (
  22         "strings"
  23         "testing"
  24 )
  25
  26 func noNickchan(t *testing.T, c *TestingConn) {
  27         if r := <-c.outbound; !strings.HasPrefix(r, ":foohost 401") {
  28                 t.Fatal("no nick/channel", r)
  29         }
  30 }
  31
  32 func noChan(t *testing.T, c *TestingConn) {
  33         if r := <-c.outbound; !strings.HasPrefix(r, ":foohost 403") {
  34                 t.Fatal("no channel", r)
  35         }
  36 }
  37
  38 func notEnoughParams(t *testing.T, c *TestingConn) {
  39         if r := <-c.outbound; !strings.HasPrefix(r, ":foohost 461") {
  40                 t.Fatal("not enough params", r)
  41         }
  42 }
  43
  44 func TestTwoUsers(t *testing.T) {
  45         logSink := make(chan LogEvent, 8)
  46         stateSink := make(chan StateEvent, 8)
  47         host := "foohost"
  48         daemon := NewDaemon("ver1", &host, nil, nil, logSink, stateSink)
  49         events := make(chan ClientEvent)
  50         go daemon.Processor(events)
  51
  52         conn1 := NewTestingConn()
  53         conn2 := NewTestingConn()
  54         client1 := NewClient(&host, conn1)
  55         client2 := NewClient(&host, conn2)
  56         go client1.Processor(events)
  57         go client2.Processor(events)
  58
  59         conn1.inbound <- "NICK nick1\r\nUSER foo1 bar1 baz1 :Long name1\r\n"
  60         conn2.inbound <- "NICK nick2\r\nUSER foo2 bar2 baz2 :Long name2\r\n"
  61         for i := 0; i < 6; i++ {
  62                 <-conn1.outbound
  63                 <-conn2.outbound
  64         }
  65
  66         daemon.SendLusers(client1)
  67         if r := <-conn1.outbound; !strings.Contains(r, "There are 2 users") {
  68                 t.Fatal("LUSERS", r)
  69         }
  70
  71         conn1.inbound <- "WHOIS"
  72         notEnoughParams(t, conn1)
  73         conn1.inbound <- "WHOIS nick3"
  74         noNickchan(t, conn1)
  75         conn1.inbound <- "WHOIS nick2"
  76         if r := <-conn1.outbound; r != ":foohost 311 nick1 nick2 foo2 Unknown * :Long name2\r\n" {
  77                 t.Fatal("first WHOIS 311", r)
  78         }
  79         if r := <-conn1.outbound; r != ":foohost 312 nick1 nick2 foohost :foohost\r\n" {
  80                 t.Fatal("first WHOIS 312", r)
  81         }
  82         if r := <-conn1.outbound; r != ":foohost 319 nick1 nick2 :\r\n" {
  83                 t.Fatal("first WHOIS 319", r)
  84         }
  85         if r := <-conn1.outbound; r != ":foohost 318 nick1 nick2 :End of /WHOIS list\r\n" {
  86                 t.Fatal("first WHOIS 318", r)
  87         }
  88
  89         conn1.inbound <- "LIST"
  90         if r := <-conn1.outbound; r != ":foohost 323 nick1 :End of /LIST\r\n" {
  91                 t.Fatal("first LIST", r)
  92         }
  93
  94         conn1.inbound <- "WHO"
  95         notEnoughParams(t, conn1)
  96         conn1.inbound <- "WHO #fooroom"
  97         noChan(t, conn1)
  98
  99         conn1.inbound <- "JOIN #foo"
 100         conn2.inbound <- "JOIN #foo"
 101         for i := 0; i < 4; i++ {
 102                 <-conn1.outbound
 103                 <-conn2.outbound
 104         }
 105         conn1.inbound <- "PRIVMSG nick2 :Hello"
 106         conn1.inbound <- "PRIVMSG #foo :world"
 107         conn1.inbound <- "NOTICE #foo :world"
 108         if r := <-conn2.outbound; r != ":nick1!foo1@someclient PRIVMSG nick2 :Hello\r\n" {
 109                 t.Fatal("first message", r)
 110         }
 111         if r := <-conn2.outbound; r != ":nick1!foo1@someclient PRIVMSG #foo :world\r\n" {
 112                 t.Fatal("second message", r)
 113         }
 114         if r := <-conn2.outbound; r != ":nick1!foo1@someclient NOTICE #foo :world\r\n" {
 115                 t.Fatal("third message", r)
 116         }
 117 }
 118
 119 func TestJoin(t *testing.T) {
 120         logSink := make(chan LogEvent, 8)
 121         stateSink := make(chan StateEvent, 8)
 122         host := "foohost"
 123         daemon := NewDaemon("ver1", &host, nil, nil, logSink, stateSink)
 124         events := make(chan ClientEvent)
 125         go daemon.Processor(events)
 126         conn := NewTestingConn()
 127         client := NewClient(&host, conn)
 128         go client.Processor(events)
 129
 130         conn.inbound <- "NICK nick2\r\nUSER foo2 bar2 baz2 :Long name2\r\n"
 131         for i := 0; i < 6; i++ {
 132                 <-conn.outbound
 133         }
 134
 135         conn.inbound <- "JOIN"
 136         notEnoughParams(t, conn)
 137         conn.inbound <- "JOIN bla/bla/bla"
 138         noChan(t, conn)
 139         conn.inbound <- "JOIN bla:bla:bla"
 140         noChan(t, conn)
 141
 142         conn.inbound <- "JOIN #foo"
 143         if r := <-conn.outbound; r != ":foohost 331 nick2 #foo :No topic is set\r\n" {
 144                 t.Fatal("no topic is set", r)
 145         }
 146         if r := <-conn.outbound; r != ":nick2!foo2@someclient JOIN #foo\r\n" {
 147                 t.Fatal("no JOIN message", r)
 148         }
 149         if r := <-conn.outbound; r != ":foohost 353 nick2 = #foo :nick2\r\n" {
 150                 t.Fatal("no NAMES list", r)
 151         }
 152         if r := <-conn.outbound; r != ":foohost 366 nick2 #foo :End of NAMES list\r\n" {
 153                 t.Fatal("no end of NAMES list", r)
 154         }
 155         if r := <-logSink; (r.what != "joined") || (r.where != "#foo") || (r.who != "nick2") || (r.meta != true) {
 156                 t.Fatal("invalid join log event", r)
 157         }
 158
 159         conn.inbound <- "JOIN #bar,#baz"
 160         for i := 0; i < 4*2; i++ {
 161                 <-conn.outbound
 162         }
 163         if _, ok := daemon.rooms["#bar"]; !ok {
 164                 t.Fatal("#bar does not exist")
 165         }
 166         if _, ok := daemon.rooms["#baz"]; !ok {
 167                 t.Fatal("#baz does not exist")
 168         }
 169         if r := <-logSink; (r.what != "joined") || (r.where != "#bar") || (r.who != "nick2") || (r.meta != true) {
 170                 t.Fatal("invalid join log event #bar", r)
 171         }
 172         if r := <-logSink; (r.what != "joined") || (r.where != "#baz") || (r.who != "nick2") || (r.meta != true) {
 173                 t.Fatal("invalid join log event #baz", r)
 174         }
 175
 176         conn.inbound <- "JOIN #barenc,#bazenc key1,key2"
 177         for i := 0; i < 4*2; i++ {
 178                 <-conn.outbound
 179         }
 180         if daemon.rooms["#barenc"].key != "key1" {
 181                 t.Fatal("no room with key1")
 182         }
 183         if daemon.rooms["#bazenc"].key != "key2" {
 184                 t.Fatal("no room with key2")
 185         }
 186         if r := <-logSink; (r.what != "joined") || (r.where != "#barenc") || (r.who != "nick2") || (r.meta != true) {
 187                 t.Fatal("invalid join log event #barenc", r)
 188         }
 189         if r := <-logSink; (r.what != "joined") || (r.where != "#bazenc") || (r.who != "nick2") || (r.meta != true) {
 190                 t.Fatal("invalid join log event #bazenc", r)
 191         }
 192         if r := <-stateSink; (r.topic != "") || (r.where != "#barenc") || (r.key != "key1") {
 193                 t.Fatal("set channel key1 state", r)
 194         }
 195         if r := <-stateSink; (r.topic != "") || (r.where != "#bazenc") || (r.key != "key2") {
 196                 t.Fatal("set channel key2 state", r)
 197         }
 198
 199         conn.inbound <- "MODE #barenc -k"
 200         if r := <-conn.outbound; r != ":nick2!foo2@someclient MODE #barenc -k\r\n" {
 201                 t.Fatal("remove #barenc key", r)
 202         }
 203         if daemon.rooms["#barenc"].key != "" {
 204                 t.Fatal("removing key from #barenc")
 205         }
 206         if r := <-logSink; (r.what != "removed channel key") || (r.where != "#barenc") || (r.who != "nick2") || (r.meta != true) {
 207                 t.Fatal("removed channel key log", r)
 208         }
 209         if r := <-stateSink; (r.topic != "") || (r.where != "#barenc") || (r.key != "") {
 210                 t.Fatal("removed channel key state", r)
 211         }
 212
 213         conn.inbound <- "PART #bazenc\r\nMODE #bazenc -k"
 214         if r := <-conn.outbound; r != ":foohost 442 #bazenc :You are not on that channel\r\n" {
 215                 t.Fatal("not on that channel", r)
 216         }
 217         if r := <-logSink; (r.what != "left") || (r.where != "#bazenc") || (r.who != "nick2") || (r.meta != true) {
 218                 t.Fatal("left #bazenc log", r)
 219         }
 220
 221         conn.inbound <- "MODE #barenc +i"
 222         if r := <-conn.outbound; r != ":foohost 472 nick2 +i :Unknown MODE flag\r\n" {
 223                 t.Fatal("unknown MODE flag", r)
 224         }
 225
 226         conn.inbound <- "MODE #barenc +k newkey"
 227         if r := <-conn.outbound; r != ":nick2!foo2@someclient MODE #barenc +k newkey\r\n" {
 228                 t.Fatal("+k MODE setting", r)
 229         }
 230         if r := <-logSink; (r.what != "set channel key to newkey") || (r.where != "#barenc") || (r.who != "nick2") || (r.meta != true) {
 231                 t.Fatal("set channel key", r)
 232         }
 233         if r := <-stateSink; (r.topic != "") || (r.where != "#barenc") || (r.key != "newkey") {
 234                 t.Fatal("set channel newkey state", r)
 235         }
 236
 237         conn.inbound <- "TOPIC #barenc :New topic"
 238         if r := <-conn.outbound; r != ":nick2!foo2@someclient TOPIC #barenc :New topic\r\n" {
 239                 t.Fatal("set TOPIC", r)
 240         }
 241         if r := <-logSink; (r.what != "set topic to New topic") || (r.where != "#barenc") || (r.who != "nick2") || (r.meta != true) {
 242                 t.Fatal("set TOPIC log", r)
 243         }
 244         if r := <-stateSink; (r.topic != "New topic") || (r.where != "#barenc") || (r.key != "newkey") {
 245                 t.Fatal("set channel TOPIC state", r)
 246         }
 247
 248         conn.inbound <- "WHO #barenc"
 249         if r := <-conn.outbound; r != ":foohost 352 nick2 #barenc foo2 someclient foohost nick2 H :0 Long name2\r\n" {
 250                 t.Fatal("WHO", r)
 251         }
 252         if r := <-conn.outbound; r != ":foohost 315 nick2 #barenc :End of /WHO list\r\n" {
 253                 t.Fatal("end of WHO", r)
 254         }
 255
 256 }