| 163 | } |
| 164 | |
| 165 | func TestRetry_NonRetryableError(t *testing.T) { |
| 166 | t.Parallel() |
| 167 | |
| 168 | calls := 0 |
| 169 | err := chatretry.Retry(context.Background(), func(_ context.Context) error { |
| 170 | calls++ |
| 171 | return xerrors.New("invalid api key") |
| 172 | }, nil) |
| 173 | |
| 174 | require.Error(t, err) |
| 175 | require.EqualError(t, err, "invalid api key") |
| 176 | require.Equal(t, 1, calls) |
| 177 | require.Equal( |
| 178 | t, |
| 179 | chaterror.Classify(xerrors.New("invalid api key")), |
| 180 | chaterror.Classify(err), |
| 181 | ) |
| 182 | } |
| 183 | |
| 184 | func TestRetry_ContextCanceledDuringWait(t *testing.T) { |
| 185 | t.Parallel() |