constant_vus_test.go

package executor

import (
	"context"
	"sync"
	"testing"
	"testing/synctest"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"gopkg.in/guregu/null.v3"

	"go.k6.io/k6/lib"
	"go.k6.io/k6/lib/types"
)

func getTestConstantVUsConfig() ConstantVUsConfig {
	return ConstantVUsConfig{
		BaseConfig: BaseConfig{GracefulStop: types.NullDurationFrom(100 * time.Millisecond)},
		VUs:        null.IntFrom(10),
		Duration:   types.NullDurationFrom(1 * time.Second),
	}
}

func TestConstantVUsRun(t *testing.T) {
	t.Parallel()

	synctest.Test(t, func(t *testing.T) {
		var result sync.Map

		runner := simpleRunner(func(ctx context.Context, state *lib.State) error {
			select {
			case <-ctx.Done():
				return nil
			default:
			}
			currIter, _ := result.LoadOrStore(state.VUID, uint64(0))
			result.Store(state.VUID, currIter.(uint64)+1)
			time.Sleep(210 * time.Millisecond)
			return nil
		})

		test := setupExecutorTest(t, "", "", lib.Options{}, runner, getTestConstantVUsConfig())
		defer test.cancel()

		require.NoError(t, test.executor.Run(test.ctx, nil))

		var totalIters uint64
		result.Range(func(_, value any) bool {
			vuIters := value.(uint64)
			assert.Equal(t, uint64(5), vuIters)
			totalIters += vuIters
			return true
		})
		assert.Equal(t, uint64(50), totalIters)
	})
}