package structs

import (
	"encoding/json"
	"testing"

	"github.com/stretchr/testify/assert"
)

func Test_MarshalJSON(t *testing.T) {
	alert := &UserAlertInfo{
		Severity: AlertSeverityMajor,
	}
	a, err := json.Marshal(alert)
	assert.Nil(t, err)
	assert.NotContains(t, string(a), "major")

	alert2 := &UserAlertInfo{
		Severity: AlertSeverityWarning,
	}
	b, err := json.Marshal(alert2)
	assert.Nil(t, err)
	assert.Contains(t, string(b), "warning")

	alert3 := &UserAlertInfo{
		Severity: AlertSeverityCritical,
	}
	c, err := json.Marshal(alert3)
	assert.Nil(t, err)
	assert.Contains(t, string(c), "critical")
}