agent/checkpoint_internal_test.go

Rama main · réplica de lectura del repositorio original.

package agent

import (
	"testing"

	"github.com/stretchr/testify/require"
	"golang.org/x/xerrors"

	"cdr.dev/slog/v3/sloggers/slogtest"
	"github.com/coder/coder/v2/testutil"
)

func TestCheckpoint_CompleteWait(t *testing.T) {
	t.Parallel()
	logger := testutil.Logger(t)
	ctx := testutil.Context(t, testutil.WaitShort)
	uut := newCheckpoint(logger)
	err := xerrors.New("test")
	uut.complete(err)
	got := uut.wait(ctx)
	require.Equal(t, err, got)
}

func TestCheckpoint_CompleteTwice(t *testing.T) {
	t.Parallel()
	logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true})
	ctx := testutil.Context(t, testutil.WaitShort)
	uut := newCheckpoint(logger)
	err := xerrors.New("test")
	uut.complete(err)
	uut.complete(nil) // drops CRITICAL log
	got := uut.wait(ctx)
	require.Equal(t, err, got)
}

func TestCheckpoint_WaitComplete(t *testing.T) {
	t.Parallel()
	logger := testutil.Logger(t)
	ctx := testutil.Context(t, testutil.WaitShort)
	uut := newCheckpoint(logger)
	err := xerrors.New("test")
	errCh := make(chan error, 1)
	go func() {
		errCh <- uut.wait(ctx)
	}()
	uut.complete(err)
	got := testutil.TryReceive(ctx, t, errCh)
	require.Equal(t, err, got)
}