health_test.go

package atomicasset

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

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

func TestClient_GetHealth(t *testing.T) {
	srv := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
		if req.URL.String() == "/health" {
			payload := `{
                "success":true,
                "data":{
                    "version":"1.0.0",
                    "postgres":{
                        "status":"OK",
                        "readers":[
                            {
                                "block_num":"167836036"
                            },
                            {
                                "block_num":"167836034"
                            }
                        ]
                    },
                    "redis":{
                        "status":"OK"
                    },
                    "chain":{
                        "status":"OK",
                        "head_block":167836035,
                        "head_time":1645374771500
                    }
                },
                "query_time":1645374772067
            }`

			res.Header().Add("Content-type", "application/json; charset=utf-8")
			_, err := res.Write([]byte(payload))
			assert.NoError(t, err)
		}
	}))

	client := New(srv.URL)

	h, err := client.GetHealth()

	require.NoError(t, err)
	assert.Equal(t, 200, h.HTTPStatusCode)

	assert.True(t, h.Success)
	assert.Equal(t, time.Date(2022, time.February, 20, 16, 32, 52, int(time.Millisecond)*67, time.UTC), h.QueryTime.Time())

	// Data
	assert.Equal(t, "1.0.0", h.Data.Version)

	// Postgres
	assert.Equal(t, "OK", h.Data.Postgres.Status)

	// Redis
	assert.Equal(t, "OK", h.Data.Redis.Status)

	// Chain
	assert.Equal(t, "OK", h.Data.Chain.Status)
	assert.Equal(t, int64(167836035), h.Data.Chain.HeadBlock)
	assert.Equal(t, time.Date(2022, time.February, 20, 16, 32, 51, int(time.Millisecond)*500, time.UTC), h.Data.Chain.HeadTime.Time())
}

func TestClient_GetHealthFailed(t *testing.T) {
	srv := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
		if req.URL.String() == "/health" {
			payload := `{
                "success":true,
                "data":{
                    "version":"1.0.0",
                    "postgres":{
                        "status":"ERROR",
                        "readers":[]
                    },
                    "redis":{
                        "status":"ERROR"
                    },
                    "chain":{
                        "status":"ERROR",
                        "head_block":0,
                        "head_time":0
                    }
                },
                "query_time":1645374772067
            }`

			res.Header().Add("Content-type", "application/json")
			_, err := res.Write([]byte(payload))
			assert.NoError(t, err)
		}
	}))

	client := New(srv.URL)

	h, err := client.GetHealth()

	require.NoError(t, err)
	assert.Equal(t, 200, h.HTTPStatusCode)

	assert.True(t, h.Success)
	assert.Equal(t, time.Date(2022, time.February, 20, 16, 32, 52, int(time.Millisecond)*67, time.UTC), h.QueryTime.Time())

	// Data
	assert.Equal(t, "1.0.0", h.Data.Version)

	// Postgres
	assert.Equal(t, "ERROR", h.Data.Postgres.Status)

	// Redis
	assert.Equal(t, "ERROR", h.Data.Redis.Status)

	// Chain
	assert.Equal(t, "ERROR", h.Data.Chain.Status)
	assert.Equal(t, int64(0), h.Data.Chain.HeadBlock)

	assert.Equal(t, time.Unix(0, 0).UTC(), h.Data.Chain.HeadTime.Time())
}