first version

internal/tinvest/retry_test.go

@@ -0,0 +1,41 @@
+package tinvest
+
+import (
+	"context"
+	"errors"
+	"testing"
+	"time"
+)
+
+func TestWithRetryRetriesUntilSuccess(t *testing.T) {
+	attempts := 0
+	err := withRetry(context.Background(), 3, 0, func() error {
+		attempts++
+		if attempts < 3 {
+			return errors.New("temporary")
+		}
+		return nil
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if attempts != 3 {
+		t.Fatalf("attempts=%d, want 3", attempts)
+	}
+}
+
+func TestWithRetryStopsOnContextCancel(t *testing.T) {
+	ctx, cancel := context.WithCancel(context.Background())
+	cancel()
+	attempts := 0
+	err := withRetry(ctx, 3, time.Millisecond, func() error {
+		attempts++
+		return errors.New("temporary")
+	})
+	if !errors.Is(err, context.Canceled) {
+		t.Fatalf("err=%v, want context.Canceled", err)
+	}
+	if attempts != 0 {
+		t.Fatalf("attempts=%d, want 0", attempts)
+	}
+}