Skip to content

Commit f582b95

Browse files
committed
pr fixes
1 parent f83e217 commit f582b95

File tree

3 files changed

+9
-10
lines changed

3 files changed

+9
-10
lines changed

http/handler.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -993,7 +993,7 @@ func forwardRequest(core *vault.Core, w http.ResponseWriter, r *http.Request) {
993993
core.Logger().Trace("cannot forward request (possibly disabled on active node), falling back to redirection to standby")
994994
case errors.Is(err, vault.StatusNotHAMember):
995995
core.Logger().Trace("this node is not a member of the HA cluster", "error", err)
996-
respondError(w, http.StatusServiceUnavailable, err)
996+
respondError(w, http.StatusInternalServerError, err)
997997
return
998998
default:
999999
core.Logger().Error("forward request error", "error", err)

vault/request_forwarding.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ func haMembershipServerCheck(ctx context.Context, c *Core, haBackend physical.Re
6666
removed, err := haBackend.IsNodeRemoved(ctx, nodeID)
6767
if err != nil {
6868
c.logger.Error("failed to check if node is removed", "error", err)
69-
return nil
69+
return err
7070
}
7171
if removed {
7272
return StatusNotHAMember

vault/request_forwarding_test.go

+7-8
Original file line numberDiff line numberDiff line change
@@ -82,38 +82,37 @@ func newMockHARemovableNodeBackend(isRemoved func(context.Context, string) (bool
8282
// when the context contains a removed node ID
8383
func Test_haMembershipServerCheck(t *testing.T) {
8484
nodeIDCtx := metadata.NewIncomingContext(context.Background(), metadata.MD{haNodeIDKey: {"node_id"}})
85+
otherErr := errors.New("error checking")
8586
testCases := []struct {
8687
name string
8788
nodeIDCtx context.Context
8889
haBackend physical.RemovableNodeHABackend
89-
wantError bool
90+
wantError error
9091
}{
9192
{
9293
name: "nil backend",
9394
haBackend: nil,
9495
nodeIDCtx: nodeIDCtx,
95-
wantError: false,
9696
}, {
9797
name: "no node ID context",
9898
haBackend: newMockHARemovableNodeBackend(func(ctx context.Context, s string) (bool, error) {
9999
return false, nil
100100
}),
101101
nodeIDCtx: context.Background(),
102-
wantError: false,
103102
}, {
104103
name: "node removed",
105104
haBackend: newMockHARemovableNodeBackend(func(ctx context.Context, s string) (bool, error) {
106105
return true, nil
107106
}),
108107
nodeIDCtx: nodeIDCtx,
109-
wantError: true,
108+
wantError: StatusNotHAMember,
110109
}, {
111110
name: "node removed err",
112111
haBackend: newMockHARemovableNodeBackend(func(ctx context.Context, s string) (bool, error) {
113-
return false, errors.New("error checking")
112+
return false, otherErr
114113
}),
115114
nodeIDCtx: nodeIDCtx,
116-
wantError: false,
115+
wantError: otherErr,
117116
},
118117
}
119118
for _, tc := range testCases {
@@ -122,8 +121,8 @@ func Test_haMembershipServerCheck(t *testing.T) {
122121
logger: hclog.NewNullLogger(),
123122
}
124123
err := haMembershipServerCheck(tc.nodeIDCtx, c, tc.haBackend)
125-
if tc.wantError {
126-
require.Error(t, err)
124+
if tc.wantError != nil {
125+
require.EqualError(t, err, tc.wantError.Error())
127126
} else {
128127
require.NoError(t, err)
129128
}

0 commit comments

Comments
 (0)