Refactor irc handler, add tests

server/irc_handler_test.go

@@ -0,0 +1,266 @@
+package server
+
+import (
+	"io/ioutil"
+	"log"
+	"os"
+	"path"
+	"testing"
+
+	"github.com/khlieng/name_pending/Godeps/_workspace/src/github.com/stretchr/testify/assert"
+
+	"github.com/khlieng/name_pending/irc"
+	"github.com/khlieng/name_pending/storage"
+)
+
+var user *storage.User
+
+func TestMain(m *testing.M) {
+	tempdir, err := ioutil.TempDir("", "test_")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	os.Mkdir(path.Join(tempdir, "logs"), 0777)
+	storage.Initialize(tempdir)
+	user = storage.NewUser("uuid")
+	channelStore = storage.NewChannelStore()
+
+	code := m.Run()
+
+	os.RemoveAll(tempdir)
+	os.Exit(code)
+}
+
+func dispatchMessage(msg *irc.Message) WSResponse {
+	c := irc.NewClient("nick", "user")
+	c.Host = "host.com"
+	s := NewSession()
+	s.user = user
+
+	newIRCHandler(c, s).dispatchMessage(msg)
+
+	return <-s.out
+}
+
+func checkResponse(t *testing.T, expectedType string, expectedResponse interface{}, res WSResponse) {
+	assert.Equal(t, expectedType, res.Type)
+	assert.Equal(t, expectedResponse, res.Response)
+}
+
+func TestHandleIRCNick(t *testing.T) {
+	res := dispatchMessage(&irc.Message{
+		Command:  irc.Nick,
+		Nick:     "old",
+		Trailing: "new",
+	})
+
+	checkResponse(t, "nick", Nick{
+		Server: "host.com",
+		Old:    "old",
+		New:    "new",
+	}, res)
+}
+
+func TestHandleIRCJoin(t *testing.T) {
+	res := dispatchMessage(&irc.Message{
+		Command: irc.Join,
+		Nick:    "joining",
+		Params:  []string{"#chan"},
+	})
+
+	checkResponse(t, "join", Join{
+		Server:   "host.com",
+		User:     "joining",
+		Channels: []string{"#chan"},
+	}, res)
+}
+
+func TestHandleIRCPart(t *testing.T) {
+	res := dispatchMessage(&irc.Message{
+		Command:  irc.Part,
+		Nick:     "parting",
+		Params:   []string{"#chan"},
+		Trailing: "the reason",
+	})
+
+	checkResponse(t, "part", Part{
+		Join: Join{
+			Server:   "host.com",
+			User:     "parting",
+			Channels: []string{"#chan"},
+		},
+		Reason: "the reason",
+	}, res)
+}
+
+func TestHandleIRCMode(t *testing.T) {
+	res := dispatchMessage(&irc.Message{
+		Command: irc.Mode,
+		Params:  []string{"#chan", "+o-v", "nick"},
+	})
+
+	checkResponse(t, "mode", &Mode{
+		Server:  "host.com",
+		Channel: "#chan",
+		User:    "nick",
+		Add:     "o",
+		Remove:  "v",
+	}, res)
+}
+
+func TestHandleIRCMessage(t *testing.T) {
+	res := dispatchMessage(&irc.Message{
+		Command:  irc.Privmsg,
+		Nick:     "nick",
+		Params:   []string{"#chan"},
+		Trailing: "the message",
+	})
+
+	checkResponse(t, "message", Chat{
+		Server:  "host.com",
+		From:    "nick",
+		To:      "#chan",
+		Message: "the message",
+	}, res)
+
+	res = dispatchMessage(&irc.Message{
+		Command:  irc.Privmsg,
+		Nick:     "someone",
+		Params:   []string{"nick"},
+		Trailing: "the message",
+	})
+
+	checkResponse(t, "pm", Chat{
+		Server:  "host.com",
+		From:    "someone",
+		Message: "the message",
+	}, res)
+}
+
+func TestHandleIRCQuit(t *testing.T) {
+	res := dispatchMessage(&irc.Message{
+		Command:  irc.Quit,
+		Nick:     "nick",
+		Trailing: "the reason",
+	})
+
+	checkResponse(t, "quit", Quit{
+		Server: "host.com",
+		User:   "nick",
+		Reason: "the reason",
+	}, res)
+}
+
+func TestHandleIRCWelcome(t *testing.T) {
+	res := dispatchMessage(&irc.Message{
+		Command: irc.ReplyWelcome,
+		Nick:    "nick",
+		Params:  []string{"target", "some", "text"},
+	})
+
+	checkResponse(t, "pm", Chat{
+		Server:  "host.com",
+		From:    "nick",
+		Message: "some text",
+	}, res)
+}
+
+func TestHandleIRCWhois(t *testing.T) {
+	c := irc.NewClient("nick", "user")
+	c.Host = "host.com"
+	s := NewSession()
+	i := newIRCHandler(c, s)
+
+	i.dispatchMessage(&irc.Message{
+		Command: irc.ReplyWhoisUser,
+		Params:  []string{"", "nick", "user", "host", "", "realname"},
+	})
+	i.dispatchMessage(&irc.Message{
+		Command: irc.ReplyWhoisServer,
+		Params:  []string{"", "", "srv.com"},
+	})
+	i.dispatchMessage(&irc.Message{
+		Command:  irc.ReplyWhoisChannels,
+		Trailing: "#chan #chan1",
+	})
+	i.dispatchMessage(&irc.Message{Command: irc.ReplyEndOfWhois})
+
+	checkResponse(t, "whois", WhoisReply{
+		Nick:     "nick",
+		Username: "user",
+		Host:     "host",
+		Realname: "realname",
+		Server:   "srv.com",
+		Channels: []string{"#chan", "#chan1"},
+	}, <-s.out)
+}
+
+func TestHandleIRCTopic(t *testing.T) {
+	res := dispatchMessage(&irc.Message{
+		Command:  irc.ReplyTopic,
+		Params:   []string{"target", "#chan"},
+		Trailing: "the topic",
+	})
+
+	checkResponse(t, "topic", Topic{
+		Server:  "host.com",
+		Channel: "#chan",
+		Topic:   "the topic",
+	}, res)
+}
+
+func TestHandleIRCNames(t *testing.T) {
+	c := irc.NewClient("nick", "user")
+	c.Host = "host.com"
+	s := NewSession()
+	i := newIRCHandler(c, s)
+
+	i.dispatchMessage(&irc.Message{
+		Command:  irc.ReplyNamReply,
+		Params:   []string{"", "", "#chan"},
+		Trailing: "a b c",
+	})
+	i.dispatchMessage(&irc.Message{
+		Command:  irc.ReplyNamReply,
+		Params:   []string{"", "", "#chan"},
+		Trailing: "d",
+	})
+	i.dispatchMessage(&irc.Message{
+		Command: irc.ReplyEndOfNames,
+		Params:  []string{"", "#chan"},
+	})
+
+	checkResponse(t, "users", Userlist{
+		Server:  "host.com",
+		Channel: "#chan",
+		Users:   []string{"a", "b", "c", "d"},
+	}, <-s.out)
+}
+
+func TestHandleIRCMotd(t *testing.T) {
+	c := irc.NewClient("nick", "user")
+	c.Host = "host.com"
+	s := NewSession()
+	i := newIRCHandler(c, s)
+
+	i.dispatchMessage(&irc.Message{
+		Command:  irc.ReplyMotdStart,
+		Trailing: "motd title",
+	})
+	i.dispatchMessage(&irc.Message{
+		Command:  irc.ReplyMotd,
+		Trailing: "line 1",
+	})
+	i.dispatchMessage(&irc.Message{
+		Command:  irc.ReplyMotd,
+		Trailing: "line 2",
+	})
+	i.dispatchMessage(&irc.Message{Command: irc.ReplyEndOfMotd})
+
+	checkResponse(t, "motd", MOTD{
+		Server:  "host.com",
+		Title:   "motd title",
+		Content: []string{"line 1", "line 2"},
+	}, <-s.out)
+}