add unit tests

lixizan · lixizan · commit febeaa304c43 · 2023-01-17T10:45:51.000+08:00
diff --git a/Makefile b/Makefile
@@ -11,5 +11,4 @@ run-testsuite-server:
 	go run github.com/lxzan/gws/examples/testsuite
 
 coverage:
-	go test -mod=readonly -covermode=count -coverprofile=./bin/coverprofile.cov -run="^Test" -coverpkg=$(go list -mod=vendor ./... | grep -v "/test" | tr '\n' ',') ./...
-
+	go test ./... -covermode=count -coverprofile=./bin/gws.out
diff --git a/internal/error.go b/internal/error.go
@@ -79,7 +79,7 @@ func (c StatusCode) Uint16() uint16 {
 
 func (c StatusCode) Bytes() []byte {
 	if c == 0 {
-		return []byte{}
+		return []byte{0, 0}
 	}
 	return []byte{uint8(c >> 8), uint8(c << 8 >> 8)}
 }
diff --git a/internal/internal_test.go b/internal/internal_test.go
@@ -0,0 +1,38 @@
+package internal
+
+import (
+	"bytes"
+	"encoding/hex"
+	"github.com/stretchr/testify/assert"
+	"io"
+	"testing"
+)
+
+func TestError(t *testing.T) {
+	var as = assert.New(t)
+	t.Run("", func(t *testing.T) {
+		var code = StatusCode(1000)
+		as.Equal(uint16(1000), code.Uint16())
+		as.Equal(hex.EncodeToString(code.Bytes()), "03e8")
+		as.Equal(code.Error() != "", true)
+	})
+
+	t.Run("", func(t *testing.T) {
+		var code = StatusCode(0)
+		as.Equal(hex.EncodeToString(code.Bytes()), "0000")
+	})
+
+	t.Run("", func(t *testing.T) {
+		var err error = NewError(CloseGoingAway, io.EOF)
+		as.Equal(err.Error() != "", true)
+	})
+}
+
+func TestBuffer_ReadFrom(t *testing.T) {
+	var b = Buffer{Buffer: bytes.NewBuffer(nil)}
+	b.ReadFrom()
+}
+
+func TestRandomString_Uint32(t *testing.T) {
+	Numeric.Uint32()
+}
diff --git a/internal/pool_test.go b/internal/pool_test.go
@@ -33,4 +33,8 @@ func TestBufferPool(t *testing.T) {
 		var buf = pool.Get(5000)
 		as.Equal(Lv4, buf.Cap())
 	}
+
+	pool.Put(nil)
+	pool.Put(NewBufferWithCap(0))
+	pool.Get(17 * 1024)
 }
diff --git a/internal/utils_test.go b/internal/utils_test.go
@@ -2,6 +2,7 @@ package internal
 
 import (
 	"bytes"
+	"encoding/hex"
 	"encoding/json"
 	"fmt"
 	"github.com/stretchr/testify/assert"
@@ -17,6 +18,18 @@ func TestNewMaskKey(t *testing.T) {
 	assert.Equal(t, 4, len(key))
 }
 
+func TestMaskByByte(t *testing.T) {
+	var data = []byte("hello")
+	MaskByByte(data, []byte{0xa, 0xb, 0xc, 0xd})
+	assert.Equal(t, "626e606165", hex.EncodeToString(data))
+}
+
+func TestStringToBytes(t *testing.T) {
+	var s1 = string(AlphabetNumeric.Generate(32))
+	var s2 = string(StringToBytes(s1))
+	assert.Equal(t, s1, s2)
+}
+
 func TestComputeAcceptKey(t *testing.T) {
 	var s = ComputeAcceptKey("PUurdSuLQj/6n4NFf/rA7A==")
 	assert.Equal(t, "HmIbwxkcLxq+A+3qnlBVtT7Bjgg=", s)
@@ -92,3 +105,55 @@ func TestFNV64(t *testing.T) {
 	_, _ = h.Write(s)
 	assert.Equal(t, h.Sum64(), FNV64(string(s)))
 }
+
+func TestIOUtil(t *testing.T) {
+	var as = assert.New(t)
+
+	t.Run("", func(t *testing.T) {
+		var dst = bytes.NewBuffer(nil)
+		var src = bytes.NewBuffer(make([]byte, 0))
+		var err = CopyN(dst, src, 0)
+		as.NoError(err)
+	})
+
+	t.Run("", func(t *testing.T) {
+		var dst = bytes.NewBuffer(nil)
+		var src = bytes.NewBuffer(make([]byte, 6))
+		var err = CopyN(dst, src, 12)
+		as.Error(err)
+	})
+
+	t.Run("", func(t *testing.T) {
+		var reader = strings.NewReader("hello")
+		var p = make([]byte, 0)
+		var err = ReadN(reader, p, 0)
+		as.NoError(err)
+	})
+
+	t.Run("", func(t *testing.T) {
+		var reader = strings.NewReader("hello")
+		var p = make([]byte, 5)
+		var err = ReadN(reader, p, 10)
+		as.Error(err)
+	})
+
+	t.Run("", func(t *testing.T) {
+		var writer = bytes.NewBufferString("")
+		var err = WriteN(writer, nil, 0)
+		as.NoError(err)
+	})
+
+	t.Run("", func(t *testing.T) {
+		var writer = bytes.NewBufferString("")
+		var p = []byte("hello")
+		var err = WriteN(writer, p, 5)
+		as.NoError(err)
+	})
+
+	t.Run("", func(t *testing.T) {
+		var buf1 = NewBufferWithCap(0)
+		as.Equal(0, buf1.Cap())
+		var buf2 = NewBufferWithCap(12)
+		as.Equal(12, buf2.Cap())
+	})
+}

Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,7 @@ func (c StatusCode) Uint16() uint16 {`
`79`	`79`
`80`	`80`	`func (c StatusCode) Bytes() []byte {`
`81`	`81`	`if c == 0 {`
`82`		`- return []byte{}`
	`82`	`+ return []byte{0, 0}`
`83`	`83`	`}`
`84`	`84`	`return []byte{uint8(c >> 8), uint8(c << 8 >> 8)}`
`85`	`85`	`}`
Original file line number	Diff line number	Diff line change
`@@ -33,4 +33,8 @@ func TestBufferPool(t *testing.T) {`
`33`	`33`	`var buf = pool.Get(5000)`
`34`	`34`	`as.Equal(Lv4, buf.Cap())`
`35`	`35`	`}`
	`36`	`+`
	`37`	`+ pool.Put(nil)`
	`38`	`+ pool.Put(NewBufferWithCap(0))`
	`39`	`+ pool.Get(17 * 1024)`
`36`	`40`	`}`