room_test.go

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