forked from mirror/redis
95 lines
1.7 KiB
Go
95 lines
1.7 KiB
Go
package proto_test
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding"
|
|
"io/ioutil"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/go-redis/redis/v7/internal/proto"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
type MyType struct{}
|
|
|
|
var _ encoding.BinaryMarshaler = (*MyType)(nil)
|
|
|
|
func (t *MyType) MarshalBinary() ([]byte, error) {
|
|
return []byte("hello"), nil
|
|
}
|
|
|
|
var _ = Describe("WriteBuffer", func() {
|
|
var buf *bytes.Buffer
|
|
var wr *proto.Writer
|
|
|
|
BeforeEach(func() {
|
|
buf = new(bytes.Buffer)
|
|
wr = proto.NewWriter(buf)
|
|
})
|
|
|
|
It("should write args", func() {
|
|
err := wr.WriteArgs([]interface{}{
|
|
"string",
|
|
12,
|
|
34.56,
|
|
[]byte{'b', 'y', 't', 'e', 's'},
|
|
true,
|
|
nil,
|
|
})
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
err = wr.Flush()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
Expect(buf.Bytes()).To(Equal([]byte("*6\r\n" +
|
|
"$6\r\nstring\r\n" +
|
|
"$2\r\n12\r\n" +
|
|
"$5\r\n34.56\r\n" +
|
|
"$5\r\nbytes\r\n" +
|
|
"$1\r\n1\r\n" +
|
|
"$0\r\n" +
|
|
"\r\n")))
|
|
})
|
|
|
|
It("should append time", func() {
|
|
tm := time.Date(2019, 01, 01, 9, 45, 10, 222125, time.UTC)
|
|
err := wr.WriteArgs([]interface{}{tm})
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
err = wr.Flush()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
Expect(buf.Len()).To(Equal(41))
|
|
})
|
|
|
|
It("should append marshalable args", func() {
|
|
err := wr.WriteArgs([]interface{}{&MyType{}})
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
err = wr.Flush()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
Expect(buf.Len()).To(Equal(15))
|
|
})
|
|
})
|
|
|
|
func BenchmarkWriteBuffer_Append(b *testing.B) {
|
|
buf := proto.NewWriter(ioutil.Discard)
|
|
args := []interface{}{"hello", "world", "foo", "bar"}
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
err := buf.WriteArgs(args)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
err = buf.Flush()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|