add more tests for zrpc (#321)

zrpc/internal/rpclogger.go

@@ -64,10 +64,10 @@ func (l *Logger) Warning(args ...interface{}) {
 	// ignore builtin grpc warning
 }
 
-func (l *Logger) Warningln(args ...interface{}) {
+func (l *Logger) Warningf(format string, args ...interface{}) {
 	// ignore builtin grpc warning
 }
 
-func (l *Logger) Warningf(format string, args ...interface{}) {
+func (l *Logger) Warningln(args ...interface{}) {
 	// ignore builtin grpc warning
 }

zrpc/internal/rpclogger_test.go

@@ -0,0 +1,83 @@
+package internal
+
+import (
+	"log"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+const content = "foo"
+
+func TestLoggerError(t *testing.T) {
+	var builder strings.Builder
+	log.SetOutput(&builder)
+	logger := new(Logger)
+	logger.Error(content)
+	assert.Contains(t, builder.String(), content)
+}
+
+func TestLoggerErrorf(t *testing.T) {
+	var builder strings.Builder
+	log.SetOutput(&builder)
+	logger := new(Logger)
+	logger.Errorf(content)
+	assert.Contains(t, builder.String(), content)
+}
+
+func TestLoggerErrorln(t *testing.T) {
+	var builder strings.Builder
+	log.SetOutput(&builder)
+	logger := new(Logger)
+	logger.Errorln(content)
+	assert.Contains(t, builder.String(), content)
+}
+
+func TestLoggerFatal(t *testing.T) {
+	var builder strings.Builder
+	log.SetOutput(&builder)
+	logger := new(Logger)
+	logger.Fatal(content)
+	assert.Contains(t, builder.String(), content)
+}
+
+func TestLoggerFatalf(t *testing.T) {
+	var builder strings.Builder
+	log.SetOutput(&builder)
+	logger := new(Logger)
+	logger.Fatalf(content)
+	assert.Contains(t, builder.String(), content)
+}
+
+func TestLoggerFatalln(t *testing.T) {
+	var builder strings.Builder
+	log.SetOutput(&builder)
+	logger := new(Logger)
+	logger.Fatalln(content)
+	assert.Contains(t, builder.String(), content)
+}
+
+func TestLoggerWarning(t *testing.T) {
+	var builder strings.Builder
+	log.SetOutput(&builder)
+	logger := new(Logger)
+	logger.Warning(content)
+	assert.Empty(t, builder.String())
+}
+
+func TestLoggerWarningf(t *testing.T) {
+	var builder strings.Builder
+	log.SetOutput(&builder)
+	logger := new(Logger)
+	logger.Warningf(content)
+	assert.Empty(t, builder.String())
+}
+
+func TestLoggerWarningln(t *testing.T) {
+	var builder strings.Builder
+	log.SetOutput(&builder)
+	logger := new(Logger)
+	logger.Warningln(content)
+	assert.Empty(t, builder.String())
+}