TrueCloudLab/frostfs-qos
14
0
Fork 6
Code Issues 1 Pull requests Projects Releases Packages Activity Actions

[#2] tagging: Add grpc and context methods

Browse source 
Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
This commit is contained in:
Dmitrii Stepanov 2025-01-21 09:56:13 +03:00
parent f4d8ebf13d
commit 1fb8b137c5
Signed by: dstepanov-yadro
GPG key ID: 237AF1A763293BC0
5 changed files with 190 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tagging/context_test.go Normal file
View file

@ -0,0 +1,23 @@
package tagging
import (
"context"
"testing"
"github.com/stretchr/testify/require"
)
func TestContext(t *testing.T) {
ctx := context.Background()
tag, ok := IOTagFromContext(ctx)
require.False(t, ok)
require.Equal(t, "", tag)
ctx = ContextWithIOTag(ctx, "tag1")
tag, ok = IOTagFromContext(ctx)
require.True(t, ok)
require.Equal(t, "tag1", tag)
ctx = ContextWithIOTag(ctx, "tag2")
tag, ok = IOTagFromContext(ctx)
require.True(t, ok)
require.Equal(t, "tag2", tag)
}