mirror of https://github.com/docker/buildx.git
64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package waitmap
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetAfter(t *testing.T) {
|
|
m := New()
|
|
|
|
m.Set("foo", "bar")
|
|
m.Set("bar", "baz")
|
|
|
|
ctx := context.TODO()
|
|
v, err := m.Get(ctx, "foo", "bar")
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 2, len(v))
|
|
require.Equal(t, "bar", v["foo"])
|
|
require.Equal(t, "baz", v["bar"])
|
|
|
|
v, err = m.Get(ctx, "foo")
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(v))
|
|
require.Equal(t, "bar", v["foo"])
|
|
}
|
|
|
|
func TestTimeout(t *testing.T) {
|
|
m := New()
|
|
|
|
m.Set("foo", "bar")
|
|
|
|
ctx, cancel := context.WithTimeoutCause(context.TODO(), 100*time.Millisecond, nil)
|
|
defer cancel()
|
|
|
|
_, err := m.Get(ctx, "bar")
|
|
require.Error(t, err)
|
|
require.ErrorIs(t, err, context.DeadlineExceeded)
|
|
}
|
|
|
|
func TestBlocking(t *testing.T) {
|
|
m := New()
|
|
|
|
m.Set("foo", "bar")
|
|
|
|
go func() {
|
|
time.Sleep(100 * time.Millisecond)
|
|
m.Set("bar", "baz")
|
|
time.Sleep(50 * time.Millisecond)
|
|
m.Set("baz", "abc")
|
|
}()
|
|
|
|
ctx := context.TODO()
|
|
v, err := m.Get(ctx, "foo", "bar", "baz")
|
|
require.NoError(t, err)
|
|
require.Equal(t, 3, len(v))
|
|
require.Equal(t, "bar", v["foo"])
|
|
require.Equal(t, "baz", v["bar"])
|
|
require.Equal(t, "abc", v["baz"])
|
|
}
|