package main

import (
	"testing"
)

func TestSingleTask(t *testing.T) {
	jsonPath := "./example/单层单任务.json"
	RunSingleTask(jsonPath)
}

func TestPreys(t *testing.T) {
	var preys []string
	var pages []string
	pages = []string{
		"test_1",
		"test_2",
	}
	chs := make([]chan string, len(pages))

	runPrey := func(task_id int, ch chan string, page string) {
		ch <- page
	}

	for task_id, page := range pages {
		// preys = append(preys, page)
		chs[task_id] = make(chan string, 1)
		go runPrey(task_id, chs[task_id], page)
	}
	for _, ch := range chs {
		s := <-ch
		preys = append(preys, s)
	}
	t.Error(preys)
}

func TestDouble(t *testing.T) {
	jsonPath := "./example/双层单任务.json"
	RunDoubleTask(jsonPath)
}