unit test

Author: ChrisSchinnerl

rhp/v4/rpc_test.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"cmp"
 	"context"
+	"errors"
 	"crypto/tls"
 	"fmt"
 	"maps"
@@ -2952,3 +2953,29 @@ func BenchmarkContractUpload(b *testing.B) {
 		b.Fatalf("expected %v sectors, got %v", b.N, appendResult.Revision.Filesize/proto4.SectorSize)
 	}
 }
+
+func TestGracefulShutdown(t *testing.T) {
+	n, genesis := testutil.V2Network()
+	cm, w := startTestNode(t, n, genesis)
+
+	ss := testutil.NewEphemeralSectorStore()
+	c := testutil.NewEphemeralContractor(cm)
+	sr := testutil.NewEphemeralSettingsReporter()
+
+	hostKey := types.GeneratePrivateKey()
+	rs := rhp4.NewServer(hostKey, cm, c, w, sr, ss, rhp4.WithPriceTableValidity(2*time.Minute))
+	hostAddr := testutil.ServeSiaMux(t, rs, zap.NewNop())
+
+	transport, err := siamux.Dial(context.Background(), hostAddr, hostKey.PublicKey())
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer transport.Close()
+
+	rs.Close()
+
+	_, err = rhp4.RPCSettings(context.Background(), transport)
+	if !errors.Is(err, proto4.ErrHostShuttingDown) {
+		t.Fatalf("expected ErrHostShuttingDown, got %v", err)
+	}
+}