realloc/frostfs-node
1
0
Fork 0
forked from TrueCloudLab/frostfs-node
Code Releases Packages Activity

[#1656] qos: Add test for SetCriticalIOTag Interceptor

Browse source 
Change-Id: I4a55fcb84e6f65408a1c0120ac917e49e23354a1
Signed-off-by: Ekaterina Lebedeva <ekaterina.lebedeva@yadro.com>
This commit is contained in:
Ekaterina Lebedeva 2025-03-22 16:36:55 +03:00
parent 115aae7c34
commit dcb2b23a7d
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
internal/qos/grpc_test.go
View file

@ -19,6 +19,7 @@ const (
var (
errTest = errors.New("mock")
errWrongTag = errors.New("wrong tag")
errResExhausted *apistatus.ResourceExhausted
)
@ -119,3 +120,18 @@ func Test_MaxActiveRPCLimiter(t *testing.T) {
require.True(t, lim.released)
})
}
func TestSetCriticalIOTagUnaryServerInterceptor_Pass(t *testing.T) {
interceptor := qos.NewSetCriticalIOTagUnaryServerInterceptor()
called := false
handler := func(ctx context.Context, req any) (any, error) {
called = true
if tag, ok := tagging.IOTagFromContext(ctx); ok && tag == qos.IOTagCritical.String() {
return nil, nil
}
return nil, errWrongTag
}
_, err := interceptor(context.Background(), nil, nil, handler)
require.NoError(t, err)
require.True(t, called)
}