更新 nezha 版本，使用其中的 MockWriter

go.mod

@@ -2,4 +2,4 @@ module github.com/q191201771/lal
 
 go 1.12
 
-require github.com/q191201771/nezha v0.0.2-0.20190904074707-20fd1445a085
+require github.com/q191201771/nezha v0.0.2-0.20190909120248-23806813af03

go.sum

@@ -1,2 +1,2 @@
-github.com/q191201771/nezha v0.0.2-0.20190904074707-20fd1445a085 h1:S6/SulRptcCXgV6hbbWGgqiKmMLf7iXHjT6Qomj5qdk=
-github.com/q191201771/nezha v0.0.2-0.20190904074707-20fd1445a085/go.mod h1:Rd4R+bJRemlSUnz7KHmSX6ZQlsHLBjT7wlzuLeOia/M=
+github.com/q191201771/nezha v0.0.2-0.20190909120248-23806813af03 h1:or7KshWRYtGDSLeFnkitJrnGqc1mt7CRHFG8uk0X4gs=
+github.com/q191201771/nezha v0.0.2-0.20190909120248-23806813af03/go.mod h1:Rd4R+bJRemlSUnz7KHmSX6ZQlsHLBjT7wlzuLeOia/M=

pkg/rtmp/pack_test.go

@@ -2,18 +2,11 @@ package rtmp
 
 import (
 	"bytes"
-	"errors"
 	"github.com/q191201771/nezha/pkg/assert"
+	"github.com/q191201771/nezha/pkg/mockwriter"
 	"testing"
 )
 
-type MockWriter struct {
-}
-
-func (w *MockWriter) Write(p []byte) (n int, err error) {
-	return 0, errors.New("mock error")
-}
-
 func TestWriteMessageHandler(t *testing.T) {
 	//buf := &bytes.Buffer{}
 	packer := NewMessagePacker()
@@ -126,7 +119,7 @@ func TestCorner(t *testing.T) {
 	}()
 
 	var err error
-	mw := &MockWriter{}
+	mw := mockwriter.NewMockWriter(mockwriter.WriterTypeReturnError)
 	packer := NewMessagePacker()
 
 	err = packer.writeProtocolControlMessage(mw, 1, 2)

pkg/rtmp/stream.go

@@ -2,7 +2,7 @@ package rtmp
 
 import "github.com/q191201771/nezha/pkg/log"
 
-var initMsgLen = 4096
+const initMsgLen = 4096
 
 type Header struct {
 	CSID   int