Modify IBD to accept various transport backends

This allows IBD to try using faster and cheaper transports first with
fallback to more reliable and expensive transports later.

1 files changed, 7 insertions, 6 deletions

diff --git a/backend/internal/ibd/search_test.go b/backend/internal/ibd/search_test.go
index ac0f578..f291033 100644
--- a/backend/internal/ibd/search_test.go
+++ b/backend/internal/ibd/search_test.go
@@ -5,6 +5,8 @@ import (
 	"net/http"
 	"testing"
 
+	"github.com/ansg191/ibd-trader-backend/internal/ibd/transport"
+	"github.com/jarcoal/httpmock"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
@@ -190,13 +192,12 @@ func TestClient_Search(t *testing.T) {
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			server := newServer(t, http.HandlerFunc(func(writer http.ResponseWriter, r *http.Request) {
-				_, _ = writer.Write([]byte(tt.response))
-			}))
-			defer server.Close()
+			tp := httpmock.NewMockTransport()
+			tp.RegisterResponder("GET", searchUrl, httpmock.NewStringResponder(200, tt.response))
 
-			client, err := NewClient(http.DefaultClient, apiKey, new(cookieSourceStub), "", WithBaseURL(server.URL))
-			require.NoError(t, err)
+			client := NewClient([]transport.Transport{
+				&http.Client{Transport: tp},
+			}, new(cookieSourceStub))
 
 			tt.f(t, client)
 		})