Function TestExecutionStateVUIDs

lib/executor/execution_test.go:21–78

(t *testing.T)

)

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

	testCases := []struct {
		seq, seg string
	}{
		{},
		{seq: "0,1/4,3/4,1", seg: "0:1/4"},
		{seq: "0,0.3,0.5,0.6,0.7,0.8,0.9,1", seg: "0.5:0.6"},
	}

	for _, tc := range testCases {
		t.Run(fmt.Sprintf("seq:%s;segment:%s", tc.seq, tc.seg), func(t *testing.T) {
			t.Parallel()
			ess, err := lib.NewExecutionSegmentSequenceFromString(tc.seq)
			require.NoError(t, err)
			segment, err := lib.NewExecutionSegmentFromString(tc.seg)
			require.NoError(t, err)
			et, err := lib.NewExecutionTuple(segment, &ess)
			require.NoError(t, err)

			start, offsets, _ := et.GetStripedOffsets()
			es := lib.NewExecutionState(nil, et, 0, 0)

			idl, idg := es.GetUniqueVUIdentifiers()
			assert.EqualValues(t, 1, idl)
			expGlobal := start + 1
			assert.EqualValues(t, expGlobal, idg)

			idl, idg = es.GetUniqueVUIdentifiers()
			assert.EqualValues(t, 2, idl)
			expGlobal += offsets[0]
			assert.EqualValues(t, expGlobal, idg)

			idl, idg = es.GetUniqueVUIdentifiers()
			assert.EqualValues(t, 3, idl)
			expGlobal += offsets[0]
			assert.EqualValues(t, expGlobal, idg)

			seed := time.Now().UnixNano()
			r := rand.New(rand.NewSource(seed))
			t.Logf("Random source seeded with %d\n", seed)
			count := 100 + r.Intn(50)

			synctest.Test(t, func(t *testing.T) {
				for range count {
					go func() {
						es.GetUniqueVUIdentifiers()
					}()
				}
				synctest.Wait()
				idl, idg = es.GetUniqueVUIdentifiers()
				assert.EqualValues(t, 4+count, idl)
				assert.EqualValues(t, (3+count)*int(offsets[0])+int(start+1), idg)
			})
		})
	}
}