package factories

import (
	"gorm.io/datatypes"

	"rocketgit.ru/rsmon/worker/app/models"
)

// CheckFactory provides functionality.
func CheckFactory(monitor *models.Monitor, kind string) models.Check {
	if monitor == nil {
		panic("CheckFactory requires monitor")
	}

	checkName := "test check"
	check := models.Check{
		Name:     &checkName,
		Kind:     kind,
		Monitor:  monitor,
		Settings: datatypes.JSON("{}"),
	}

	return check
}

// PersistedCheck provides functionality.
func PersistedCheck(monitor *models.Monitor, kind string) models.Check {
	check := CheckFactory(monitor, kind)

	err := models.DB().Save(&check).Error
	if err != nil {
		panic(err)
	}
	return check
}