Function Test_MsgDryRun

transport/msg_test.go:60–104

(t *testing.T)

}

func Test_MsgDryRun(t *testing.T) {
	tutils.CheckSkip(t, tutils.SkipTestArgs{Long: true})
	t.Setenv("AIS_STREAM_DRY_RUN", "true")

	// fill in common shared read-only bug
	random := newRand(mono.NanoTime())
	buf, slab := memsys.PageMM().AllocSize(cos.MiB)
	defer slab.Free(buf)
	random.Read(buf)

	wg := &sync.WaitGroup{}
	num := atomic.NewInt64(0)
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go func(idx int) {
			defer wg.Done()
			myrand := newRand(int64(idx * idx))
			tsize, prevsize, off := int64(0), int64(0), 0
			total := int64(cos.GiB * 4)
			if testing.Short() {
				total = cos.GiB
			}
			stream := transport.NewMsgStream(nil, "dry-msg"+strconv.Itoa(idx), cos.GenTie())
			for tsize < total {
				msize := myrand.Intn(memsys.PageSize - 64) // <= s.maxheader, zero-length OK
				if off+msize > len(buf) {
					off = 0
				}
				msg := &transport.Msg{Body: buf[off : off+msize]}
				off += msize
				err := stream.Send(msg)
				tassert.CheckFatal(t, err)
				num.Inc()
				tsize += int64(msize)
				if tsize-prevsize > total/2 {
					prevsize = tsize
					tlog.Logf("%s: %s\n", stream, cos.B2S(tsize, 0))
				}
			}
			stream.Fin()
		}(i)
	}
	wg.Wait()
	tlog.Logf("total messages: %d\n", num.Load())
}