busretry_test.go RawHistoryFolder

1package busretry
2
3import (
4	"context"
5	"errors"
6	"testing"
7	"time"
8
9	"github.com/bnema/neferwl/internal/adapters/logging"
10)
11
12// A failing or returning attempt runs again until ctx ends, and Run returns
13// without waiting out the delay.
14func TestRunRetriesUntilCancelled(t *testing.T) {
15	ctx, cancel := context.WithCancel(context.Background())
16	calls := make(chan int, 8)
17	n := 0
18	done := make(chan struct{})
19	go func() {
20		defer close(done)
21		Run(ctx, time.Millisecond, "test feature", logging.For(ctx, "test"), func(context.Context) error {
22			n++
23			select {
24			case calls <- n:
25			case <-ctx.Done():
26			}
27			if n%2 == 0 {
28				return nil
29			}
30			return errors.New("bus gone")
31		})
32	}()
33	for want := 1; want <= 3; want++ {
34		select {
35		case got := <-calls:
36			if got != want {
37				t.Fatalf("attempt %d, want %d", got, want)
38			}
39		case <-time.After(5 * time.Second):
40			t.Fatalf("attempt %d never ran", want)
41		}
42	}
43	cancel()
44	select {
45	case <-done:
46	case <-time.After(5 * time.Second):
47		t.Fatal("Run outlived its context")
48	}
49}
50
51// A long delay does not hold Run once ctx ends.
52func TestRunStopsDuringDelay(t *testing.T) {
53	ctx, cancel := context.WithCancel(context.Background())
54	started := make(chan struct{})
55	done := make(chan struct{})
56	go func() {
57		defer close(done)
58		Run(ctx, time.Hour, "test feature", logging.For(ctx, "test"), func(context.Context) error {
59			close(started)
60			return errors.New("fail")
61		})
62	}()
63	<-started
64	cancel()
65	select {
66	case <-done:
67	case <-time.After(5 * time.Second):
68		t.Fatal("Run waits out the retry delay after cancel")
69	}
70}