64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
|
package wallet
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/common"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestFetchAllCollectionsByOwner(t *testing.T) {
|
||
|
|
expected := []OpenseaCollection{OpenseaCollection{Name: "Rocky", Slug: "rocky", ImageURL: "ImageUrl", OwnedAssetCount: 1}}
|
||
|
|
response, _ := json.Marshal(expected)
|
||
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
w.WriteHeader(200)
|
||
|
|
_, err := w.Write(response)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}))
|
||
|
|
defer srv.Close()
|
||
|
|
|
||
|
|
opensea := &OpenseaClient{
|
||
|
|
client: srv.Client(),
|
||
|
|
url: srv.URL,
|
||
|
|
}
|
||
|
|
res, err := opensea.fetchAllCollectionsByOwner(common.Address{1})
|
||
|
|
assert.Equal(t, expected, res)
|
||
|
|
assert.Nil(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFetchAllAssetsByOwnerAndCollection(t *testing.T) {
|
||
|
|
expected := []OpenseaAsset{OpenseaAsset{
|
||
|
|
ID: 1,
|
||
|
|
Name: "Rocky",
|
||
|
|
Description: "Rocky Balboa",
|
||
|
|
Permalink: "permalink",
|
||
|
|
ImageThumbnailURL: "ImageThumbnailURL",
|
||
|
|
ImageURL: "ImageUrl",
|
||
|
|
Contract: OpenseaContract{Address: "1"},
|
||
|
|
Collection: OpenseaAssetCollection{Name: "Rocky"},
|
||
|
|
}}
|
||
|
|
response, _ := json.Marshal(OpenseaAssetContainer{Assets: expected})
|
||
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
w.WriteHeader(200)
|
||
|
|
_, err := w.Write(response)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}))
|
||
|
|
defer srv.Close()
|
||
|
|
|
||
|
|
opensea := &OpenseaClient{
|
||
|
|
client: srv.Client(),
|
||
|
|
url: srv.URL,
|
||
|
|
}
|
||
|
|
res, err := opensea.fetchAllAssetsByOwnerAndCollection(common.Address{1}, "rocky", 200)
|
||
|
|
assert.Equal(t, expected, res)
|
||
|
|
assert.Nil(t, err)
|
||
|
|
}
|