Function TestDefaultClientRetryPolicy_NextRetry

retry_policy_test.go:19–84

(t *testing.T)

var _ ClientRetryPolicy = &DefaultClientRetryPolicy{}

func TestDefaultClientRetryPolicy_NextRetry(t *testing.T) {
	t.Parallel()

	type testBundle struct {
		now time.Time
	}

	setup := func(t *testing.T) (*DefaultClientRetryPolicy, *testBundle) {
		t.Helper()

		var (
			now         = time.Now().UTC()
			timeNowFunc = func() time.Time { return now }
		)

		return &DefaultClientRetryPolicy{timeNowFunc: timeNowFunc}, &testBundle{
			now: now,
		}
	}

	t.Run("Schedule", func(t *testing.T) {
		t.Parallel()

		retryPolicy, bundle := setup(t)

		for attempt := 1; attempt < 10; attempt++ {
			retrySecondsWithoutJitter := retryPolicy.retrySecondsWithoutJitter(attempt)
			allowedDelta := timeutil.SecondsAsDuration(retrySecondsWithoutJitter * 0.2)

			nextRetryAt := retryPolicy.NextRetry(&rivertype.JobRow{
				Attempt:     attempt,
				AttemptedAt: &bundle.now,
				Errors:      make([]rivertype.AttemptError, attempt-1),
			})
			require.WithinDuration(t, bundle.now.Add(timeutil.SecondsAsDuration(retrySecondsWithoutJitter)), nextRetryAt, allowedDelta)
		}
	})

	t.Run("MaxRetryDuration", func(t *testing.T) {
		t.Parallel()

		retryPolicy, bundle := setup(t)

		maxRetryDuration := timeutil.SecondsAsDuration(maxDurationSeconds)

		// First time the maximum will be hit.
		require.Equal(t,
			bundle.now.Add(maxRetryDuration),
			retryPolicy.NextRetry(&rivertype.JobRow{
				Attempt:     310,
				AttemptedAt: &bundle.now,
				Errors:      make([]rivertype.AttemptError, 310-1),
			}),
		)

		// And will be hit on all subsequent attempts as well.
		require.Equal(t,
			bundle.now.Add(maxRetryDuration),