package jsonip import ( "net" "net/http" "net/http/httptest" "testing" "github.com/stretchr/testify/assert" ) func TestService_Name(t *testing.T) { s := Service{} assert.Equal(t, "jsonip", s.Name()) } func TestService_Lookup(t *testing.T) { server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { _, err := w.Write([]byte(`{"ip":"211.46.32.214","geo-ip":"https://getjsonip.com/#plus","API Help":"https://getjsonip.com/#docs"}`)) assert.NoError(t, err) })) defer server.Close() s := Service{url: server.URL} ip, err := s.Lookup() assert.NoError(t, err) assert.Equal(t, net.IPv4(211, 46, 32, 214), ip) }