浏览代码

chore: add more tests (#2547)

Kevin Wan 2 年之前
父节点
当前提交
46e236fef7

+ 15 - 0
core/fs/files_test.go

@@ -0,0 +1,15 @@
+package fs
+
+import (
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCloseOnExec(t *testing.T) {
+	file := os.NewFile(0, os.DevNull)
+	assert.NotPanics(t, func() {
+		CloseOnExec(file)
+	})
+}

+ 4 - 0
core/stat/alert_test.go

@@ -4,6 +4,7 @@
 package stat
 
 import (
+	"os"
 	"strconv"
 	"sync/atomic"
 	"testing"
@@ -12,6 +13,9 @@ import (
 )
 
 func TestReport(t *testing.T) {
+	os.Setenv(clusterNameKey, "test-cluster")
+	defer os.Unsetenv(clusterNameKey)
+
 	var count int32
 	SetReporter(func(s string) {
 		atomic.AddInt32(&count, 1)

+ 9 - 1
zrpc/resolver/internal/directbuilder_test.go

@@ -1,6 +1,7 @@
 package internal
 
 import (
+	"errors"
 	"fmt"
 	"net/url"
 	"strconv"
@@ -35,10 +36,17 @@ func TestDirectBuilder_Build(t *testing.T) {
 			target := fmt.Sprintf("%s:///%s", DirectScheme, strings.Join(servers, ","))
 			uri, err := url.Parse(target)
 			assert.Nil(t, err)
+			cc.err = errors.New("foo")
 			_, err = b.Build(resolver.Target{
 				URL: *uri,
 			}, cc, resolver.BuildOptions{})
-			assert.Nil(t, err)
+			assert.NotNil(t, err)
+			cc.err = nil
+			_, err = b.Build(resolver.Target{
+				URL: *uri,
+			}, cc, resolver.BuildOptions{})
+			assert.NoError(t, err)
+
 			size := mathx.MinInt(test, subsetSize)
 			assert.Equal(t, size, len(cc.state.Addresses))
 			m := make(map[string]lang.PlaceholderType)

+ 2 - 1
zrpc/resolver/internal/resolver_test.go

@@ -16,11 +16,12 @@ func TestNopResolver(t *testing.T) {
 
 type mockedClientConn struct {
 	state resolver.State
+	err   error
 }
 
 func (m *mockedClientConn) UpdateState(state resolver.State) error {
 	m.state = state
-	return nil
+	return m.err
 }
 
 func (m *mockedClientConn) ReportError(err error) {