98 lines
1.8 KiB
Go
98 lines
1.8 KiB
Go
package irc
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestParseMessage(t *testing.T) {
|
|
cases := []struct {
|
|
input string
|
|
expected *Message
|
|
}{
|
|
{
|
|
":user CMD #chan :some message\r\n",
|
|
&Message{
|
|
Prefix: "user",
|
|
Nick: "user",
|
|
Command: "CMD",
|
|
Params: []string{"#chan", "some message"},
|
|
},
|
|
}, {
|
|
":nick!user@host.com CMD a b\r\n",
|
|
&Message{
|
|
Prefix: "nick!user@host.com",
|
|
Nick: "nick",
|
|
Command: "CMD",
|
|
Params: []string{"a", "b"},
|
|
},
|
|
}, {
|
|
"CMD a b :\r\n",
|
|
&Message{
|
|
Command: "CMD",
|
|
Params: []string{"a", "b", ""},
|
|
},
|
|
}, {
|
|
"CMD a b\r\n",
|
|
&Message{
|
|
Command: "CMD",
|
|
Params: []string{"a", "b"},
|
|
},
|
|
}, {
|
|
"CMD\r\n",
|
|
&Message{
|
|
Command: "CMD",
|
|
},
|
|
}, {
|
|
"CMD :tests and stuff\r\n",
|
|
&Message{
|
|
Command: "CMD",
|
|
Params: []string{"tests and stuff"},
|
|
},
|
|
}, {
|
|
":nick@host.com CMD\r\n",
|
|
&Message{
|
|
Prefix: "nick@host.com",
|
|
Nick: "nick",
|
|
Command: "CMD",
|
|
},
|
|
}, {
|
|
":ni@ck!user!name@host!.com CMD\r\n",
|
|
&Message{
|
|
Prefix: "ni@ck!user!name@host!.com",
|
|
Nick: "ni@ck",
|
|
Command: "CMD",
|
|
},
|
|
}, {
|
|
"CMD #cake pie \r\n",
|
|
&Message{
|
|
Command: "CMD",
|
|
Params: []string{"#cake", "pie"},
|
|
},
|
|
}, {
|
|
" CMD #cake pie\r\n",
|
|
&Message{
|
|
Command: "CMD",
|
|
Params: []string{"#cake", "pie"},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
assert.Equal(t, tc.expected, parseMessage(tc.input))
|
|
}
|
|
}
|
|
|
|
func TestLastParam(t *testing.T) {
|
|
assert.Equal(t, "some message", parseMessage(":user CMD #chan :some message\r\n").LastParam())
|
|
assert.Equal(t, "", parseMessage("NO_PARAMS").LastParam())
|
|
}
|
|
|
|
func TestBadMessagePanic(t *testing.T) {
|
|
parseMessage(":user\r\n")
|
|
parseMessage(":\r\n")
|
|
parseMessage(":")
|
|
parseMessage("")
|
|
}
|