nomadcode/services/core/internal/notification/service_test.go

package notification_test

import (
	"context"
	"errors"
	"log/slog"
	"testing"

	"github.com/nomadcode/nomadcode-core/internal/notification"
)

type recordingSink struct {
	events []notification.TaskEvent
	err    error
}

func (s *recordingSink) HandleTaskEvent(_ context.Context, event notification.TaskEvent) error {
	s.events = append(s.events, event)
	return s.err
}

type logRecord struct {
	msg  string
	args map[string]any
}

type spyHandler struct {
	records []logRecord
}

func (h *spyHandler) Enabled(ctx context.Context, level slog.Level) bool {
	return true
}

func (h *spyHandler) Handle(ctx context.Context, r slog.Record) error {
	args := make(map[string]any)
	r.Attrs(func(a slog.Attr) bool {
		args[a.Key] = a.Value.Any()
		return true
	})
	h.records = append(h.records, logRecord{
		msg:  r.Message,
		args: args,
	})
	return nil
}

func (h *spyHandler) WithAttrs(attrs []slog.Attr) slog.Handler {
	return h
}

func (h *spyHandler) WithGroup(name string) slog.Handler {
	return h
}

func TestNotifyTaskCompletedUsesTaskEventCompleted(t *testing.T) {
	spy := &spyHandler{}
	logger := slog.New(spy)

	service := notification.NewService(nil, logger)

	err := service.NotifyTaskCompleted(context.Background(), notification.TaskNotification{
		TaskID:  "task-123",
		Title:   "Test Task",
		Status:  "completed",
		Message: "Finished successfully",
	})
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}

	if len(spy.records) == 0 {
		t.Fatal("expected log records, got none")
	}

	found := false
	for _, rec := range spy.records {
		if rec.msg == "task event notification requested" {
			found = true
			if rec.args["type"] != notification.TaskEventCompleted {
				t.Errorf("expected event type %s, got %v", notification.TaskEventCompleted, rec.args["type"])
			}
			if rec.args["task_id"] != "task-123" {
				t.Errorf("expected task_id task-123, got %v", rec.args["task_id"])
			}
		}
	}
	if !found {
		t.Error("expected 'task event notification requested' log message, not found")
	}
}

func TestNotifyTaskEventFansOutToSinksForRunningAndFailed(t *testing.T) {
	logger := slog.New(&spyHandler{})
	sink := &recordingSink{}
	service := notification.NewService(nil, logger, sink)

	events := []notification.TaskEvent{
		{Type: notification.TaskEventRunning, TaskID: "task-1", Status: "running"},
		{Type: notification.TaskEventFailed, TaskID: "task-2", Status: "failed"},
		{Type: notification.TaskEventCompleted, TaskID: "task-3", Status: "completed"},
	}
	for _, e := range events {
		if err := service.NotifyTaskEvent(context.Background(), e); err != nil {
			t.Fatalf("unexpected error for %s: %v", e.Type, err)
		}
	}

	if len(sink.events) != 3 {
		t.Fatalf("expected sink to receive 3 events, got %d", len(sink.events))
	}
	if sink.events[0].Type != notification.TaskEventRunning || sink.events[1].Type != notification.TaskEventFailed {
		t.Errorf("sink did not receive non-completed events: %+v", sink.events)
	}
	if sink.events[2].Type != notification.TaskEventCompleted {
		t.Errorf("sink did not receive completed event: %+v", sink.events)
	}
}

func TestNotifyTaskEventReturnsSinkErrors(t *testing.T) {
	logger := slog.New(&spyHandler{})
	sinkErr := errors.New("sink boom")
	sink := &recordingSink{err: sinkErr}
	service := notification.NewService(nil, logger, sink)

	err := service.NotifyTaskEvent(context.Background(), notification.TaskEvent{
		Type:   notification.TaskEventRunning,
		TaskID: "task-1",
		Status: "running",
	})
	if err == nil {
		t.Fatal("expected sink error to be returned")
	}
	if !errors.Is(err, sinkErr) {
		t.Errorf("expected joined error to wrap sink error, got %v", err)
	}
}

func TestNotifyTaskEventAllowsFailedWithoutMattermost(t *testing.T) {
	spy := &spyHandler{}
	logger := slog.New(spy)

	service := notification.NewService(nil, logger)

	err := service.NotifyTaskEvent(context.Background(), notification.TaskEvent{
		Type:    notification.TaskEventFailed,
		TaskID:  "task-456",
		Title:   "Failed Task",
		Status:  "failed",
		Message: "Something went wrong",
	})
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}

	found := false
	for _, rec := range spy.records {
		if rec.msg == "task event notification requested" {
			found = true
			if rec.args["type"] != notification.TaskEventFailed {
				t.Errorf("expected event type %s, got %v", notification.TaskEventFailed, rec.args["type"])
			}
			if rec.args["task_id"] != "task-456" {
				t.Errorf("expected task_id task-456, got %v", rec.args["task_id"])
			}
		}
	}
	if !found {
		t.Error("expected 'task event notification requested' log message, not found")
	}
}