package jobs

import (
	"database/sql"
	"testing"
	"time"

	"quyun/v2/app/commands/testx"
	jobs_args "quyun/v2/app/jobs/args"
	"quyun/v2/app/services"
	"quyun/v2/database"
	"quyun/v2/database/models"
	"quyun/v2/pkg/consts"

	"github.com/riverqueue/river"
	"github.com/riverqueue/river/rivertype"
	. "github.com/smartystreets/goconvey/convey"
	"github.com/stretchr/testify/suite"

	_ "go.ipao.vip/atom"
	"go.ipao.vip/atom/contracts"
	"go.ipao.vip/gen/types"
	"go.uber.org/dig"
)

type MediaAssetProcessJobSuiteInjectParams struct {
	dig.In

	DB       *sql.DB
	Initials []contracts.Initial `group:"initials"` // nolint:structcheck
}

type MediaAssetProcessJobSuite struct {
	suite.Suite
	MediaAssetProcessJobSuiteInjectParams
}

func Test_MediaAssetProcessJob(t *testing.T) {
	// 注意：testx.Default() 已内置一个测试用的 job worker 注册器，避免和 jobs.Provide 重复注册同 kind worker。
	providers := testx.Default().With(services.Provide)

	testx.Serve(providers, t, func(p MediaAssetProcessJobSuiteInjectParams) {
		suite.Run(t, &MediaAssetProcessJobSuite{MediaAssetProcessJobSuiteInjectParams: p})
	})
}

func (s *MediaAssetProcessJobSuite) Test_Work_ProcessingToReady() {
	Convey("MediaAssetProcessJobWorker processing -> ready", s.T(), func() {
		ctx := s.T().Context()
		now := time.Now().UTC()
		tenantID := int64(1)
		userID := int64(2)

		database.Truncate(ctx, s.DB, models.TableNameMediaAsset)

		asset := &models.MediaAsset{
			TenantID:  tenantID,
			UserID:    userID,
			Type:      consts.MediaAssetTypeVideo,
			Status:    consts.MediaAssetStatusProcessing,
			Provider:  "test",
			Bucket:    "b",
			ObjectKey: "k",
			Meta:      types.JSON([]byte("{}")),
			CreatedAt: now,
			UpdatedAt: now,
		}
		So(asset.Create(ctx), ShouldBeNil)

		worker := &MediaAssetProcessJobWorker{}
		err := worker.Work(ctx, &river.Job[jobs_args.MediaAssetProcessJob]{
			JobRow: &rivertype.JobRow{Attempt: 1},
			Args:   jobs_args.MediaAssetProcessJob{TenantID: tenantID, AssetID: asset.ID},
		})
		So(err, ShouldBeNil)

		tbl, query := models.MediaAssetQuery.QueryContext(ctx)
		got, err := query.Where(
			tbl.TenantID.Eq(tenantID),
			tbl.ID.Eq(asset.ID),
		).First()
		So(err, ShouldBeNil)
		So(got.Status, ShouldEqual, consts.MediaAssetStatusReady)
	})
}