room_test.go

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