room_test.go

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