package ipecho import ( "net" "net/http" "net/http/httptest" "testing" "github.com/stretchr/testify/assert" ) func TestService_Name(t *testing.T) { s := Service{} assert.Equal(t, "ipecho", s.Name()) } func TestService_Lookup(t *testing.T) { server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { _, err := w.Write([]byte("188.242.103.22")) 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(188, 242, 103, 22), ip) }