diff options
Diffstat (limited to 'backend/internal/ibd/search_test.go')
| -rw-r--r-- | backend/internal/ibd/search_test.go | 13 |
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) }) |