executor_test.go

package executor

import (
	"context"
	"fmt"
	"log"
	"math/rand"
	"testing"
	"time"
)

func TestCanRunExecutor(t *testing.T) {
	//TODO simple test NEED TO CHANGE
	ctx, cancel := context.WithTimeout(context.Background(), 20000*time.Second)
	defer cancel()

	executor := NewService(ctx, 10)
	defer executor.Stop()

	ts := 20
	runnables := make([]Runnable, ts)
	for i := 0; i < ts; i++ {
		runnables[i] = NewTask(fmt.Sprintf("test_%v", i+1), fmt.Sprintf("testValue_%v", i+1), exec(ctx), complete, errH)
	}

	executor.Start()

	executor.AddTasks(runnables...)

}

func exec(ctx context.Context) func(v string) error {
	return func(v string) error {
		for i := 0; i < 5; i++ {
			select {
			case <-ctx.Done():
				return nil
			default:
				rand.Seed(time.Now().UnixNano())
				time.Sleep(time.Duration(rand.Intn(2)) * time.Second)
				fmt.Printf("executing job %s\n", v)
			}
		}

		return nil
	}
}

func complete(v string) error {
	fmt.Printf("Complete successfull %s\n", v)

	return nil
}

func errH(v string, err error) {
	if err != nil {
		log.Fatal(v + err.Error())
	}
}