@@ -888,7 +888,6 @@ func TestVaultHandler_HandleJSONRPCUserMessage(t *testing.T) {
888888 })
889889
890890 t .Run ("empty delete batch" , func (t * testing.T ) {
891- var wg sync.WaitGroup
892891 h , callback , don , _ := setupHandler (t )
893892 don .AssertNotCalled (t , "SendToNode" )
894893
@@ -899,22 +898,16 @@ func TestVaultHandler_HandleJSONRPCUserMessage(t *testing.T) {
899898 Params : & invalidParams ,
900899 }
901900
902- wg .Add (1 )
903- go func () {
904- defer wg .Done ()
905- resp , err := callback .Wait (t .Context ())
906- assert .NoError (t , err )
907- var secretsResponse jsonrpc.Response [vaultcommon.DeleteSecretsResponse ]
908- err = json .Unmarshal (resp .RawResponse , & secretsResponse )
909- assert .NoError (t , err )
910- assert .Equal (t , invalidParamsRequest .ID , secretsResponse .ID , "Request ID should match" )
911- assert .Equal (t , "invalid params error: failed to validate delete secrets request: request batch must contain at least 1 item" , secretsResponse .Error .Message , "Error message should match" )
912- assert .Equal (t , api .ToJSONRPCErrorCode (api .InvalidParamsError ), secretsResponse .Error .Code , "Error code should match" )
913- }()
914-
915901 err := h .HandleJSONRPCUserMessage (t .Context (), invalidParamsRequest , callback )
916902 require .NoError (t , err )
917- wg .Wait ()
903+
904+ resp , err := callback .Wait (t .Context ())
905+ require .NoError (t , err )
906+ var secretsResponse jsonrpc.Response [vaultcommon.DeleteSecretsResponse ]
907+ require .NoError (t , json .Unmarshal (resp .RawResponse , & secretsResponse ))
908+ assert .Equal (t , invalidParamsRequest .ID , secretsResponse .ID , "Request ID should match" )
909+ assert .Equal (t , "invalid params error: failed to validate delete secrets request: request batch must contain at least 1 item" , secretsResponse .Error .Message , "Error message should match" )
910+ assert .Equal (t , api .ToJSONRPCErrorCode (api .InvalidParamsError ), secretsResponse .Error .Code , "Error code should match" )
918911 })
919912
920913 t .Run ("invalid params error" , func (t * testing.T ) {
@@ -1191,17 +1184,15 @@ func TestVaultHandler_PreAuthValidationSkipsAuthorization(t *testing.T) {
11911184 Method : vaulttypes .MethodSecretsCreate ,
11921185 }
11931186
1194- wg .Add (1 )
1195- go func () {
1196- defer wg .Done ()
1187+ wg .Go (func () {
11971188 resp , waitErr := callback .Wait (t .Context ())
1198- require .NoError (t , waitErr )
1189+ assert .NoError (t , waitErr )
11991190 var secretsResponse jsonrpc.Response [vaultcommon.CreateSecretsResponse ]
1200- require .NoError (t , json .Unmarshal (resp .RawResponse , & secretsResponse ))
1201- require .Equal (t , req .ID , secretsResponse .ID )
1202- require .Equal (t , api .ToJSONRPCErrorCode (api .InvalidParamsError ), secretsResponse .Error .Code )
1203- require .Contains (t , secretsResponse .Error .Message , "request params must not be nil" )
1204- }( )
1191+ assert .NoError (t , json .Unmarshal (resp .RawResponse , & secretsResponse ))
1192+ assert .Equal (t , req .ID , secretsResponse .ID )
1193+ assert .Equal (t , api .ToJSONRPCErrorCode (api .InvalidParamsError ), secretsResponse .Error .Code )
1194+ assert .Contains (t , secretsResponse .Error .Message , "request params must not be nil" )
1195+ })
12051196
12061197 require .NoError (t , h .HandleJSONRPCUserMessage (t .Context (), req , callback ))
12071198 wg .Wait ()
@@ -1219,17 +1210,15 @@ func TestVaultHandler_PreAuthValidationSkipsAuthorization(t *testing.T) {
12191210 Params : & invalidParams ,
12201211 }
12211212
1222- wg .Add (1 )
1223- go func () {
1224- defer wg .Done ()
1213+ wg .Go (func () {
12251214 resp , waitErr := callback .Wait (t .Context ())
1226- require .NoError (t , waitErr )
1215+ assert .NoError (t , waitErr )
12271216 var secretsResponse jsonrpc.Response [vaultcommon.DeleteSecretsResponse ]
1228- require .NoError (t , json .Unmarshal (resp .RawResponse , & secretsResponse ))
1229- require .Equal (t , req .ID , secretsResponse .ID )
1230- require .Equal (t , api .ToJSONRPCErrorCode (api .InvalidParamsError ), secretsResponse .Error .Code )
1231- require .Contains (t , secretsResponse .Error .Message , "request batch must contain at least 1 item" )
1232- }( )
1217+ assert .NoError (t , json .Unmarshal (resp .RawResponse , & secretsResponse ))
1218+ assert .Equal (t , req .ID , secretsResponse .ID )
1219+ assert .Equal (t , api .ToJSONRPCErrorCode (api .InvalidParamsError ), secretsResponse .Error .Code )
1220+ assert .Contains (t , secretsResponse .Error .Message , "request batch must contain at least 1 item" )
1221+ })
12331222
12341223 require .NoError (t , h .HandleJSONRPCUserMessage (t .Context (), req , callback ))
12351224 wg .Wait ()
0 commit comments