package report_test

import (
	"bytes"
	"encoding/csv"
	"encoding/json"
	"os"
	"path/filepath"
	"regexp"
	"slices"
	"strings"
	"testing"
	"testing/synctest"
	"time"

	"github.com/gruntwork-io/terragrunt/internal/report"
	"github.com/gruntwork-io/terragrunt/pkg/log"
	"github.com/gruntwork-io/terragrunt/test/helpers"
	"github.com/gruntwork-io/terragrunt/test/helpers/logger"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"github.com/xeipuuv/gojsonschema"
)

func TestNewReport(t *testing.T) {
	t.Parallel()

	report := report.NewReport()
	assert.NotNil(t, report)
	assert.NotNil(t, report.Runs)
	assert.Empty(t, report.Runs)
}

func TestNewRun(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	path := filepath.Join(tmp, "test-run")
	run := newRun(t, path)
	assert.NotNil(t, run)
	assert.Equal(t, path, run.Path)
	assert.False(t, run.Started.IsZero())
	assert.True(t, run.Ended.IsZero())
	assert.Empty(t, run.Result)
	assert.Nil(t, run.Reason)
	assert.Nil(t, run.Cause)
}

func TestAddRun(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	l := logger.CreateLogger()

	path := filepath.Join(tmp, "test-run")

	r := report.NewReport()

	err := r.AddRun(l, newRun(t, path))
	require.NoError(t, err)
	assert.Len(t, r.Runs, 1)

	err = r.AddRun(l, newRun(t, path))
	require.Error(t, err)
	assert.ErrorIs(t, err, report.ErrRunAlreadyExists)
}

func TestGetRun(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	l := logger.CreateLogger()

	r := report.NewReport()
	run := newRun(t, filepath.Join(tmp, "test-run"))
	r.AddRun(l, run)

	tests := []struct {
		expectedErr error
		name        string
		runName     string
	}{
		{
			name:    "existing run",
			runName: filepath.Join(tmp, "test-run"),
		},
		{
			name:        "non-existent run",
			runName:     filepath.Join(tmp, "non-existent"),
			expectedErr: report.ErrRunNotFound,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			run, err := r.GetRun(tt.runName)
			if tt.expectedErr != nil {
				assert.ErrorIs(t, err, tt.expectedErr)
			} else {
				require.NoError(t, err)
				assert.Equal(t, tt.runName, run.Path)
			}
		})
	}
}

func TestEnsureRun(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	l := logger.CreateLogger()

	tests := []struct {
		expectedErrIs error
		setupFunc     func(*report.Report) *report.Run
		name          string
		runName       string
		existingRun   bool
		expectError   bool
	}{
		{
			name:        "creates new run when run does not exist",
			runName:     filepath.Join(tmp, "new-run"),
			existingRun: false,
			expectError: false,
		},
		{
			name:        "returns existing run when it exists",
			runName:     filepath.Join(tmp, "existing-run"),
			existingRun: true,
			expectError: false,
			setupFunc: func(r *report.Report) *report.Run {
				run := newRun(t, filepath.Join(tmp, "existing-run"))
				err := r.AddRun(l, run)
				require.NoError(t, err)

				return run
			},
		},
		{
			name:          "returns error for invalid path",
			runName:       "relative-path",
			existingRun:   false,
			expectError:   true,
			expectedErrIs: report.ErrPathMustBeAbsolute,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			r := report.NewReport()

			var existingRun *report.Run

			if tt.setupFunc != nil {
				existingRun = tt.setupFunc(r)
			}

			run, err := r.EnsureRun(l, tt.runName)

			if tt.expectError {
				require.Error(t, err)
				assert.Nil(t, run)

				if tt.expectedErrIs != nil {
					require.ErrorIs(t, err, tt.expectedErrIs)
				}
			} else {
				require.NoError(t, err)
				require.NotNil(t, run)
				assert.Equal(t, tt.runName, run.Path)
				assert.False(t, run.Started.IsZero())

				if tt.existingRun {
					// Should return the same instance as the existing run
					assert.Equal(t, existingRun.Started, run.Started)
				}

				// Verify the run was added to the report
				retrievedRun, err := r.GetRun(tt.runName)
				require.NoError(t, err)
				assert.Equal(t, run, retrievedRun)

				// Verify that calling EnsureRun again returns the same run
				secondRun, err := r.EnsureRun(l, tt.runName)
				require.NoError(t, err)
				assert.Equal(t, run, secondRun)
			}
		})
	}
}

func TestEndRun(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	l := logger.CreateLogger()

	tests := []struct {
		wantReason *report.Reason
		wantCause  *report.Cause
		name       string
		runName    string
		wantResult report.Result
		options    []report.EndOption
		wantErr    bool
	}{
		{
			name:       "successful end",
			runName:    filepath.Join(tmp, "test-run"),
			options:    []report.EndOption{},
			wantErr:    false,
			wantResult: report.ResultSucceeded,
		},
		{
			name:    "non-existent run",
			runName: filepath.Join(tmp, "non-existent"),
			options: []report.EndOption{},
			wantErr: true,
		},
		{
			name:       "with result",
			runName:    filepath.Join(tmp, "with-result"),
			options:    []report.EndOption{report.WithResult(report.ResultFailed)},
			wantErr:    false,
			wantResult: report.ResultFailed,
		},
		{
			name:       "with reason",
			runName:    filepath.Join(tmp, "with-reason"),
			options:    []report.EndOption{report.WithReason(report.ReasonRunError)},
			wantErr:    false,
			wantResult: report.ResultSucceeded,
			wantReason: func() *report.Reason { r := report.ReasonRunError; return &r }(),
		},
		{
			name:       "with cause",
			runName:    filepath.Join(tmp, "with-cause"),
			options:    []report.EndOption{report.WithCauseRetryBlock("test-block")},
			wantErr:    false,
			wantResult: report.ResultSucceeded,
			wantCause:  func() *report.Cause { c := report.Cause("test-block"); return &c }(),
		},
	}

	r := report.NewReport()

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			if !tt.wantErr {
				run := newRun(t, tt.runName)
				r.AddRun(l, run)
			}

			err := r.EndRun(l, tt.runName, tt.options...)
			if tt.wantErr {
				require.Error(t, err)
			} else {
				require.NoError(t, err)

				run, err := r.GetRun(tt.runName)
				require.NoError(t, err)

				assert.Equal(t, tt.wantResult, run.Result)

				if tt.wantReason != nil {
					assert.NotNil(t, run.Reason)
					assert.Equal(t, *tt.wantReason, *run.Reason)
				}

				if tt.wantCause != nil {
					assert.NotNil(t, run.Cause)
					assert.Equal(t, *tt.wantCause, *run.Cause)
				}

				assert.False(t, run.Ended.IsZero())
			}
		})
	}
}

func TestEndRunAlreadyEnded(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	l := logger.CreateLogger()

	tests := []struct {
		name           string
		initialResult  report.Result
		expectedResult report.Result
		secondResult   report.Result
		initialOptions []report.EndOption
		secondOptions  []report.EndOption
	}{
		{
			name:           "already ended with early exit is not overwritten",
			initialResult:  report.ResultEarlyExit,
			secondResult:   report.ResultSucceeded,
			expectedResult: report.ResultEarlyExit,
		},
		{
			name:           "already ended with excluded is not overwritten",
			initialResult:  report.ResultExcluded,
			secondResult:   report.ResultSucceeded,
			expectedResult: report.ResultExcluded,
		},
		{
			name:           "already ended with retry succeeded is overwritten",
			initialResult:  report.ResultSucceeded,
			initialOptions: []report.EndOption{report.WithReason(report.ReasonRetrySucceeded)},
			secondResult:   report.ResultSucceeded,
			expectedResult: report.ResultSucceeded,
		},
		{
			name:           "already ended with retry failed is overwritten",
			initialResult:  report.ResultSucceeded,
			initialOptions: []report.EndOption{report.WithReason(report.ReasonRetrySucceeded)},
			secondResult:   report.ResultFailed,
			expectedResult: report.ResultFailed,
		},
		{
			name:           "already ended with error ignored is overwritten",
			initialResult:  report.ResultSucceeded,
			initialOptions: []report.EndOption{report.WithReason(report.ReasonErrorIgnored)},
			secondResult:   report.ResultSucceeded,
			expectedResult: report.ResultSucceeded,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			// Create a new report and run for each test case
			r := report.NewReport()
			runName := filepath.Join(tmp, tt.name)
			run := newRun(t, runName)
			r.AddRun(l, run)

			// Set up initial options with the initial result
			initialOptions := slices.Concat(tt.initialOptions, []report.EndOption{report.WithResult(tt.initialResult)})

			// End the run with the initial state
			err := r.EndRun(l, runName, initialOptions...)
			require.NoError(t, err)

			// Set up second options with the second result
			secondOptions := slices.Concat(tt.secondOptions, []report.EndOption{report.WithResult(tt.secondResult)})

			// Then try to end it again with a different state
			err = r.EndRun(l, runName, secondOptions...)
			require.NoError(t, err)

			// Verify that the result is the expected one
			endedRun, err := r.GetRun(runName)
			require.NoError(t, err)
			assert.Equal(t, tt.expectedResult, endedRun.Result)
		})
	}
}

func TestSummarize(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	l := logger.CreateLogger()

	tests := []struct {
		name    string
		results []struct {
			name   string
			result report.Result
		}
		wantTotalUnits int
		wantSucceeded  int
		wantFailed     int
		wantEarlyExits int
		wantExcluded   int
	}{
		{
			name: "empty report",
			results: []struct {
				name   string
				result report.Result
			}{},
			wantTotalUnits: 0,
		},
		{
			name: "single successful run",
			results: []struct {
				name   string
				result report.Result
			}{
				{filepath.Join(tmp, "single-successful-run"), report.ResultSucceeded},
			},
			wantTotalUnits: 1,
			wantSucceeded:  1,
		},
		{
			name: "mixed results",
			results: []struct {
				name   string
				result report.Result
			}{
				{filepath.Join(tmp, "successful-run"), report.ResultSucceeded},
				{filepath.Join(tmp, "failed-run"), report.ResultFailed},
				{filepath.Join(tmp, "early-exit-run"), report.ResultEarlyExit},
				{filepath.Join(tmp, "excluded-run"), report.ResultExcluded},
			},
			wantTotalUnits: 4,
			wantSucceeded:  1,
			wantFailed:     1,
			wantEarlyExits: 1,
			wantExcluded:   1,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			r := report.NewReport()

			for _, result := range tt.results {
				run := newRun(t, result.name)
				r.AddRun(l, run)
				r.EndRun(l, result.name, report.WithResult(result.result))
			}

			summary := r.Summarize()
			assert.Equal(t, tt.wantTotalUnits, summary.TotalUnits())
			assert.Equal(t, tt.wantSucceeded, summary.UnitsSucceeded)
			assert.Equal(t, tt.wantFailed, summary.UnitsFailed)
			assert.Equal(t, tt.wantEarlyExits, summary.EarlyExits)
			assert.Equal(t, tt.wantExcluded, summary.Excluded)
		})
	}
}

func TestWriteCSV(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name     string
		setup    func(l log.Logger, dir string, r *report.Report)
		expected [][]string
	}{
		{
			name: "single successful run",
			setup: func(l log.Logger, dir string, r *report.Report) {
				run := newRun(t, filepath.Join(dir, "successful-run"))
				r.AddRun(l, run)
				r.EndRun(l, run.Path)
			},
			expected: [][]string{
				{"Name", "Started", "Ended", "Result", "Reason", "Cause", "Ref", "Cmd", "Args"},
				{"successful-run", "", "", "succeeded", "", "", "", "", ""},
			},
		},
		{
			name: "complex mixed results",
			setup: func(l log.Logger, dir string, r *report.Report) {
				// Add successful run
				successRun := newRun(t, filepath.Join(dir, "success-run"))
				r.AddRun(l, successRun)
				r.EndRun(l, successRun.Path)

				// Add failed run with reason
				failedRun := newRun(t, filepath.Join(dir, "failed-run"))
				r.AddRun(l, failedRun)
				r.EndRun(l, failedRun.Path, report.WithResult(report.ResultFailed), report.WithReason(report.ReasonRunError))

				// Add excluded run with cause
				excludedRun := newRun(t, filepath.Join(dir, "excluded-run"))
				r.AddRun(l, excludedRun)
				r.EndRun(l, excludedRun.Path, report.WithResult(report.ResultExcluded), report.WithCauseRetryBlock("test-block"))

				// Add early exit run with both reason and cause
				earlyExitRun := newRun(t, filepath.Join(dir, "early-exit-run"))
				r.AddRun(l, earlyExitRun)
				r.EndRun(l, earlyExitRun.Path,
					report.WithResult(report.ResultEarlyExit),
					report.WithReason(report.ReasonRunError),
					report.WithCauseRetryBlock("another-block"),
				)
			},
			expected: [][]string{
				{"Name", "Started", "Ended", "Result", "Reason", "Cause", "Ref", "Cmd", "Args"},
				{"success-run", "", "", "succeeded", "", "", "", "", ""},
				{"failed-run", "", "", "failed", "run error", "", "", "", ""},
				{"excluded-run", "", "", "excluded", "", "test-block", "", "", ""},
				{"early-exit-run", "", "", "early exit", "run error", "another-block", "", "", ""},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			tmp := helpers.TmpDirWOSymlinks(t)

			l := logger.CreateLogger()

			// Create a temporary file for the CSV
			csvFile := filepath.Join(tmp, "report.csv")
			file, err := os.Create(csvFile)
			require.NoError(t, err)

			defer file.Close()

			// Setup and write the report
			r := report.NewReport().WithWorkingDir(tmp)
			tt.setup(l, tmp, r)

			err = r.WriteCSV(file)
			require.NoError(t, err)

			// Close the file before reading
			file.Close()

			// Read the CSV file
			file, err = os.Open(csvFile)
			require.NoError(t, err)

			defer file.Close()

			reader := csv.NewReader(file)
			records, err := reader.ReadAll()
			require.NoError(t, err)

			// Verify the number of records
			require.Len(t, records, len(tt.expected))

			// Verify each record
			for i, record := range records {
				expected := tt.expected[i]
				require.Len(t, record, len(expected), "Record %d has wrong number of fields", i)

				// For the header row, verify exact match
				if i == 0 {
					assert.Equal(t, expected, record)
					continue
				}

				// For data rows, verify fields individually
				assert.Equal(t, expected[0], record[0], "Name mismatch in record %d", i)
				// Skip timestamp verification for Started and Ended fields
				assert.Equal(t, expected[3], record[3], "Result mismatch in record %d", i)
				assert.Equal(t, expected[4], record[4], "Reason mismatch in record %d", i)
				assert.Equal(t, expected[5], record[5], "Cause mismatch in record %d", i)
				assert.Equal(t, expected[6], record[6], "Ref mismatch in record %d", i)
				assert.Equal(t, expected[7], record[7], "Cmd mismatch in record %d", i)
				assert.Equal(t, expected[8], record[8], "Args mismatch in record %d", i)

				// Verify that timestamps are in RFC3339 format
				if record[1] != "" {
					_, err := time.Parse(time.RFC3339, record[1])
					require.NoError(t, err, "Started timestamp in record %d is not in RFC3339 format", i)
				}

				if record[2] != "" {
					_, err := time.Parse(time.RFC3339, record[2])
					require.NoError(t, err, "Ended timestamp in record %d is not in RFC3339 format", i)
				}
			}
		})
	}
}

func TestWriteJSON(t *testing.T) {
	t.Parallel()

	l := logger.CreateLogger()

	tests := []struct {
		name     string
		setup    func(l log.Logger, dir string, r *report.Report)
		expected string
	}{
		{
			name: "single successful run",
			setup: func(l log.Logger, dir string, r *report.Report) {
				run := newRun(t, filepath.Join(dir, "successful-run"))
				r.AddRun(l, run)
				r.EndRun(l, run.Path)
			},
			expected: `[
  {
    "Name": "successful-run",
    "Started": "2024-03-21T10:00:00Z",
    "Ended": "2024-03-21T10:01:00Z",
    "Result": "succeeded"
  }
]`,
		},
		{
			name: "complex mixed results",
			setup: func(l log.Logger, dir string, r *report.Report) {
				// Add successful run
				successRun := newRun(t, filepath.Join(dir, "success-run"))
				r.AddRun(l, successRun)
				r.EndRun(l, successRun.Path)

				// Add failed run with reason
				failedRun := newRun(t, filepath.Join(dir, "failed-run"))
				r.AddRun(l, failedRun)
				r.EndRun(
					l,
					failedRun.Path,
					report.WithResult(report.ResultFailed),
					report.WithReason(report.ReasonRunError),
				)

				// Add excluded run with cause
				retriedRun := newRun(t, filepath.Join(dir, "retried-run"))
				r.AddRun(l, retriedRun)
				r.EndRun(
					l,
					retriedRun.Path,
					report.WithResult(report.ResultSucceeded),
					report.WithReason(report.ReasonRetrySucceeded),
				)

				// Add excluded run with cause
				excludedRun := newRun(t, filepath.Join(dir, "excluded-run"))
				r.AddRun(l, excludedRun)
				r.EndRun(
					l,
					excludedRun.Path,
					report.WithResult(report.ResultExcluded),
					report.WithReason(report.ReasonExcludeBlock),
					report.WithCauseExcludeBlock("test-block"),
				)
			},
			expected: `[
  {
    "Name": "success-run",
    "Started": "2024-03-21T10:00:00Z",
    "Ended": "2024-03-21T10:01:00Z",
    "Result": "succeeded"
  },
  {
    "Name": "failed-run",
    "Started": "2024-03-21T10:01:00Z",
    "Ended": "2024-03-21T10:02:00Z",
    "Result": "failed",
    "Reason": "run error"
  },
  {
    "Name": "retried-run",
    "Started": "2024-03-21T10:03:00Z",
    "Ended": "2024-03-21T10:04:00Z",
    "Result": "succeeded",
    "Reason": "retry succeeded"
  },
  {
    "Name": "excluded-run",
    "Started": "2024-03-21T10:02:00Z",
    "Ended": "2024-03-21T10:03:00Z",
    "Result": "excluded",
    "Reason": "exclude block",
    "Cause": "test-block"
  }
]`,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			tmp := helpers.TmpDirWOSymlinks(t)

			// Create a temporary file for the JSON
			jsonFile := filepath.Join(tmp, "report.json")
			file, err := os.Create(jsonFile)
			require.NoError(t, err)

			defer file.Close()

			// Setup and write the report
			r := report.NewReport().WithWorkingDir(tmp)
			tt.setup(l, tmp, r)

			err = r.WriteJSON(file)
			require.NoError(t, err)

			// Close the file before reading
			file.Close()

			// Read the JSON file
			file, err = os.Open(jsonFile)
			require.NoError(t, err)

			defer file.Close()

			// Read the actual output
			actualBytes, err := os.ReadFile(jsonFile)
			require.NoError(t, err)

			// Parse both expected and actual JSON to compare them
			var expectedJSON, actualJSON []map[string]any

			err = json.Unmarshal([]byte(tt.expected), &expectedJSON)
			require.NoError(t, err)
			err = json.Unmarshal(actualBytes, &actualJSON)
			require.NoError(t, err)

			// Verify the number of records
			require.Len(t, actualJSON, len(expectedJSON))

			// Verify each record
			for i, actualRecord := range actualJSON {
				expectedRecord := expectedJSON[i]

				// Verify name
				assert.Equal(t, expectedRecord["Name"], actualRecord["Name"], "Name mismatch in record %d", i)

				// Verify result
				assert.Equal(t, expectedRecord["Result"], actualRecord["Result"], "Result mismatch in record %d", i)

				// Verify reason if present
				if expectedReason, ok := expectedRecord["Reason"]; ok {
					assert.Equal(t, expectedReason, actualRecord["Reason"], "Reason mismatch in record %d", i)
				} else {
					assert.NotContains(t, actualRecord, "Reason", "Unexpected reason in record %d", i)
				}

				// Verify cause if present
				if expectedCause, ok := expectedRecord["Cause"]; ok {
					assert.Equal(t, expectedCause, actualRecord["Cause"], "Cause mismatch in record %d", i)
				} else {
					assert.NotContains(t, actualRecord, "Cause", "Unexpected cause in record %d", i)
				}

				// Verify timestamps are in RFC3339 format
				if started, ok := actualRecord["Started"].(string); ok {
					_, err := time.Parse(time.RFC3339, started)
					require.NoError(t, err, "Started timestamp in record %d is not in RFC3339 format", i)
				}

				if ended, ok := actualRecord["Ended"].(string); ok {
					_, err := time.Parse(time.RFC3339, ended)
					require.NoError(t, err, "Ended timestamp in record %d is not in RFC3339 format", i)
				}
			}
		})
	}
}

const ExpectedSchema = `{
  "$schema": "https://json-schema.org/draft/2020-12/schema",
  "$id": "https://docs.terragrunt.com/schemas/run/report/v4/schema.json",
  "items": {
    "properties": {
      "Started": {
        "type": "string",
        "format": "date-time"
      },
      "Ended": {
        "type": "string",
        "format": "date-time"
      },
      "Reason": {
        "type": "string",
        "enum": [
          "retry succeeded",
          "error ignored",
          "run error",
          "exclude block",
          "ancestor error"
        ]
      },
      "Cause": {
        "type": "string"
      },
      "Name": {
        "type": "string"
      },
      "Result": {
        "type": "string",
        "enum": [
          "succeeded",
          "failed",
          "early exit",
          "excluded"
        ]
      },
      "Ref": {
        "type": "string"
      },
      "Cmd": {
        "type": "string"
      },
      "Args": {
        "items": {
          "type": "string"
        },
        "type": "array"
      }
    },
    "additionalProperties": false,
    "type": "object",
    "required": [
      "Started",
      "Ended",
      "Name",
      "Result"
    ],
    "title": "Terragrunt Run Report Schema",
    "description": "Schema for Terragrunt run report"
  },
  "type": "array",
  "title": "Terragrunt Run Report Schema",
  "description": "Array of Terragrunt runs"
}
`

func TestWriteSchema(t *testing.T) {
	t.Parallel()

	// Create a buffer to write the schema to
	var buf bytes.Buffer

	// Write the schema
	err := report.WriteSchema(&buf)
	require.NoError(t, err)

	// Assert the contents of the schema
	assert.JSONEq(t, ExpectedSchema, buf.String())

	// Parse the schema
	var schema map[string]any

	err = json.Unmarshal(buf.Bytes(), &schema)
	require.NoError(t, err)

	// Verify the schema structure
	assert.Equal(t, "array", schema["type"])
	assert.Equal(t, "Array of Terragrunt runs", schema["description"])
	assert.Equal(t, "Terragrunt Run Report Schema", schema["title"])

	// Verify the items schema
	items, ok := schema["items"].(map[string]any)
	require.True(t, ok)

	// Verify the properties
	properties, ok := items["properties"].(map[string]any)
	require.True(t, ok)

	// Verify required fields
	required, ok := items["required"].([]any)
	require.True(t, ok)
	assert.Contains(t, required, "Name")
	assert.Contains(t, required, "Started")
	assert.Contains(t, required, "Ended")
	assert.Contains(t, required, "Result")

	// Verify field types
	assert.Equal(t, "string", properties["Name"].(map[string]any)["type"])
	assert.Equal(t, "string", properties["Result"].(map[string]any)["type"])
	assert.Equal(t, "string", properties["Started"].(map[string]any)["type"])
	assert.Equal(t, "string", properties["Ended"].(map[string]any)["type"])

	// Verify optional fields
	reason, ok := properties["Reason"].(map[string]any)
	require.True(t, ok)
	assert.Equal(t, "string", reason["type"])

	cause, ok := properties["Cause"].(map[string]any)
	require.True(t, ok)
	assert.Equal(t, "string", cause["type"])
}

func TestExpectedSchemaIsInDocs(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name string
		file string
	}{
		{
			name: "starlight",
			file: filepath.Join(
				"..",
				"..",
				"docs",
				"public",
				"schemas",
				"run",
				"report",
				"v4",
				"schema.json",
			),
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			schema, err := os.ReadFile(tt.file)
			require.NoError(t, err)

			assert.JSONEq(t, ExpectedSchema, string(schema))
		})
	}
}

func TestWriteSummary(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	l := logger.CreateLogger()

	tests := []struct {
		name     string
		setup    func(l log.Logger, r *report.Report)
		expected string
	}{
		{
			name: "single successful run",
			setup: func(l log.Logger, r *report.Report) {
				run := newRun(t, filepath.Join(tmp, "successful-run"))
				r.AddRun(l, run)
				r.EndRun(l, run.Path)
			},
			expected: `
❯❯ Run Summary  1 units  x
   ────────────────────────────
   Succeeded    1
`,
		},
		{
			name: "complex mixed results",
			setup: func(l log.Logger, r *report.Report) {
				// Add successful runs
				firstSuccessfulRun := newRun(t, filepath.Join(tmp, "first-successful-run"))
				r.AddRun(l, firstSuccessfulRun)
				r.EndRun(l, firstSuccessfulRun.Path)

				secondSuccessfulRun := newRun(t, filepath.Join(tmp, "second-successful-run"))
				r.AddRun(l, secondSuccessfulRun)
				r.EndRun(l, secondSuccessfulRun.Path)

				// Add failed runs
				firstFailedRun := newRun(t, filepath.Join(tmp, "first-failed-run"))
				r.AddRun(l, firstFailedRun)
				r.EndRun(l, firstFailedRun.Path, report.WithResult(report.ResultFailed))

				secondFailedRun := newRun(t, filepath.Join(tmp, "second-failed-run"))
				r.AddRun(l, secondFailedRun)
				r.EndRun(l, secondFailedRun.Path, report.WithResult(report.ResultFailed))

				// Add excluded runs
				firstExcludedRun := newRun(t, filepath.Join(tmp, "first-excluded-run"))
				r.AddRun(l, firstExcludedRun)
				r.EndRun(l, firstExcludedRun.Path, report.WithResult(report.ResultExcluded))

				secondExcludedRun := newRun(t, filepath.Join(tmp, "second-excluded-run"))
				r.AddRun(l, secondExcludedRun)
				r.EndRun(l, secondExcludedRun.Path, report.WithResult(report.ResultExcluded))

				// Add early exit runs
				firstEarlyExitRun := newRun(t, filepath.Join(tmp, "first-early-exit-run"))
				r.AddRun(l, firstEarlyExitRun)
				r.EndRun(l, firstEarlyExitRun.Path, report.WithResult(report.ResultEarlyExit))

				secondEarlyExitRun := newRun(t, filepath.Join(tmp, "second-early-exit-run"))
				r.AddRun(l, secondEarlyExitRun)
				r.EndRun(l, secondEarlyExitRun.Path, report.WithResult(report.ResultEarlyExit))
			},
			expected: `
❯❯ Run Summary  8 units  x
   ────────────────────────────
   Succeeded    2
   Failed       2
   Early Exits  2
   Excluded     2
`,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			r := report.NewReport().WithDisableColor()
			tt.setup(l, r)

			var buf bytes.Buffer

			err := r.WriteSummary(&buf)
			require.NoError(t, err)

			output := buf.String()

			// Replace the duration in the header with x
			// Pattern matches: "❯❯ Run Summary  8 units  42µs" -> "❯❯ Run Summary  8 units  x"
			re := regexp.MustCompile(`(❯❯ Run Summary\s+\d+\s+units\s+)[^\n]+`)
			output = re.ReplaceAllString(output, "${1}x")

			expected := strings.TrimSpace(tt.expected)
			assert.Equal(t, expected, strings.TrimSpace(output))
		})
	}
}

func TestSchemaIsValid(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	l := logger.CreateLogger()

	// Create a new report with working directory
	r := report.NewReport().WithWorkingDir(tmp)

	// Add a simple run that succeeds
	simpleRun := newRun(t, filepath.Join(tmp, "simple-run"))
	r.AddRun(l, simpleRun)
	r.EndRun(l, simpleRun.Path,
		report.WithResult(report.ResultSucceeded),
	)

	// Add a complex run that tests all possible fields and states
	complexRun := newRun(t, filepath.Join(tmp, "complex-run"))
	r.AddRun(l, complexRun)
	r.EndRun(l, complexRun.Path,
		report.WithResult(report.ResultFailed),
		report.WithReason(report.ReasonRunError),
		report.WithCauseAncestorExit("some-error"),
	)

	// Create an excluded run with exclude block
	excludedRun := newRun(t, filepath.Join(tmp, "excluded-run"))
	r.AddRun(l, excludedRun)
	r.EndRun(l, excludedRun.Path,
		report.WithResult(report.ResultExcluded),
		report.WithReason(report.ReasonExcludeBlock),
		report.WithCauseExcludeBlock("test-block"),
	)

	// Create a retry run that succeeded
	retryRun := newRun(t, filepath.Join(tmp, "retry-run"))
	r.AddRun(l, retryRun)
	r.EndRun(l, retryRun.Path,
		report.WithResult(report.ResultSucceeded),
		report.WithReason(report.ReasonRetrySucceeded),
		report.WithCauseRetryBlock("retry-block"),
	)

	// Create an early exit run
	earlyExitRun := newRun(t, filepath.Join(tmp, "early-exit-run"))
	r.AddRun(l, earlyExitRun)
	r.EndRun(l, earlyExitRun.Path,
		report.WithResult(report.ResultEarlyExit),
		report.WithReason(report.ReasonAncestorError),
		report.WithCauseAncestorExit("parent-unit"),
	)

	// Create a run with ignored error
	ignoredRun := newRun(t, filepath.Join(tmp, "ignored-run"))
	r.AddRun(l, ignoredRun)
	r.EndRun(l, ignoredRun.Path,
		report.WithResult(report.ResultSucceeded),
		report.WithReason(report.ReasonErrorIgnored),
		report.WithCauseIgnoreBlock("ignore-block"),
	)

	// Write the report to a JSON file
	reportFile := filepath.Join(tmp, "report.json")
	file, err := os.Create(reportFile)
	require.NoError(t, err)

	defer file.Close()

	err = r.WriteJSON(file)
	require.NoError(t, err)
	file.Close()

	// Write the schema to a file
	schemaFile := filepath.Join(tmp, "schema.json")
	file, err = os.Create(schemaFile)
	require.NoError(t, err)

	defer file.Close()

	err = report.WriteSchema(file)
	require.NoError(t, err)
	file.Close()

	// Read the schema and report files
	schemaBytes, err := os.ReadFile(schemaFile)
	require.NoError(t, err)

	reportBytes, err := os.ReadFile(reportFile)
	require.NoError(t, err)

	// Create a new schema loader
	schemaLoader := gojsonschema.NewBytesLoader(schemaBytes)
	documentLoader := gojsonschema.NewBytesLoader(reportBytes)

	// Validate the report against the schema
	result, err := gojsonschema.Validate(schemaLoader, documentLoader)
	require.NoError(t, err)

	// Check if the validation was successful
	assert.True(t, result.Valid(), "JSON report does not match schema: %v", result.Errors())

	// Additional validation of the report content
	var reportData []report.JSONRun

	err = json.Unmarshal(reportBytes, &reportData)
	require.NoError(t, err)

	// Verify we have all the expected runs
	require.Len(t, reportData, 6)

	// Helper function to find a run by name
	findRun := func(name string) *report.JSONRun {
		for _, run := range reportData {
			if run.Name == name {
				return &run
			}
		}

		return nil
	}

	// Verify simple run
	simple := findRun("simple-run")
	require.NotNil(t, simple)
	assert.Equal(t, "succeeded", simple.Result)
	assert.Nil(t, simple.Reason)
	assert.Nil(t, simple.Cause)
	assert.False(t, simple.Started.IsZero())
	assert.False(t, simple.Ended.IsZero())

	// Verify complex run
	complex := findRun("complex-run")
	require.NotNil(t, complex)
	assert.Equal(t, "failed", complex.Result)
	assert.Equal(t, "run error", *complex.Reason)
	assert.Equal(t, "some-error", *complex.Cause)
	assert.False(t, complex.Started.IsZero())
	assert.False(t, complex.Ended.IsZero())

	// Verify excluded run
	excluded := findRun("excluded-run")
	require.NotNil(t, excluded)
	assert.Equal(t, "excluded", excluded.Result)
	assert.Equal(t, "exclude block", *excluded.Reason)
	assert.Equal(t, "test-block", *excluded.Cause)

	// Verify retry run
	retry := findRun("retry-run")
	require.NotNil(t, retry)
	assert.Equal(t, "succeeded", retry.Result)
	assert.Equal(t, "retry succeeded", *retry.Reason)
	assert.Equal(t, "retry-block", *retry.Cause)

	// Verify early exit run
	earlyExit := findRun("early-exit-run")
	require.NotNil(t, earlyExit)
	assert.Equal(t, "early exit", earlyExit.Result)
	assert.Equal(t, "ancestor error", *earlyExit.Reason)
	assert.Equal(t, "parent-unit", *earlyExit.Cause)

	// Verify ignored run
	ignored := findRun("ignored-run")
	require.NotNil(t, ignored)
	assert.Equal(t, "succeeded", ignored.Result)
	assert.Equal(t, "error ignored", *ignored.Reason)
	assert.Equal(t, "ignore-block", *ignored.Cause)
}

func TestWriteUnitLevelSummary(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	l := logger.CreateLogger()

	tests := []struct {
		name     string
		setup    func(l log.Logger, r *report.Report)
		expected string
	}{
		{
			name: "empty runs",
			setup: func(l log.Logger, r *report.Report) {
				// No runs added
			},
			expected: ``,
		},
		{
			name: "single run",
			setup: func(l log.Logger, r *report.Report) {
				run := newRun(t, filepath.Join(tmp, "single-run"))
				r.AddRun(l, run)
				r.EndRun(l, run.Path)
			},
			expected: `
❯❯ Run Summary  1 units  x
   ────────────────────────────
   Succeeded (1)
      single-run ....... x
`,
		},
		{
			name: "multiple runs sorted by duration",
			setup: func(l log.Logger, r *report.Report) {
				// Use syntest.Test so that we can artificially manipulate the clock for duration testing.
				synctest.Test(t, func(t *testing.T) {
					t.Helper()

					longRun := newRun(t, filepath.Join(tmp, "long-run"))
					r.AddRun(l, longRun)

					time.Sleep(1 * time.Second)

					mediumRun := newRun(t, filepath.Join(tmp, "medium-run"))
					r.AddRun(l, mediumRun)

					time.Sleep(1 * time.Second)

					shortRun := newRun(t, filepath.Join(tmp, "short-run"))
					r.AddRun(l, shortRun)

					time.Sleep(1 * time.Second)

					r.EndRun(l, shortRun.Path)

					time.Sleep(1 * time.Second)

					r.EndRun(l, mediumRun.Path)

					time.Sleep(1 * time.Second)

					r.EndRun(l, longRun.Path)
				})
			},
			expected: `
❯❯ Run Summary  3 units  x
   ────────────────────────────
   Succeeded (3)
      long-run ......... x
      medium-run ....... x
      short-run ........ x
`,
		},
		{
			name: "mixed results grouped by category",
			setup: func(l log.Logger, r *report.Report) {
				// Use syntest.Test so that we can artificially manipulate the clock for duration testing.
				synctest.Test(t, func(t *testing.T) {
					t.Helper()

					successRun1 := newRun(t, filepath.Join(tmp, "success-1"))

					time.Sleep(1 * time.Second)

					successRun2 := newRun(t, filepath.Join(tmp, "success-2"))

					time.Sleep(1 * time.Second)

					failRun := newRun(t, filepath.Join(tmp, "fail-run"))

					time.Sleep(1 * time.Second)

					excludedRun := newRun(t, filepath.Join(tmp, "excluded-run"))

					r.AddRun(l, successRun1)

					time.Sleep(1 * time.Second)

					r.AddRun(l, successRun2)

					time.Sleep(1 * time.Second)

					r.AddRun(l, failRun)

					time.Sleep(1 * time.Second)

					r.AddRun(l, excludedRun)

					time.Sleep(1 * time.Second)
					r.EndRun(l, successRun1.Path)

					time.Sleep(1 * time.Second)
					r.EndRun(l, successRun2.Path)

					time.Sleep(1 * time.Second)
					r.EndRun(l, failRun.Path, report.WithResult(report.ResultFailed))

					time.Sleep(1 * time.Second)
					r.EndRun(l, excludedRun.Path, report.WithResult(report.ResultExcluded))
				})
			},
			expected: `
❯❯ Run Summary  4 units  x
   ────────────────────────────
   Succeeded (2)
      success-1 ........ x
      success-2 ........ x
   Failed (1)
      fail-run ......... x
   Excluded (1)
      excluded-run ..... x
`,
		},
		{
			name: "very short unit names",
			setup: func(l log.Logger, r *report.Report) {
				// Use syntest.Test so that we can artificially manipulate the clock for duration testing.
				synctest.Test(t, func(t *testing.T) {
					t.Helper()

					a := newRun(t, filepath.Join(tmp, "a"))

					time.Sleep(1 * time.Second)

					b := newRun(t, filepath.Join(tmp, "b"))

					time.Sleep(1 * time.Second)

					c := newRun(t, filepath.Join(tmp, "c"))

					r.AddRun(l, a)

					time.Sleep(1 * time.Second)

					r.AddRun(l, b)

					time.Sleep(1 * time.Second)

					r.AddRun(l, c)

					time.Sleep(1 * time.Second)

					r.EndRun(l, a.Path)

					time.Sleep(1 * time.Second)

					r.EndRun(l, b.Path)

					time.Sleep(1 * time.Second)

					r.EndRun(l, c.Path)
				})
			},
			expected: `
❯❯ Run Summary  3 units  x
   ────────────────────────────
   Succeeded (3)
      a ................ x
      b ................ x
      c ................ x
`,
		},
		{
			name: "very long unit names",
			setup: func(l log.Logger, r *report.Report) {
				// Use syntest.Test so that we can artificially manipulate the clock for duration testing.
				synctest.Test(t, func(t *testing.T) {
					t.Helper()

					longName1 := newRun(t, filepath.Join(tmp, "this-is-a-very-long-name-1"))

					time.Sleep(1 * time.Second)

					longName2 := newRun(t, filepath.Join(tmp, "this-is-a-very-long-name-2"))

					time.Sleep(1 * time.Second)

					longName3 := newRun(t, filepath.Join(tmp, "this-is-a-very-long-name-3"))

					time.Sleep(1 * time.Second)

					r.AddRun(l, longName1)

					time.Sleep(1 * time.Second)

					r.AddRun(l, longName2)

					time.Sleep(1 * time.Second)

					r.AddRun(l, longName3)

					time.Sleep(1 * time.Second)

					r.EndRun(l, longName1.Path)

					time.Sleep(1 * time.Second)

					r.EndRun(l, longName2.Path)

					time.Sleep(1 * time.Second)

					r.EndRun(l, longName3.Path)
				})
			},
			expected: `
❯❯ Run Summary  3 units           x
   ───────────────────────────────────
   Succeeded (3)
      this-is-a-very-long-name-1  x
      this-is-a-very-long-name-2  x
      this-is-a-very-long-name-3  x
`,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			r := report.NewReport().
				WithDisableColor().
				WithShowUnitLevelSummary().
				WithWorkingDir(tmp)

			tt.setup(l, r)

			var buf bytes.Buffer

			err := r.WriteSummary(&buf)
			require.NoError(t, err)

			// Replace the duration with x since we can't control the actual duration in tests
			output := buf.String()

			// Replace the header duration with x
			re := regexp.MustCompile(`❯❯ Run Summary  (\d+) units(\s+)(\d+.+)`)
			output = re.ReplaceAllString(output, "❯❯ Run Summary  ${1} units${2}x")

			// Replace all the unit level summaries
			re = regexp.MustCompile(`([ ]{6})([^ ]+)( )([^ ]*)( )(\d+.+)`)
			output = re.ReplaceAllString(output, "${1}${2}${3}${4}${5}x")

			expected := strings.TrimSpace(tt.expected)
			assert.Equal(t, expected, strings.TrimSpace(output))
		})
	}
}

// TestWriteJSONWithDiscoveryWorkingDir verifies that when a run has a DiscoveryWorkingDir set,
// the report writer uses it instead of the report's workingDir for path computation.
// This is critical for worktree scenarios where units are discovered in temporary worktree directories.
func TestWriteJSONWithDiscoveryWorkingDir(t *testing.T) {
	t.Parallel()

	l := logger.CreateLogger()

	// Simulate a worktree scenario:
	// - Original working dir: /original/repo
	// - Worktree path: /tmp/terragrunt-worktree-xxx/original/repo
	// - Unit path in worktree: /tmp/terragrunt-worktree-xxx/original/repo/module/unit

	// Create temp directories
	originalRepoDir := helpers.TmpDirWOSymlinks(t)
	worktreeDir := helpers.TmpDirWOSymlinks(t)

	// Create the "unit" path in the worktree
	unitPath := filepath.Join(worktreeDir, "module", "unit")
	err := os.MkdirAll(unitPath, 0755)
	require.NoError(t, err)

	// Create a report with the original repo as working dir (simulating non-worktree scenario)
	r := report.NewReport().WithWorkingDir(originalRepoDir)

	// Add a run that was discovered in the worktree
	// Set the DiscoveryWorkingDir to the worktree path
	run := newRun(t, unitPath)
	r.AddRun(l, run)

	// Ensure the run and set the DiscoveryWorkingDir
	r.EnsureRun(l, unitPath, report.WithDiscoveryWorkingDir(worktreeDir))

	// End the run
	r.EndRun(l, unitPath, report.WithResult(report.ResultSucceeded))

	// Write JSON and verify the name is relative to DiscoveryWorkingDir, not report.workingDir
	var buf bytes.Buffer

	err = r.WriteJSON(&buf)
	require.NoError(t, err)

	// Parse the JSON
	var runs []report.JSONRun

	err = json.Unmarshal(buf.Bytes(), &runs)
	require.NoError(t, err)

	require.Len(t, runs, 1)

	// The name should be relative to the worktree dir, not the original repo dir
	// Without the fix, this would be the full absolute path since unitPath doesn't start with originalRepoDir
	assert.Equal(t, "module/unit", runs[0].Name,
		"Run name should be relative to DiscoveryWorkingDir, not report.workingDir")
}

// TestWriteCSVWithDiscoveryWorkingDir verifies that CSV output also uses DiscoveryWorkingDir.
func TestWriteCSVWithDiscoveryWorkingDir(t *testing.T) {
	t.Parallel()

	l := logger.CreateLogger()

	// Create temp directories (simulating worktree scenario)
	originalRepoDir := helpers.TmpDirWOSymlinks(t)
	worktreeDir := helpers.TmpDirWOSymlinks(t)

	// Create the "unit" path in the worktree
	unitPath := filepath.Join(worktreeDir, "module", "unit")
	err := os.MkdirAll(unitPath, 0755)
	require.NoError(t, err)

	// Create a report with the original repo as working dir
	r := report.NewReport().WithWorkingDir(originalRepoDir)

	// Add a run with DiscoveryWorkingDir set to worktree path
	run := newRun(t, unitPath)
	r.AddRun(l, run)
	r.EnsureRun(l, unitPath, report.WithDiscoveryWorkingDir(worktreeDir))
	r.EndRun(l, unitPath, report.WithResult(report.ResultSucceeded))

	// Write CSV
	var buf bytes.Buffer

	err = r.WriteCSV(&buf)
	require.NoError(t, err)

	// Parse the CSV
	reader := csv.NewReader(&buf)
	records, err := reader.ReadAll()
	require.NoError(t, err)

	require.Len(t, records, 2) // header + 1 data row

	// The name (first column) should be relative to worktree dir
	assert.Equal(t, "module/unit", records[1][0],
		"Run name should be relative to DiscoveryWorkingDir, not report.workingDir")
}

// TestParseJSONRuns verifies that JSON report data can be parsed from bytes.
func TestParseJSONRuns(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name        string
		input       string
		expected    report.JSONRuns
		expectError bool
	}{
		{
			name:     "empty array",
			input:    "[]",
			expected: report.JSONRuns{},
		},
		{
			name: "single run",
			input: `[{
				"Name": "module/unit",
				"Started": "2024-01-01T10:00:00Z",
				"Ended": "2024-01-01T10:01:00Z",
				"Result": "succeeded"
			}]`,
			expected: report.JSONRuns{
				{
					Name:   "module/unit",
					Result: "succeeded",
				},
			},
		},
		{
			name: "multiple runs with all fields",
			input: `[
				{
					"Name": "unit-a",
					"Started": "2024-01-01T10:00:00Z",
					"Ended": "2024-01-01T10:01:00Z",
					"Result": "succeeded"
				},
				{
					"Name": "unit-b",
					"Started": "2024-01-01T10:01:00Z",
					"Ended": "2024-01-01T10:02:00Z",
					"Result": "failed",
					"Reason": "run error",
					"Cause": "some error"
				}
			]`,
			expected: report.JSONRuns{
				{Name: "unit-a", Result: "succeeded"},
				{Name: "unit-b", Result: "failed"},
			},
		},
		{
			name:        "invalid json",
			input:       "not valid json",
			expectError: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			runs, err := report.ParseJSONRuns([]byte(tt.input))

			if tt.expectError {
				require.Error(t, err)
				return
			}

			require.NoError(t, err)
			require.Len(t, runs, len(tt.expected))

			for i, expected := range tt.expected {
				assert.Equal(t, expected.Name, runs[i].Name)
				assert.Equal(t, expected.Result, runs[i].Result)
			}
		})
	}
}

// TestParseJSONRunsFromFile verifies that JSON report data can be parsed from a file.
func TestParseJSONRunsFromFile(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	t.Run("valid file", func(t *testing.T) {
		t.Parallel()

		reportFile := filepath.Join(tmp, "valid-report.json")
		content := `[{"Name": "test-unit", "Started": "2024-01-01T10:00:00Z",` +
			` "Ended": "2024-01-01T10:01:00Z", "Result": "succeeded"}]`

		err := os.WriteFile(reportFile, []byte(content), 0644)
		require.NoError(t, err)

		runs, err := report.ParseJSONRunsFromFile(reportFile)
		require.NoError(t, err)
		require.Len(t, runs, 1)
		assert.Equal(t, "test-unit", runs[0].Name)
	})

	t.Run("non-existent file", func(t *testing.T) {
		t.Parallel()

		_, err := report.ParseJSONRunsFromFile(filepath.Join(tmp, "does-not-exist.json"))
		require.Error(t, err)
	})
}

// TestJSONRunsFindByName verifies that runs can be found by name.
func TestJSONRunsFindByName(t *testing.T) {
	t.Parallel()

	runs := report.JSONRuns{
		{Name: "unit-a", Result: "succeeded"},
		{Name: "module/unit-b", Result: "failed"},
		{Name: "nested/path/unit-c", Result: "excluded"},
	}

	tests := []struct {
		expected *report.JSONRun
		name     string
		search   string
	}{
		{
			name:     "find first unit",
			search:   "unit-a",
			expected: &report.JSONRun{Name: "unit-a", Result: "succeeded"},
		},
		{
			name:     "find nested path",
			search:   "module/unit-b",
			expected: &report.JSONRun{Name: "module/unit-b", Result: "failed"},
		},
		{
			name:     "find deeply nested",
			search:   "nested/path/unit-c",
			expected: &report.JSONRun{Name: "nested/path/unit-c", Result: "excluded"},
		},
		{
			name:     "not found",
			search:   "does-not-exist",
			expected: nil,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			result := runs.FindByName(tt.search)

			if tt.expected == nil {
				assert.Nil(t, result)
			} else {
				require.NotNil(t, result)
				assert.Equal(t, tt.expected.Name, result.Name)
				assert.Equal(t, tt.expected.Result, result.Result)
			}
		})
	}
}

// TestJSONRunsNames verifies that run names can be extracted from a slice of runs.
func TestJSONRunsNames(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name     string
		runs     report.JSONRuns
		expected []string
	}{
		{
			name:     "empty slice",
			runs:     report.JSONRuns{},
			expected: []string{},
		},
		{
			name: "multiple runs",
			runs: report.JSONRuns{
				{Name: "unit-a"},
				{Name: "module/unit-b"},
				{Name: "nested/path/unit-c"},
			},
			expected: []string{"unit-a", "module/unit-b", "nested/path/unit-c"},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			names := tt.runs.Names()
			assert.Equal(t, tt.expected, names)
		})
	}
}

// TestParseCSVRuns verifies that CSV report data can be parsed from bytes.
func TestParseCSVRuns(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name        string
		input       string
		expected    report.CSVRuns
		expectError bool
	}{
		{
			name:     "header only",
			input:    "Name,Started,Ended,Result,Reason,Cause,Ref,Cmd,Args\n",
			expected: report.CSVRuns{},
		},
		{
			name: "single run",
			input: "Name,Started,Ended,Result,Reason,Cause,Ref,Cmd,Args\n" +
				"module/unit,2024-01-01T10:00:00Z,2024-01-01T10:01:00Z,succeeded,,,,,\n",
			expected: report.CSVRuns{{
				Name:    "module/unit",
				Started: "2024-01-01T10:00:00Z",
				Ended:   "2024-01-01T10:01:00Z",
				Result:  "succeeded",
			}},
		},
		{
			name: "multiple runs with all fields",
			input: `Name,Started,Ended,Result,Reason,Cause,Ref,Cmd,Args
unit-a,2024-01-01T10:00:00Z,2024-01-01T10:01:00Z,succeeded,,,HEAD~1,plan,-out=plan.tfplan|-var=foo=bar
unit-b,2024-01-01T10:01:00Z,2024-01-01T10:02:00Z,failed,run error,some error,main,apply,
`,
			expected: report.CSVRuns{
				{
					Name:    "unit-a",
					Started: "2024-01-01T10:00:00Z",
					Ended:   "2024-01-01T10:01:00Z",
					Result:  "succeeded",
					Ref:     "HEAD~1",
					Cmd:     "plan",
					Args:    "-out=plan.tfplan|-var=foo=bar",
				},
				{
					Name:    "unit-b",
					Started: "2024-01-01T10:01:00Z",
					Ended:   "2024-01-01T10:02:00Z",
					Result:  "failed",
					Reason:  "run error",
					Cause:   "some error",
					Ref:     "main",
					Cmd:     "apply",
				},
			},
		},
		{
			name:        "invalid csv - missing fields",
			input:       "Name,Started,Ended,Result,Reason,Cause,Ref,Cmd,Args\nunit-a,2024-01-01T10:00:00Z\n",
			expectError: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			runs, err := report.ParseCSVRuns([]byte(tt.input))

			if tt.expectError {
				require.Error(t, err)
				return
			}

			require.NoError(t, err)
			require.Len(t, runs, len(tt.expected))

			for i, expected := range tt.expected {
				assert.Equal(t, expected.Name, runs[i].Name)
				assert.Equal(t, expected.Result, runs[i].Result)
				assert.Equal(t, expected.Reason, runs[i].Reason)
				assert.Equal(t, expected.Cause, runs[i].Cause)
				assert.Equal(t, expected.Ref, runs[i].Ref)
				assert.Equal(t, expected.Cmd, runs[i].Cmd)
				assert.Equal(t, expected.Args, runs[i].Args)
			}
		})
	}
}

// TestParseCSVRunsFromFile verifies that CSV report data can be parsed from a file.
func TestParseCSVRunsFromFile(t *testing.T) {
	t.Parallel()

	tmp := helpers.TmpDirWOSymlinks(t)

	t.Run("valid file", func(t *testing.T) {
		t.Parallel()

		reportFile := filepath.Join(tmp, "valid-report.csv")
		content := "Name,Started,Ended,Result,Reason,Cause,Ref,Cmd,Args\n" +
			"test-unit,2024-01-01T10:00:00Z,2024-01-01T10:01:00Z,succeeded,,,,,\n"

		err := os.WriteFile(reportFile, []byte(content), 0644)
		require.NoError(t, err)

		runs, err := report.ParseCSVRunsFromFile(reportFile)
		require.NoError(t, err)
		require.Len(t, runs, 1)
		assert.Equal(t, "test-unit", runs[0].Name)
	})

	t.Run("non-existent file", func(t *testing.T) {
		t.Parallel()

		_, err := report.ParseCSVRunsFromFile(filepath.Join(tmp, "does-not-exist.csv"))
		require.Error(t, err)
	})
}

// TestCSVRunsFindByName verifies that CSV runs can be found by name.
func TestCSVRunsFindByName(t *testing.T) {
	t.Parallel()

	runs := report.CSVRuns{
		{Name: "unit-a", Result: "succeeded"},
		{Name: "module/unit-b", Result: "failed"},
		{Name: "nested/path/unit-c", Result: "excluded"},
	}

	tests := []struct {
		expected *report.CSVRun
		name     string
		search   string
	}{
		{
			name:     "find first unit",
			search:   "unit-a",
			expected: &report.CSVRun{Name: "unit-a", Result: "succeeded"},
		},
		{
			name:     "find nested path",
			search:   "module/unit-b",
			expected: &report.CSVRun{Name: "module/unit-b", Result: "failed"},
		},
		{
			name:     "not found",
			search:   "does-not-exist",
			expected: nil,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			result := runs.FindByName(tt.search)

			if tt.expected == nil {
				assert.Nil(t, result)
			} else {
				require.NotNil(t, result)
				assert.Equal(t, tt.expected.Name, result.Name)
				assert.Equal(t, tt.expected.Result, result.Result)
			}
		})
	}
}

// TestCSVRunsNames verifies that run names can be extracted from a slice of CSV runs.
func TestCSVRunsNames(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name     string
		runs     report.CSVRuns
		expected []string
	}{
		{
			name:     "empty slice",
			runs:     report.CSVRuns{},
			expected: []string{},
		},
		{
			name: "multiple runs",
			runs: report.CSVRuns{
				{Name: "unit-a"},
				{Name: "module/unit-b"},
				{Name: "nested/path/unit-c"},
			},
			expected: []string{"unit-a", "module/unit-b", "nested/path/unit-c"},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			names := tt.runs.Names()
			assert.Equal(t, tt.expected, names)
		})
	}
}

// TestParseJSONRunsFromFileValidation verifies that JSON report validation works correctly
// when parsing files. Validation is performed as the first step in parsing.
func TestParseJSONRunsFromFileValidation(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name        string
		input       string
		expectError bool
	}{
		{
			name:        "valid empty report",
			input:       "[]",
			expectError: false,
		},
		{
			name: "valid single run",
			input: `[{
				"Name": "module/unit",
				"Started": "2024-01-01T10:00:00Z",
				"Ended": "2024-01-01T10:01:00Z",
				"Result": "succeeded"
			}]`,
			expectError: false,
		},
		{
			name: "valid run with all fields",
			input: `[{
				"Name": "module/unit",
				"Started": "2024-01-01T10:00:00Z",
				"Ended": "2024-01-01T10:01:00Z",
				"Result": "failed",
				"Reason": "run error",
				"Cause": "some error"
			}]`,
			expectError: false,
		},
		{
			name: "valid multiple runs",
			input: "[" +
				`{"Name": "unit-a", "Started": "2024-01-01T10:00:00Z",` +
				` "Ended": "2024-01-01T10:01:00Z", "Result": "succeeded"},` +
				`{"Name": "unit-b", "Started": "2024-01-01T10:01:00Z",` +
				` "Ended": "2024-01-01T10:02:00Z", "Result": "failed",` +
				` "Reason": "run error"}` +
				"]",
			expectError: false,
		},
		{
			name:        "invalid - not an array",
			input:       `{"Name": "unit", "Result": "succeeded"}`,
			expectError: true,
		},
		{
			name:        "invalid - missing required field Name",
			input:       `[{"Started": "2024-01-01T10:00:00Z", "Ended": "2024-01-01T10:01:00Z", "Result": "succeeded"}]`,
			expectError: true,
		},
		{
			name:        "invalid - missing required field Result",
			input:       `[{"Name": "unit", "Started": "2024-01-01T10:00:00Z", "Ended": "2024-01-01T10:01:00Z"}]`,
			expectError: true,
		},
		{
			name:        "invalid json",
			input:       "not valid json",
			expectError: true,
		},
	}

	tmp := helpers.TmpDirWOSymlinks(t)

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			reportFile := filepath.Join(tmp, strings.ReplaceAll(tt.name, " ", "-")+".json")
			err := os.WriteFile(reportFile, []byte(tt.input), 0644)
			require.NoError(t, err)

			_, err = report.ParseJSONRunsFromFile(reportFile)

			if tt.expectError {
				require.Error(t, err)
			} else {
				require.NoError(t, err)
			}
		})
	}

	t.Run("schema validation error details", func(t *testing.T) {
		t.Parallel()

		reportFile := filepath.Join(tmp, "schema-error-details.json")
		content := `[{"Name": "test-unit"}]` // missing required fields

		err := os.WriteFile(reportFile, []byte(content), 0644)
		require.NoError(t, err)

		_, err = report.ParseJSONRunsFromFile(reportFile)
		require.Error(t, err)

		var schemaErr *report.SchemaValidationError
		require.ErrorAs(t, err, &schemaErr)
		assert.NotEmpty(t, schemaErr.Errors)
	})
}

// TestSchemaValidationError verifies the error type works correctly.
func TestSchemaValidationError(t *testing.T) {
	t.Parallel()

	err := &report.SchemaValidationError{
		Errors: []string{"error 1", "error 2"},
	}

	assert.Contains(t, err.Error(), "2 error(s)")
	assert.Contains(t, err.Error(), "error 1")
	assert.Contains(t, err.Error(), "error 2")
}

// newRun creates a new run, and asserts that it doesn't error.
func newRun(t *testing.T, name string) *report.Run {
	t.Helper()

	run, err := report.NewRun(name)
	require.NoError(t, err)

	return run
}
