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         hostname = &host
  49         events := make(chan ClientEvent)
  50         rooms = make(map[string]*Room)
  51         clients = make(map[*Client]struct{})
  52         roomSinks = make(map[*Room]chan ClientEvent)
  53         finished := make(chan struct{})
  54         go Processor(events, finished)
  55         defer func() {
  56                 events <- ClientEvent{eventType: EventTerm}
  57                 <-finished
  58         }()
  59
  60         conn1 := NewTestingConn()
  61         conn2 := NewTestingConn()
  62         client1 := NewClient(conn1)
  63         client2 := NewClient(conn2)
  64         go client1.Processor(events)
  65         go client2.Processor(events)
  66
  67         conn1.inbound <- "NICK nick1\r\nUSER foo1 bar1 baz1 :Long name1"
  68         conn2.inbound <- "NICK nick2\r\nUSER foo2 bar2 baz2 :Long name2"
  69         for i := 0; i < 6; i++ {
  70                 <-conn1.outbound
  71                 <-conn2.outbound
  72         }
  73
  74         SendLusers(client1)
  75         if r := <-conn1.outbound; !strings.Contains(r, "There are 2 users") {
  76                 t.Fatal("LUSERS", r)
  77         }
  78
  79         conn1.inbound <- "WHOIS"
  80         notEnoughParams(t, conn1)
  81         conn1.inbound <- "WHOIS nick3"
  82         noNickchan(t, conn1)
  83         conn1.inbound <- "WHOIS nick2"
  84         if r := <-conn1.outbound; r != ":foohost 311 nick1 nick2 foo2 Unknown * :Long name2\r\n" {
  85                 t.Fatal("first WHOIS 311", r)
  86         }
  87         if r := <-conn1.outbound; r != ":foohost 312 nick1 nick2 foohost :foohost\r\n" {
  88                 t.Fatal("first WHOIS 312", r)
  89         }
  90         if r := <-conn1.outbound; r != ":foohost 319 nick1 nick2 :\r\n" {
  91                 t.Fatal("first WHOIS 319", r)
  92         }
  93         if r := <-conn1.outbound; r != ":foohost 318 nick1 nick2 :End of /WHOIS list\r\n" {
  94                 t.Fatal("first WHOIS 318", r)
  95         }
  96
  97         conn1.inbound <- "LIST"
  98         if r := <-conn1.outbound; r != ":foohost 323 nick1 :End of /LIST\r\n" {
  99                 t.Fatal("first LIST", r)
 100         }
 101
 102         conn1.inbound <- "WHO"
 103         notEnoughParams(t, conn1)
 104         conn1.inbound <- "WHO #fooroom"
 105         noChan(t, conn1)
 106
 107         conn1.inbound <- "JOIN #foo"
 108         conn2.inbound <- "JOIN #foo"
 109         for i := 0; i < 4; i++ {
 110                 <-conn1.outbound
 111                 <-conn2.outbound
 112         }
 113         conn1.inbound <- "PRIVMSG nick2 :Hello"
 114         conn1.inbound <- "PRIVMSG #foo :world"
 115         conn1.inbound <- "NOTICE #foo :world"
 116         m1 := <-conn2.outbound
 117         m2 := <-conn2.outbound
 118         mNeeded := ":nick1!foo1@someclient PRIVMSG nick2 :Hello\r\n"
 119         if !(m1 == mNeeded || m2 == mNeeded) {
 120                 t.Fatal("first message", m1, m2)
 121         }
 122         if m2 == mNeeded {
 123                 m2 = <-conn2.outbound
 124         }
 125         if m2 != ":nick1!foo1@someclient PRIVMSG #foo :world\r\n" {
 126                 t.Fatal("second message", m2)
 127         }
 128         if m2 = <-conn2.outbound; m2 != ":nick1!foo1@someclient NOTICE #foo :world\r\n" {
 129                 t.Fatal("third message", m2)
 130         }
 131 }
 132
 133 func TestJoin(t *testing.T) {
 134         logSink = make(chan LogEvent, 8)
 135         stateSink = make(chan StateEvent, 8)
 136         host := "foohost"
 137         hostname = &host
 138         events := make(chan ClientEvent)
 139         rooms = make(map[string]*Room)
 140         clients = make(map[*Client]struct{})
 141         roomSinks = make(map[*Room]chan ClientEvent)
 142         finished := make(chan struct{})
 143         go Processor(events, finished)
 144         defer func() {
 145                 events <- ClientEvent{eventType: EventTerm}
 146                 <-finished
 147         }()
 148         conn := NewTestingConn()
 149         client := NewClient(conn)
 150         go client.Processor(events)
 151
 152         conn.inbound <- "NICK nick2\r\nUSER foo2 bar2 baz2 :Long name2"
 153         for i := 0; i < 6; i++ {
 154                 <-conn.outbound
 155         }
 156
 157         conn.inbound <- "JOIN"
 158         notEnoughParams(t, conn)
 159         conn.inbound <- "JOIN bla/bla/bla"
 160         noChan(t, conn)
 161         conn.inbound <- "JOIN bla:bla:bla"
 162         noChan(t, conn)
 163
 164         conn.inbound <- "JOIN #foo"
 165         if r := <-conn.outbound; r != ":foohost 331 nick2 #foo :No topic is set\r\n" {
 166                 t.Fatal("no topic is set", r)
 167         }
 168         if r := <-conn.outbound; r != ":nick2!foo2@someclient JOIN #foo\r\n" {
 169                 t.Fatal("no JOIN message", r)
 170         }
 171         if r := <-conn.outbound; r != ":foohost 353 nick2 = #foo :nick2\r\n" {
 172                 t.Fatal("no NAMES list", r)
 173         }
 174         if r := <-conn.outbound; r != ":foohost 366 nick2 #foo :End of NAMES list\r\n" {
 175                 t.Fatal("no end of NAMES list", r)
 176         }
 177         if r := <-logSink; (r.what != "joined") || (r.where != "#foo") || (r.who != "nick2") || (r.meta != true) {
 178                 t.Fatal("invalid join log event", r)
 179         }
 180
 181         conn.inbound <- "JOIN #bar,#baz"
 182         for i := 0; i < 4*2; i++ {
 183                 <-conn.outbound
 184         }
 185         if _, ok := rooms["#bar"]; !ok {
 186                 t.Fatal("#bar does not exist")
 187         }
 188         if _, ok := rooms["#baz"]; !ok {
 189                 t.Fatal("#baz does not exist")
 190         }
 191         if r := <-logSink; (r.what != "joined") || (r.where != "#bar") || (r.who != "nick2") || (r.meta != true) {
 192                 t.Fatal("invalid join log event #bar", r)
 193         }
 194         if r := <-logSink; (r.what != "joined") || (r.where != "#baz") || (r.who != "nick2") || (r.meta != true) {
 195                 t.Fatal("invalid join log event #baz", r)
 196         }
 197
 198         conn.inbound <- "JOIN #barenc,#bazenc key1,key2"
 199         for i := 0; i < 4*2; i++ {
 200                 <-conn.outbound
 201         }
 202         if *rooms["#barenc"].key != "key1" {
 203                 t.Fatal("no room with key1")
 204         }
 205         if *rooms["#bazenc"].key != "key2" {
 206                 t.Fatal("no room with key2")
 207         }
 208         if r := <-logSink; (r.what != "joined") || (r.where != "#barenc") || (r.who != "nick2") || (r.meta != true) {
 209                 t.Fatal("invalid join log event #barenc", r)
 210         }
 211         if r := <-logSink; (r.what != "joined") || (r.where != "#bazenc") || (r.who != "nick2") || (r.meta != true) {
 212                 t.Fatal("invalid join log event #bazenc", r)
 213         }
 214         if r := <-stateSink; (r.topic != "") || (r.where != "#barenc") || (r.key != "key1") {
 215                 t.Fatal("set channel key1 state", r)
 216         }
 217         if r := <-stateSink; (r.topic != "") || (r.where != "#bazenc") || (r.key != "key2") {
 218                 t.Fatal("set channel key2 state", r)
 219         }
 220
 221         conn.inbound <- "MODE #barenc -k"
 222         if r := <-conn.outbound; r != ":nick2!foo2@someclient MODE #barenc -k\r\n" {
 223                 t.Fatal("remove #barenc key", r)
 224         }
 225         if *rooms["#barenc"].key != "" {
 226                 t.Fatal("removing key from #barenc")
 227         }
 228         if r := <-logSink; (r.what != "removed channel key") || (r.where != "#barenc") || (r.who != "nick2") || (r.meta != true) {
 229                 t.Fatal("removed channel key log", r)
 230         }
 231         if r := <-stateSink; (r.topic != "") || (r.where != "#barenc") || (r.key != "") {
 232                 t.Fatal("removed channel key state", r)
 233         }
 234
 235         conn.inbound <- "PART #bazenc\r\nMODE #bazenc -k"
 236         if r := <-conn.outbound; r != ":foohost 442 #bazenc :You are not on that channel\r\n" {
 237                 t.Fatal("not on that channel", r)
 238         }
 239         if r := <-logSink; (r.what != "left") || (r.where != "#bazenc") || (r.who != "nick2") || (r.meta != true) {
 240                 t.Fatal("left #bazenc log", r)
 241         }
 242
 243         conn.inbound <- "MODE #barenc +i"
 244         if r := <-conn.outbound; r != ":foohost 472 nick2 +i :Unknown MODE flag\r\n" {
 245                 t.Fatal("unknown MODE flag", r)
 246         }
 247
 248         conn.inbound <- "MODE #barenc +k newkey"
 249         if r := <-conn.outbound; r != ":nick2!foo2@someclient MODE #barenc +k newkey\r\n" {
 250                 t.Fatal("+k MODE setting", r)
 251         }
 252         if r := <-logSink; (r.what != "set channel key to newkey") || (r.where != "#barenc") || (r.who != "nick2") || (r.meta != true) {
 253                 t.Fatal("set channel key", r)
 254         }
 255         if r := <-stateSink; (r.topic != "") || (r.where != "#barenc") || (r.key != "newkey") {
 256                 t.Fatal("set channel newkey state", r)
 257         }
 258
 259         conn.inbound <- "TOPIC #barenc :New topic"
 260         if r := <-conn.outbound; r != ":nick2!foo2@someclient TOPIC #barenc :New topic\r\n" {
 261                 t.Fatal("set TOPIC", r)
 262         }
 263         if r := <-logSink; (r.what != "set topic to New topic") || (r.where != "#barenc") || (r.who != "nick2") || (r.meta != true) {
 264                 t.Fatal("set TOPIC log", r)
 265         }
 266         if r := <-stateSink; (r.topic != "New topic") || (r.where != "#barenc") || (r.key != "newkey") {
 267                 t.Fatal("set channel TOPIC state", r)
 268         }
 269
 270         conn.inbound <- "WHO #barenc"
 271         if r := <-conn.outbound; r != ":foohost 352 nick2 #barenc foo2 someclient foohost nick2 H :0 Long name2\r\n" {
 272                 t.Fatal("WHO", r)
 273         }
 274         if r := <-conn.outbound; r != ":foohost 315 nick2 #barenc :End of /WHO list\r\n" {
 275                 t.Fatal("end of WHO", r)
 276         }
 277 }