From 575a0a53b149ac2da21e368ef809dd2180a878f5 Mon Sep 17 00:00:00 2001 From: jwijenbergh Date: Fri, 24 May 2024 12:35:42 +0200 Subject: API Test: Mock Transport by passing it around --- internal/api/cache.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'internal/api/cache.go') diff --git a/internal/api/cache.go b/internal/api/cache.go index 4777334..420a4b0 100644 --- a/internal/api/cache.go +++ b/internal/api/cache.go @@ -2,6 +2,7 @@ package api import ( "context" + "net/http" "sync" "time" @@ -16,7 +17,7 @@ type EndpointCache struct { } // Get() returns a cached or fresh endpoint cache copy -func (ec *EndpointCache) Get(ctx context.Context, wk string) (*endpoints.Endpoints, error) { +func (ec *EndpointCache) Get(ctx context.Context, wk string, transport http.RoundTripper) (*endpoints.Endpoints, error) { ec.mu.Lock() defer ec.mu.Unlock() @@ -35,7 +36,7 @@ func (ec *EndpointCache) Get(ctx context.Context, wk string) (*endpoints.Endpoin } // get fresh API endpoints - ep, err := getEndpoints(ctx, wk) + ep, err := getEndpoints(ctx, wk, transport) if err != nil { return nil, err } -- cgit v1.2.3