mackerel_test.go

package shimesaba_test

import (
	"context"
	"testing"
	"time"

	"github.com/mashiike/shimesaba"
	"github.com/stretchr/testify/require"
)

func TestRepositoryFetchVirtualAlerts(t *testing.T) {
	client := newMockMackerelClient(t)
	repo := shimesaba.NewRepository(client)

	cases := []struct {
		name        string
		serviceName string
		sloID       string
		startAt     time.Time
		endAt       time.Time
		expected    shimesaba.Alerts
	}{
		{
			name:        "SLO:*",
			serviceName: "shimesaba",
			sloID:       "hoge",
			startAt:     time.Date(2021, 10, 1, 0, 5, 0, 0, time.UTC),
			endAt:       time.Date(2021, 10, 1, 0, 15, 0, 0, time.UTC),
			expected: shimesaba.Alerts{
				{
					Reason:   "SLO:*",
					OpenedAt: time.Date(2021, 10, 1, 0, 10, 0, 0, time.UTC),
					ClosedAt: ptrTime(time.Date(2021, 10, 1, 0, 15, 0, 0, time.UTC)),
				},
			},
		},
		{
			name:        "SLO:availability,quarity ",
			serviceName: "shimesaba",
			sloID:       "availability",
			startAt:     time.Date(2021, 10, 1, 0, 5, 0, 0, time.UTC),
			endAt:       time.Date(2021, 10, 1, 0, 15, 0, 0, time.UTC),
			expected: shimesaba.Alerts{
				{
					Reason:   "SLO:*",
					OpenedAt: time.Date(2021, 10, 1, 0, 10, 0, 0, time.UTC),
					ClosedAt: ptrTime(time.Date(2021, 10, 1, 0, 15, 0, 0, time.UTC)),
				},
				{
					Reason:   "ALB Failures SLO:availability,quarity affected.",
					OpenedAt: time.Date(2021, 10, 1, 0, 10, 0, 0, time.UTC),
					ClosedAt: ptrTime(time.Date(2021, 10, 1, 0, 15, 0, 0, time.UTC)),
				},
			},
		},
	}
	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			vAlerts, err := repo.FetchVirtualAlerts(context.Background(), c.serviceName, c.sloID, c.startAt, c.endAt)
			require.NoError(t, err)
			require.EqualValues(t, c.expected, vAlerts)
		})
	}
}

func TestRepositoryFetchAlerts(t *testing.T) {
	client := newMockMackerelClient(t)
	repo := shimesaba.NewRepository(client)

	cases := []struct {
		name     string
		startAt  time.Time
		endAt    time.Time
		expected shimesaba.Alerts
	}{
		{
			name:    "Alerts service",
			startAt: time.Date(2021, 10, 1, 0, 5, 0, 0, time.UTC),
			endAt:   time.Date(2021, 10, 1, 0, 15, 0, 0, time.UTC),
			expected: shimesaba.Alerts{
				{
					OpenedAt: time.Date(2021, 10, 1, 0, 10, 0, 0, time.UTC),
					Monitor:  shimesaba.NewMonitor("dummyMonitorID", "Dummy Service Metric Monitor", "service"),
					ClosedAt: ptrTime(time.Date(2021, 10, 1, 0, 15, 0, 0, time.UTC)),
				},
			},
		},
		{
			name:     "No alerts",
			startAt:  time.Date(2022, 10, 1, 0, 5, 0, 0, time.UTC),
			endAt:    time.Date(2022, 10, 1, 0, 15, 0, 0, time.UTC),
			expected: shimesaba.Alerts{},
		},
		{
			name:    "check monitor",
			startAt: time.Date(2021, 10, 1, 0, 17, 0, 0, time.UTC),
			endAt:   time.Date(2021, 10, 1, 0, 18, 0, 0, time.UTC),
			expected: shimesaba.Alerts{
				{
					OpenedAt: time.Date(2021, 10, 1, 0, 17, 0, 0, time.UTC),
					Monitor:  shimesaba.NewMonitor("dummyCheckMonitorID", "check monitor dummyCheckMonitorID", "check"),
					ClosedAt: nil,
				},
			},
		},
	}
	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			alerts, err := repo.FetchAlerts(context.Background(), c.startAt, c.endAt)
			require.NoError(t, err)
			for _, a := range alerts {
				if a.Monitor != nil {
					a.Monitor = a.Monitor.WithEvaluator(nil)
				}
			}
			require.EqualValues(t, c.expected, alerts)
		})
	}
}