gopeed/pkg/download/engine/cancellation_test.go

package engine

import (
	"context"
	"errors"
	"github.com/dop251/goja"
	"testing"
	"time"
)

func TestCloseUnblocksPendingPromise(t *testing.T) {
	e := NewEngine(nil)
	entered := make(chan struct{})
	_, err := e.runOnLoop(func(r *goja.Runtime) (goja.Value, error) {
		return goja.Undefined(), r.Set("entered", func() { close(entered) })
	})
	if err != nil {
		t.Fatal(err)
	}
	done := make(chan error, 1)
	go func() { _, err := e.RunString(`entered(); new Promise(()=>{});`); done <- err }()
	select {
	case <-entered:
	case <-time.After(time.Second):
		t.Fatal("script did not enter")
	}
	e.Close()
	select {
	case err := <-done:
		if err == nil {
			t.Fatal("closed execution returned success")
		}
	case <-time.After(300 * time.Millisecond):
		t.Fatal("Engine.Close left the Promise waiter blocked")
	}
}

func TestExecutionDeadlineInterruptsSynchronousScript(t *testing.T) {
	e := NewEngine(nil)
	defer e.Close()
	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Millisecond)
	defer cancel()
	_, err := e.RunStringContext(ctx, `while (true) {}`)
	if !errors.Is(err, context.DeadlineExceeded) {
		t.Fatalf("got %v, want deadline exceeded", err)
	}
}