client_test.go

package updown

import (
	"net"
	"net/http"
	"os"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

const TQToken = "s7su"

func newClient() *Client {
	apiKey := os.Getenv("UPDOWN_API_KEY")
	if apiKey == "" {
		panic("API key is not set")
	}
	return NewClient(apiKey, nil)
}

func TestTokenForAlias(t *testing.T) {
	client := newClient()
	// Cache miss + alias not found
	token, err := client.Check.TokenForAlias("foo")
	assert.Equal(t, "", token)
	assert.Equal(t, ErrTokenNotFound, err)

	// - Cache miss + match found after request
	// - Cache hit
	for i := 0; i < 2; i++ {
		token, err = client.Check.TokenForAlias("Teen Quotes")
		assert.Nil(t, err)
		assert.Equal(t, TQToken, token)
	}
}

func TestList(t *testing.T) {
	client := newClient()
	checks, resp, _ := client.Check.List()

	assert.Equal(t, http.StatusOK, resp.StatusCode)
	assert.True(t, len(checks) > 0)
	found := false
	for _, element := range checks {
		if element.Alias == "Teen Quotes" {
			found = true
			break
		}
	}
	assert.True(t, found, "Cannot found the Teen Quotes check")
}

func TestGet(t *testing.T) {
	client := newClient()
	check, resp, _ := client.Check.Get(TQToken)

	assert.Equal(t, http.StatusOK, resp.StatusCode)
	assert.Equal(t, "Teen Quotes", check.Alias)

	check, resp, err := client.Check.Get("aaaaaa")
	assert.Equal(t, http.StatusNotFound, resp.StatusCode)
	assert.Equal(t, "GET https://updown.io/api/checks/aaaaaa: 404 ", err.Error())
}

func TestListDowntimes(t *testing.T) {
	client := newClient()
	// Page should be set to 1 automatically
	downs, resp, _ := client.Downtime.List(TQToken, -1)

	assert.Equal(t, http.StatusOK, resp.StatusCode)
	assert.True(t, len(downs) > 1)

	// Page with no downtimes
	downs, resp, _ = client.Downtime.List(TQToken, 200)
	assert.Equal(t, http.StatusOK, resp.StatusCode)
	assert.Equal(t, 0, len(downs))
}

func TestAddUpdateRemoveCheck(t *testing.T) {
	client := newClient()
	res, resp, _ := client.Check.Add(CheckItem{URL: "https://google.fr"})
	assert.Equal(t, http.StatusCreated, resp.StatusCode)
	assert.Equal(t, "https://google.fr", res.URL)

	res, resp, _ = client.Check.Update(res.Token, CheckItem{URL: "https://google.com"})
	assert.Equal(t, http.StatusOK, resp.StatusCode)
	assert.Equal(t, "https://google.com", res.URL)

	result, resp, _ := client.Check.Remove(res.Token)
	assert.Equal(t, http.StatusOK, resp.StatusCode)
	assert.True(t, result)
}

func TestListMetrics(t *testing.T) {
	client := newClient()
	now := time.Now()
	timeFormat := "2006-01-02 15:04:05 -0700"
	from, to := now.AddDate(0, 0, -1).Format(timeFormat), now.Format(timeFormat)
	metricRes, resp, _ := client.Metric.List(TQToken, "host", from, to)

	assert.Equal(t, http.StatusOK, resp.StatusCode)
	for _, location := range updownLocations() {
		assert.Contains(t, metricRes, location)
	}
	assert.True(t, len(metricRes) > 1)
}

func TestListNodes(t *testing.T) {
	client := newClient()
	nodeRes, resp, _ := client.Node.List()

	assert.Equal(t, http.StatusOK, resp.StatusCode)
	for _, location := range updownLocations() {
		assert.Contains(t, nodeRes, location)
	}
	assert.True(t, len(nodeRes) > 1)
}

func TestListIPv4(t *testing.T) {
	client := newClient()
	IPs, resp, _ := client.Node.ListIPv4()

	for _, ip := range IPs {
		assert.True(t, isIPv4(net.ParseIP(ip)))
	}

	assert.Equal(t, http.StatusOK, resp.StatusCode)
	assert.Equal(t, len(IPs), len(updownLocations()))
}

func TestListIPv6(t *testing.T) {
	client := newClient()
	IPs, resp, _ := client.Node.ListIPv6()

	for _, ip := range IPs {
		assert.True(t, isIPv6(net.ParseIP(ip)))
	}

	assert.Equal(t, http.StatusOK, resp.StatusCode)
	assert.Equal(t, len(IPs), len(updownLocations()))
}

func updownLocations() []string {
	return []string{"lan", "mia", "bhs", "gra", "fra", "sin", "tok", "syd"}
}

func isIPv4(ip net.IP) bool {
	return ip.To4().String() == ip.String()
}

func isIPv6(ip net.IP) bool {
	return ip.To16().String() == ip.String()
}