incident_test.go

//go:build unit
// +build unit

package tools

import (
	"context"
	"testing"

	"github.com/grafana/incident-go"
	mcpgrafana "github.com/grafana/mcp-grafana"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func newIncidentTestContext() context.Context {
	client := incident.NewTestClient()
	return mcpgrafana.WithIncidentClient(context.Background(), client)
}

func TestIncidentTools(t *testing.T) {
	t.Run("list incidents", func(t *testing.T) {
		ctx := newIncidentTestContext()
		result, err := listIncidents(ctx, ListIncidentsParams{
			Limit: 2,
		})
		require.NoError(t, err)
		assert.Len(t, result.IncidentPreviews, 2)
	})

	t.Run("create incident", func(t *testing.T) {
		ctx := newIncidentTestContext()
		result, err := createIncident(ctx, CreateIncidentParams{
			Title:         "high latency in web requests",
			Severity:      "minor",
			RoomPrefix:    "test",
			IsDrill:       true,
			Status:        "active",
			AttachCaption: "Test attachment",
			AttachURL:     "https://grafana.com",
		})
		require.NoError(t, err)
		assert.Equal(t, "high latency in web requests", result.Title)
		assert.Equal(t, "minor", result.Severity)
		assert.True(t, result.IsDrill)
		assert.Equal(t, "active", result.Status)
	})

	t.Run("add activity to incident", func(t *testing.T) {
		ctx := newIncidentTestContext()
		result, err := addActivityToIncident(ctx, AddActivityToIncidentParams{
			IncidentID: "123",
			Body:       "The incident was created by user-123",
			EventTime:  "2021-08-07T11:58:23Z",
		})
		require.NoError(t, err)
		assert.Equal(t, "The incident was created by user-123", result.Body)
		assert.Equal(t, "2021-08-07T11:58:23Z", result.EventTime)
	})
}