Add context test

go.mod

@@ -1 +1,3 @@
 module github.com/Jeffail/tunny
+
+go 1.13

tunny.go

@@ -21,6 +21,7 @@
 package tunny
 
 import (
+	"context"
 	"errors"
 	"sync"
 	"sync/atomic"
@@ -218,7 +219,7 @@ func (p *Pool) ProcessTimed(
 // the result. If the context cancels before the job has finished the worker will
 // be interrupted and ErrJobTimedOut will be returned. ProcessCtx can be
 // called safely by any goroutines.
-func (p *Pool) ProcessCtx(ctx context.Context, i interface{}) (interface{}, error) {
+func (p *Pool) ProcessCtx(ctx context.Context, payload interface{}) (interface{}, error) {
 	atomic.AddInt64(&p.queuedJobs, 1)
 	defer atomic.AddInt64(&p.queuedJobs, -1)
 
@@ -231,14 +232,14 @@ func (p *Pool) ProcessCtx(ctx context.Context, i interface{}) (interface{}, erro
 			return nil, ErrPoolNotRunning
 		}
 	case <-ctx.Done():
-		return nil, ErrJobTimedOut
+		return nil, ctx.Err()
 	}
 
 	select {
 	case request.jobChan <- payload:
 	case <-ctx.Done():
 		request.interruptFunc()
-		return nil, ErrJobTimedOut
+		return nil, ctx.Err()
 	}
 
 	select {
@@ -248,7 +249,7 @@ func (p *Pool) ProcessCtx(ctx context.Context, i interface{}) (interface{}, erro
 		}
 	case <-ctx.Done():
 		request.interruptFunc()
-		return nil, ErrJobTimedOut
+		return nil, ctx.Err()
 	}
 
 	return payload, nil

tunny_test.go

@@ -21,6 +21,7 @@
 package tunny
 
 import (
+	"context"
 	"sync"
 	"sync/atomic"
 	"testing"
@@ -106,6 +107,24 @@ func TestFuncJobTimed(t *testing.T) {
 	}
 }
 
+func TestFuncJobCtx(t *testing.T) {
+	pool := NewFunc(10, func(in interface{}) interface{} {
+		intVal := in.(int)
+		return intVal * 2
+	})
+	defer pool.Close()
+
+	for i := 0; i < 10; i++ {
+		ret, err := pool.ProcessCtx(context.Background(), 10)
+		if err != nil {
+			t.Fatalf("Failed to process: %v", err)
+		}
+		if exp, act := 20, ret.(int); exp != act {
+			t.Errorf("Wrong result: %v != %v", act, exp)
+		}
+	}
+}
+
 func TestCallbackJob(t *testing.T) {
 	pool := NewCallback(10)
 	defer pool.Close()