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, "[]")
|
||
|
})
|
||
|
|
||
|
}
|