diff options
-rw-r--r-- | internal/source/gitlab/cache/cache_test.go | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/internal/source/gitlab/cache/cache_test.go b/internal/source/gitlab/cache/cache_test.go index 97b4c0cb..d0fbd4e1 100644 --- a/internal/source/gitlab/cache/cache_test.go +++ b/internal/source/gitlab/cache/cache_test.go @@ -168,15 +168,26 @@ func TestResolve(t *testing.T) { }) }) - t.Run("when retrieval failed because of a timeout", func(t *testing.T) { + t.Run("when retrieval failed because of an external context timeout", func(t *testing.T) { ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(-time.Hour)) defer cancel() - withTestCache(resolverConfig{failure: errors.New("resolver 500")}, func(cache *Cache, resolver *client) { + withTestCache(resolverConfig{}, func(cache *Cache, resolver *client) { lookup := cache.Resolve(ctx, "my.gitlab.com") assert.Equal(t, uint64(0), resolver.resolutions) assert.EqualError(t, lookup.Error, "context timeout") }) }) + + t.Run("when retrieval failed because of an internal context timeout", func(t *testing.T) { + retrievalTimeout = 0 + + withTestCache(resolverConfig{}, func(cache *Cache, resolver *client) { + lookup := cache.Resolve(context.Background(), "my.gitlab.com") + + assert.Equal(t, uint64(0), resolver.resolutions) + assert.EqualError(t, lookup.Error, "context timeout") + }) + }) } |