room_test.go

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