Use interface

0xsequence · Jun 5, 2024 · 9f53ea7 · 9f53ea7
1 parent dc8818b
commit 9f53ea7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/providers.go b/providers.go
@@ -17,7 +17,11 @@ type Provider interface {
 	FetchTokenList(ctx context.Context, chainID uint64, source SourceType) (*TokenList, error)
 }
 
-func NewSequenceProvider(client *http.Client, rootURL string) (Provider, error) {
+type HTTPClient interface {
+	Do(req *http.Request) (*http.Response, error)
+}
+
+func NewSequenceProvider(client HTTPClient, rootURL string) (Provider, error) {
 	if client == nil {
 		client = http.DefaultClient
 	}
@@ -31,7 +35,7 @@ func NewSequenceProvider(client *http.Client, rootURL string) (Provider, error)
 
 type sequenceProvider struct {
 	id      string
-	client  *http.Client
+	client  HTTPClient
 	rootURL string
 }