package store

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/pingcap/test-infra/common/model"

	"github.com/pingcap/test-infra/services/tcms/utils"
)

func TestGetReleaseSummary(t *testing.T) {
	s := setup(t)
	defer s.Shutdown()
	ctx := utils.FakeCtx()

	res, err := s.Summary.GetReleaseSummary(ctx, 3, nil)
	if assert.NoError(t, err) {
		assert.Equal(t, 5, res.PlanExecTotal)
		assert.Equal(t, 2, res.PlanExecFailureCnt)
		assert.Equal(t, 0, res.CaseExecTotal)
	}

	triagePhase := model.TriagePhaseTriaging
	res, err = s.Summary.GetReleaseSummary(ctx, 3, &triagePhase)
	if assert.NoError(t, err) {
		assert.Equal(t, 0, res.PlanExecTotal)
		assert.Equal(t, 0, res.PlanExecFailureCnt)
		assert.Equal(t, 0, res.CaseExecTotal)
	}
}