Files
gochat/internal/model/direct_upload_test.go
T
2026-06-04 15:44:48 +08:00

43 lines
1.4 KiB
Go

package model
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestDirectUploadStatusConstants(t *testing.T) {
assert.Equal(t, DirectUploadStatus("pending"), DirectUploadStatusPending)
assert.Equal(t, DirectUploadStatus("completed"), DirectUploadStatusCompleted)
assert.Equal(t, DirectUploadStatus("expired"), DirectUploadStatusExpired)
}
func TestDirectUploadSourceConstants(t *testing.T) {
assert.Equal(t, DirectUploadSource("account"), DirectUploadSourceAccount)
assert.Equal(t, DirectUploadSource("widget"), DirectUploadSourceWidget)
}
func TestDirectUpload_StructFields(t *testing.T) {
upload := DirectUpload{
UploadUUID: "test-uuid-123",
AccountID: 1,
Status: DirectUploadStatusPending,
Source: DirectUploadSourceAccount,
OriginalName: "test.png",
FileType: "image",
MimeType: "image/png",
FileSize: 1024,
FileURL: "/uploads/account/1/test.png",
}
assert.Equal(t, "test-uuid-123", upload.UploadUUID)
assert.Equal(t, uint(1), upload.AccountID)
assert.Equal(t, DirectUploadStatusPending, upload.Status)
assert.Equal(t, DirectUploadSourceAccount, upload.Source)
assert.Equal(t, "test.png", upload.OriginalName)
assert.Equal(t, "image", upload.FileType)
assert.Equal(t, "image/png", upload.MimeType)
assert.Equal(t, int64(1024), upload.FileSize)
assert.Equal(t, "/uploads/account/1/test.png", upload.FileURL)
}