package org

import (
	"testing"
	"time"

	"github.com/onsi/gomega"
	"gorm.io/gorm"

	"github.com/pingcap-inc/clinic/pkg/db"
)

func Test_SortUploadFiles2(t *testing.T) {
	sortedByCreateTime := make([]db.UploadFile, 0)

	now := time.Unix(1644854400, 0)
	sortedByCreateTime = append(sortedByCreateTime, db.UploadFile{
		Model: gorm.Model{
			CreatedAt: now.Add(time.Duration(1)),
		},
		FileUUID: "Test_SortUploadFiles2_1",
	})

	sortedByUpdateTime := make([]db.UploadFile, 0)
	sortedByUpdateTime = append(sortedByUpdateTime, db.UploadFile{
		Model: gorm.Model{
			UpdatedAt: now.Add(time.Duration(1)),
		},
		FileUUID: "Test_SortUploadFiles2_1",
	})

	files := SortUploadFiles(sortedByUpdateTime, sortedByCreateTime, 5)

	g := gomega.NewWithT(t)
	g.Expect(len(files)).To(gomega.Equal(2))
}