GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-06-11 10:49:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: unittest

Browse Source
This commit is contained in:
xucong053
2022-05-20 14:48:55 +08:00
committed by 徐聪
parent 6ad03e81bb
commit ab91feedbc
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
hrp/internal/boomer/runner_test.go
View File

@@ -129,6 +129,7 @@ func TestSpawnWorkers(t *testing.T) {
runner.client = newClient("localhost", 5557, runner.nodeID) runner.client = newClient("localhost", 5557, runner.nodeID)
runner.setTasks(tasks) runner.setTasks(tasks)
runner.stopChan = make(chan bool) runner.stopChan = make(chan bool)
runner.rebalance = make(chan bool)
go runner.spawnWorkers(10, 10, runner.stopChan, runner.spawnComplete) go runner.spawnWorkers(10, 10, runner.stopChan, runner.spawnComplete)
time.Sleep(2 * time.Second) time.Sleep(2 * time.Second)
@@ -167,6 +168,7 @@ func TestSpawnWorkersWithManyTasks(t *testing.T) {
const numToSpawn int64 = 30 const numToSpawn int64 = 30
runner.stopChan = make(chan bool) runner.stopChan = make(chan bool)
runner.rebalance = make(chan bool)
go runner.spawnWorkers(numToSpawn, float64(numToSpawn), runner.stopChan, runner.spawnComplete) go runner.spawnWorkers(numToSpawn, float64(numToSpawn), runner.stopChan, runner.spawnComplete)
time.Sleep(2 * time.Second) time.Sleep(2 * time.Second)
@@ -257,6 +259,7 @@ func TestStop(t *testing.T) {
tasks := []*Task{taskA} tasks := []*Task{taskA}
runner := newWorkerRunner("localhost", 5557) runner := newWorkerRunner("localhost", 5557)
runner.stopChan = make(chan bool) runner.stopChan = make(chan bool)
runner.rebalance = make(chan bool)
runner.setTasks(tasks) runner.setTasks(tasks)
runner.spawn.setSpawn(10, 10) runner.spawn.setSpawn(10, 10)
runner.updateState(StateSpawning) runner.updateState(StateSpawning)
@@ -308,6 +311,7 @@ func TestOnQuitMessage(t *testing.T) {
runner.updateState(StateRunning) runner.updateState(StateRunning)
runner.closeChan = make(chan bool) runner.closeChan = make(chan bool)
runner.stopChan = make(chan bool) runner.stopChan = make(chan bool)
runner.rebalance = make(chan bool)
runner.client.shutdownChan = make(chan bool) runner.client.shutdownChan = make(chan bool)
runner.onMessage(newGenericMessage("quit", nil, runner.nodeID)) runner.onMessage(newGenericMessage("quit", nil, runner.nodeID))
<-runner.closeChan <-runner.closeChan