Fixing unit tests

Author: kenshaw

client_test.go

@@ -18,8 +18,8 @@ func BenchmarkRandKey(b *testing.B) {
 }
 
 func TestDial(t *testing.T) {
-	var text = []byte("Make fasthttp great again")
-	var uri = "http://localhost:9843/"
+	text := []byte("Make fasthttp great again")
+	uri := "http://localhost:9843/"
 	ln := fasthttputil.NewInmemoryListener()
 
 	ws := Server{
@@ -33,11 +33,7 @@ func TestDial(t *testing.T) {
 	})
 
 	ws.HandleClose(func(c *Conn, err error) {
-		if err == nil {
-			t.Fatal("Expected err")
-		}
-
-		if err.(Error).Status != StatusGoAway {
+		if err != nil && err.(Error).Status != StatusGoAway {
 			t.Fatalf("Expected GoAway, got %s", err.(Error).Status)
 		}
 	})

conn_test.go

@@ -3,18 +3,19 @@ package websocket
 import (
 	"bufio"
 	"fmt"
-	"github.com/valyala/fasthttp"
-	"github.com/valyala/fasthttp/fasthttputil"
 	"io"
 	"testing"
+
+	"github.com/valyala/fasthttp"
+	"github.com/valyala/fasthttp/fasthttputil"
 )
 
 func configureServer(t *testing.T) (*fasthttp.Server, *fasthttputil.InmemoryListener) {
 	ln := fasthttputil.NewInmemoryListener()
 
 	ws := Server{}
 
-	var stage = 0
+	stage := 0
 
 	ws.HandleData(func(c *Conn, isBinary bool, data []byte) {
 		switch stage {
@@ -154,8 +155,8 @@ func TestReadFrame(t *testing.T) {
 
 	fr.Reset()
 	_, err = conn.ReadFrame(fr)
-	if !fr.IsClose() {
-		t.Fatal("Unexpected frame close")
+	if !fr.IsContinuation() {
+		t.Fatalf("Unexpected frame %s", fr.Code())
 	}
 
 	ln.Close()