forked from TrueCloudLab/frostfs-qos
24 lines
482 B
Go
24 lines
482 B
Go
|
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)
|
||
|
|
}
|