package company

import (
	"testing"

	"tm/pkg/ai_summarizer"
)

func TestIsPipelineDailyRunInProgress(t *testing.T) {
	tests := []struct {
		name   string
		report *ai_summarizer.PipelineReportResponse
		want   bool
	}{
		{name: "nil report", report: nil, want: true},
		{name: "empty status", report: &ai_summarizer.PipelineReportResponse{}, want: true},
		{name: "running", report: &ai_summarizer.PipelineReportResponse{Status: "running"}, want: true},
		{name: "in progress", report: &ai_summarizer.PipelineReportResponse{Status: "in_progress"}, want: true},
		{name: "completed", report: &ai_summarizer.PipelineReportResponse{Status: "completed"}, want: false},
		{name: "failed", report: &ai_summarizer.PipelineReportResponse{Status: "failed"}, want: false},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := isPipelineDailyRunInProgress(tt.report); got != tt.want {
				t.Fatalf("isPipelineDailyRunInProgress() = %v, want %v", got, tt.want)
			}
		})
	}
}