NewMockWriter return *MockWriter -> MockWriter

pkg/bele/bele_test.go

@@ -229,4 +229,3 @@ func BenchmarkWriteBE(b *testing.B) {
 		_ = WriteBE(out, in)
 	}
 }
-

pkg/log/log_test.go

@@ -2,9 +2,9 @@ package log
 
 import (
 	"github.com/q191201771/nezha/pkg/assert"
+	originLog "log"
 	"os"
 	"testing"
-	originLog "log"
 )
 
 func TestLogger(t *testing.T) {

pkg/mockwriter/mockwriter.go

@@ -26,8 +26,8 @@ type MockWriter struct {
 	b     bytes.Buffer
 }
 
-func NewMockWriter(t WriterType) *MockWriter {
+func NewMockWriter(t WriterType) MockWriter {
-	return &MockWriter{
+	return MockWriter{
 		t: t,
 	}
 }
@@ -53,6 +53,5 @@ func (w *MockWriter) Write(b []byte) (int, error) {
 	}
 
 	return w.b.Write(b)
-
 	//panic("never reach here.")
 }

pkg/mockwriter/mockwriter_test.go

@@ -11,7 +11,7 @@ func TestNewMockWriter(t *testing.T) {
 
 func TestMockWriter_Write(t *testing.T) {
 	var (
-		w *MockWriter
+		w   MockWriter
 		n   int
 		err error
 		b   = []byte("hello")
@@ -35,7 +35,7 @@ func TestMockWriter_Write(t *testing.T) {
 
 func TestMockWriter_SetSpecificType(t *testing.T) {
 	var (
-		w *MockWriter
+		w   MockWriter
 		n   int
 		err error
 		b   = []byte("hello")
@@ -70,3 +70,27 @@ func TestMockWriter_SetSpecificType(t *testing.T) {
 		assert.Equal(t, expectedErr[i], err)
 	}
 }
+
+func BenchmarkNewMockWriter(b *testing.B) {
+	b.ReportAllocs()
+	var tmp uint32
+	for i := 0; i < b.N; i++ {
+		mw := NewMockWriter(WriterTypeDoNothing)
+		tmp = tmp + mw.count
+	}
+}
+
+func newMockWriter2(t WriterType) *MockWriter {
+	return &MockWriter{
+		t: t,
+	}
+}
+
+func BenchmarkNewMockWriter2(b *testing.B) {
+	b.ReportAllocs()
+	var tmp uint32
+	for i := 0; i < b.N; i++ {
+		mw := newMockWriter2(WriterTypeDoNothing)
+		tmp = tmp + mw.count
+	}
+}