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) }