room_test.go

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