fix gracefulShutdownConfig unittest bug

config/graceful_shutdown_test.go

@@ -21,13 +21,24 @@ import (
 	"testing"
 )
 
+import (
+	"go.uber.org/atomic"
+)
+
 import (
 	"dubbo.apache.org/dubbo-go/v3/common/constant"
 	"dubbo.apache.org/dubbo-go/v3/common/extension"
+	"dubbo.apache.org/dubbo-go/v3/filter"
 	"dubbo.apache.org/dubbo-go/v3/protocol"
 )
 
 func TestGracefulShutdownInit(t *testing.T) {
+	extension.SetFilter(constant.CONSUMER_SHUTDOWN_FILTER, func() filter.Filter {
+		return &mockGracefulShutdownFilter{}
+	})
+	extension.SetFilter(constant.PROVIDER_SHUTDOWN_FILTER, func() filter.Filter {
+		return &mockGracefulShutdownFilter{}
+	})
 	GracefulShutdownInit()
 }
 
@@ -49,13 +60,16 @@ func TestBeforeShutdown(t *testing.T) {
 	}
 
 	// without configuration
+	consumerConfig = nil
+	providerConfig = nil
 	BeforeShutdown()
 
 	consumerConfig = &ConsumerConfig{
 		References: consumerReferences,
 		ShutdownConfig: &ShutdownConfig{
 			Timeout:     "1",
 			StepTimeout: "1s",
+			RequestsFinished: &atomic.Bool{},
 		},
 	}
 
@@ -72,6 +86,7 @@ func TestBeforeShutdown(t *testing.T) {
 		ShutdownConfig: &ShutdownConfig{
 			Timeout:     "1",
 			StepTimeout: "1s",
+			RequestsFinished: &atomic.Bool{},
 		},
 		Protocols: providerProtocols,
 	}
@@ -82,6 +97,7 @@ func TestBeforeShutdown(t *testing.T) {
 		ShutdownConfig: &ShutdownConfig{
 			Timeout:     "1",
 			StepTimeout: "-1s",
+			RequestsFinished: &atomic.Bool{},
 		},
 		Protocols: providerProtocols,
 	}
@@ -91,6 +107,7 @@ func TestBeforeShutdown(t *testing.T) {
 		ShutdownConfig: &ShutdownConfig{
 			Timeout:     "1",
 			StepTimeout: "-1s",
+			RequestsFinished: &atomic.Bool{},
 		},
 	}