This commit is contained in:
2026-10-27 22:55:36 +08:00
parent 1231251ea4
commit 1f6cd63815
5 changed files with 203 additions and 23 deletions
+46 -10
View File
@@ -12,16 +12,17 @@ import (
"portal-mini-go/internal/model"
"portal-mini-go/internal/pkg/config"
"portal-mini-go/internal/pkg/storage"
"portal-mini-go/internal/repository/file"
filesvc "portal-mini-go/internal/service/file"
)
// --- 测试替身 ---
type fakeStorage struct {
putErr error
gotKey string
putErr error
gotKey string
gotSize int64
data map[string][]byte
data map[string][]byte
}
func newFakeStorage() *fakeStorage { return &fakeStorage{data: map[string][]byte{}} }
@@ -50,12 +51,16 @@ func (s *fakeStorage) Delete(_ context.Context, key string) error {
func (s *fakeStorage) URL(key string) string { return "/files/static/" + key }
type fakeRepo struct {
created *model.File
byID *model.File
getErr error
createErr error
delErr error
deletedID uint64
created *model.File
byID *model.File
getErr error
createErr error
delErr error
deletedID uint64
listed []model.File
listTotal int64
listErr error
listFilter file.ListFilter
}
func (r *fakeRepo) Create(_ context.Context, f *model.File) error { r.created = f; return r.createErr }
@@ -66,11 +71,18 @@ func (r *fakeRepo) GetByID(_ context.Context, id uint64) (*model.File, error) {
return r.byID, nil
}
func (r *fakeRepo) Delete(_ context.Context, id uint64) error { r.deletedID = id; return r.delErr }
func (r *fakeRepo) List(_ context.Context, f file.ListFilter) ([]model.File, int64, error) {
r.listFilter = f
if r.listErr != nil {
return nil, 0, r.listErr
}
return r.listed, r.listTotal, nil
}
// 编译期接口断言。
var (
_ filesvc.FileRepository = (*fakeRepo)(nil)
_ storage.Storage = (*fakeStorage)(nil)
_ storage.Storage = (*fakeStorage)(nil)
)
func newSvc(cfg config.UploadConfig, store *fakeStorage, repo *fakeRepo) *filesvc.Service {
@@ -192,3 +204,27 @@ func TestDelete(t *testing.T) {
require.Equal(t, uint64(9), repo.deletedID)
require.Empty(t, store.data)
}
func TestList(t *testing.T) {
want := []model.File{{ID: 1, OriginName: "a.png"}, {ID: 2, OriginName: "b.jpg"}}
repo := &fakeRepo{listed: want, listTotal: 2}
svc := newSvc(config.UploadConfig{}, newFakeStorage(), repo)
got, total, err := svc.List(context.Background(), filesvc.ListReq{Keyword: "png", Page: 2, Size: 15})
require.NoError(t, err)
require.Equal(t, int64(2), total)
require.Equal(t, want, got)
require.Equal(t, "png", repo.listFilter.Keyword)
require.Equal(t, 2, repo.listFilter.Page)
require.Equal(t, 15, repo.listFilter.Size)
}
func TestList_Defaults(t *testing.T) {
repo := &fakeRepo{listed: nil, listTotal: 0}
svc := newSvc(config.UploadConfig{}, newFakeStorage(), repo)
_, _, err := svc.List(context.Background(), filesvc.ListReq{})
require.NoError(t, err)
require.Equal(t, 1, repo.listFilter.Page, "缺省页码应为 1")
require.Equal(t, 20, repo.listFilter.Size, "缺省每页应为 20")
}