main_test.go

package main

import (
	"testing"
	"time"

	"github.com/tinywell/loadgen/lib"
	"github.com/tinywell/loadgen/mock"
	"github.com/tinywell/loadgen/model"
)

func TestAll(t *testing.T) {
	caller := mock.NewMockCaller()
	tickets, _ := lib.NewTickets(2)
	set := lib.ParamSet{
		Caller:     caller,
		Tickets:    tickets,
		Lps:        10,
		TimeoutNS:  time.Second * 2,
		DurationNS: time.Second * 10,
		ResultChan: make(chan *model.LDResult),
	}
	gen := lib.NewGenerator(set)
	if ok := gen.Start(); ok {
		t.Log("OK")
	}
	for res := range set.ResultChan {
		t.Log(res)
	}
}

func TestStop(t *testing.T) {
	caller := mock.NewMockCaller()
	tickets, _ := lib.NewTickets(2)
	set := lib.ParamSet{
		Caller:     caller,
		Tickets:    tickets,
		Lps:        10,
		TimeoutNS:  time.Second * 2,
		DurationNS: time.Second * 10,
		ResultChan: make(chan *model.LDResult),
	}
	gen := lib.NewGenerator(set)
	if ok := gen.Start(); ok {
		t.Log("OK")
	}
	time.Sleep(time.Second * 2)
	gen.Stop()
}