package models

import (
	"testing"
	"time"

	"github.com/lib/pq"
	"github.com/stretchr/testify/require"
)

func TestMigratePreservesEveryLegacyStatusPageMaintenance(t *testing.T) {
	Drop()
	Migrate()
	plan := Plan{Name: "legacy migration plan"}
	require.NoError(t, DB().Create(&plan).Error)
	account := Account{Name: "legacy migration account", PlanID: &plan.ID}
	require.NoError(t, DB().Create(&account).Error)
	page := StatusPage{AccountID: account.ID, Slug: "legacy-maintenance-migration", Name: "Legacy"}
	require.NoError(t, DB().Create(&page).Error)
	start := time.Date(2026, 7, 1, 12, 0, 0, 0, time.UTC)
	for i := 0; i < 2; i++ {
		row := StatusPageMaintenance{StatusPageID: page.ID, Title: "same", StartsAt: start, EndsAt: start.Add(time.Hour), MonitorIDs: pq.Int64Array{}}
		require.NoError(t, DB().Create(&row).Error)
	}

	Migrate()
	var migrated []Maintenance
	require.NoError(t, DB().Where("legacy_status_page_maintenance_id IS NOT NULL").Find(&migrated).Error)
	require.Len(t, migrated, 2)
	var joins int64
	require.NoError(t, DB().Table("maintenance_status_pages").Where("status_page_id = ?", page.ID).Count(&joins).Error)
	require.EqualValues(t, 2, joins)
}