package timeout

import (
	"fmt"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
)

func TestTimeout(t *testing.T) {

	sleep := func(t time.Duration) func() error {
		return func() error {
			time.Sleep(t)
			return nil
		}
	}
	require.Equal(t, ErrTimedOut, Run(sleep(50*time.Millisecond), 10*time.Millisecond))
	require.Equal(t, nil, Run(sleep(10*time.Millisecond), 50*time.Millisecond))
	err := Run(func() error {
		return fmt.Errorf("blah")
	}, 100*time.Second)
	require.NotNil(t, err)
	require.Equal(t, "blah", err.Error())
}