JakeHillion/storj
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

satellite/gracefulexit: check for unknown error in graceful exit disable test

Browse Source 
Allow error in graceful exit disable test to be rpcstatus.Unimplemented (grpc) or rpcstatus.Unknown (drpc)
This commit is contained in:
Maximillian von Briesen 2019-11-01 17:21:30 -04:00 committed by GitHub
parent be2dd1ca72
commit f9df0ea591
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
satellite/gracefulexit/endpoint_test.go
View File

@ -824,7 +824,10 @@ func TestExitDisabled(t *testing.T) {
// Process endpoint should return immediately if GE is disabled
response, err := processClient.Recv()
require.True(t, errs2.IsRPC(err, rpcstatus.Unimplemented))
require.Error(t, err)
// grpc will return "Unimplemented", drpc will return "Unknown"
unimplementedOrUnknown := errs2.IsRPC(err, rpcstatus.Unimplemented) || errs2.IsRPC(err, rpcstatus.Unknown)
require.True(t, unimplementedOrUnknown)
require.Nil(t, response)
})
}