health_test.go

package health

import (
	"bytes"
	"encoding/json"
	"errors"
	"strings"
	"testing"

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

var (
	untabber = strings.NewReplacer("\n\t\t", "\n")
)

func TestHealth_Alert_SavesError(t *testing.T) {
	test := assert.New(t)

	health := NewHealth()
	health.Alert(errors.New("a"), "key1", "key2")

	test.EqualValues([]error{Error("a")}, health.GetErrors())
}

func TestHealth_Alert_UsesKeys(t *testing.T) {
	test := assert.New(t)

	health := NewHealth()
	health.Alert(errors.New("a"), "key1", "key2")

	test.EqualValues([]error{Error("a")}, health.GetErrors())
	test.EqualValues([]string{"key1@key2"}, health.keys)
}

func TestHealth_Resolve_RemovesError(t *testing.T) {
	test := assert.New(t)

	health := NewHealth()
	health.Alert(errors.New("a"), "key1", "key2")
	health.Alert(errors.New("b"), "key3", "key4")

	health.Resolve("key3", "key4")

	test.EqualValues([]error{Error("a")}, health.GetErrors())
}

func TestHealth_MarshalJSON_ReturnsZeroForNoErrors(t *testing.T) {
	test := assert.New(t)

	health := NewHealth()

	buffer := bytes.NewBuffer(nil)

	encoder := json.NewEncoder(buffer)
	encoder.SetIndent("", "\t")

	err := encoder.Encode(health)
	test.NoError(err)

	test.Equal(
		untabber.Replace(`{
			"status": 0
		}
		`),
		string(buffer.Bytes()),
	)
}

func TestHealth_HasErrors_ReturnsTrueIfErrorExists(t *testing.T) {
	test := assert.New(t)

	health := NewHealth()
	health.Alert(nil, "a")

	test.True(health.HasErrors())
}

func TestHealth_HasErrors_ReturnsFalseIfNoErrorExists(t *testing.T) {
	test := assert.New(t)

	health := NewHealth()

	test.False(health.HasErrors())
}

func BenchmarkHealth_Resolve(b *testing.B) {
	health := NewHealth()

	health.Alert(errors.New("a"), "key1", "key2")
	health.Alert(errors.New("b"), "key1", "key2")
	health.Alert(errors.New("c"), "key1", "key2")

	for i := 0; i < b.N; i++ {
		health.Resolve("x")
	}
}

func BenchmarkHealth_Alert(b *testing.B) {
	health := NewHealth()

	for i := 0; i < b.N; i++ {
		health.Alert(errors.New("a"), "key1", "key2")
	}
}