55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package socketio
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
)
|
|
|
|
func TestMessageReader(t *testing.T) {
|
|
|
|
Convey("Read with args", t, func() {
|
|
buf := bufio.NewReader(bytes.NewBufferString(`["message",1]`))
|
|
reader, err := newMessageReader(buf)
|
|
So(err, ShouldBeNil)
|
|
So(reader.Message(), ShouldEqual, "message")
|
|
b, err := ioutil.ReadAll(reader)
|
|
So(err, ShouldBeNil)
|
|
So(string(b), ShouldEqual, "[1]")
|
|
})
|
|
|
|
Convey("Read with args, space", t, func() {
|
|
buf := bufio.NewReader(bytes.NewBufferString(`["message" , 1]`))
|
|
reader, err := newMessageReader(buf)
|
|
So(err, ShouldBeNil)
|
|
So(reader.Message(), ShouldEqual, "message")
|
|
b, err := ioutil.ReadAll(reader)
|
|
So(err, ShouldBeNil)
|
|
So(string(b), ShouldEqual, "[ 1]")
|
|
})
|
|
|
|
Convey("Read only message", t, func() {
|
|
buf := bufio.NewReader(bytes.NewBufferString(`["message"]`))
|
|
reader, err := newMessageReader(buf)
|
|
So(err, ShouldBeNil)
|
|
So(reader.Message(), ShouldEqual, "message")
|
|
b, err := ioutil.ReadAll(reader)
|
|
So(err, ShouldBeNil)
|
|
So(string(b), ShouldEqual, "[]")
|
|
})
|
|
|
|
Convey("Read only message", t, func() {
|
|
buf := bufio.NewReader(bytes.NewBufferString(`["message" ]`))
|
|
reader, err := newMessageReader(buf)
|
|
So(err, ShouldBeNil)
|
|
So(reader.Message(), ShouldEqual, "message")
|
|
b, err := ioutil.ReadAll(reader)
|
|
So(err, ShouldBeNil)
|
|
So(string(b), ShouldEqual, "[]")
|
|
})
|
|
|
|
}
|