tamakyi
e3e59e7c69
Repo .git was damaged (refs/ missing, index truncated, object store near-empty). Repaired via git init (rebuilt refs), stale index preserved as .git/index.corrupt-20260910.bak, then full worktree committed as a rollback baseline.
134 lines
3.5 KiB
Go
134 lines
3.5 KiB
Go
// Copyright 2023 E99p1ant. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package db
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUploadImages(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db, cleanup := newTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
uploadImagesStore := NewUploadImagesStore(db)
|
|
|
|
for _, tc := range []struct {
|
|
name string
|
|
test func(t *testing.T, ctx context.Context, db *uploadImages)
|
|
}{
|
|
{"Create", testUploadImagesCreate},
|
|
{"GetByQuestionID", testUploadImagesGetByQuestionID},
|
|
{"GetByTypeQuestionID", testUploadImagesGetByTypeQuestionID},
|
|
} {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Cleanup(func() {
|
|
if err := cleanup("upload_images"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
tc.test(t, ctx, uploadImagesStore.(*uploadImages))
|
|
})
|
|
}
|
|
}
|
|
|
|
func testUploadImagesCreate(t *testing.T, ctx context.Context, db *uploadImages) {
|
|
got, err := db.Create(ctx, CreateUploadImageOptions{
|
|
Type: UploadImageQuestionTypeAsk,
|
|
QuestionID: 1,
|
|
UploaderUserID: 1,
|
|
Name: "test.png",
|
|
FileSize: 12345,
|
|
Md5: "d41d8cd98f00b204e9800998ecf8427e",
|
|
Key: "uploads/test.png",
|
|
IsDeletingPrevious: false,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
want := &UploadImage{
|
|
Model: got.Model,
|
|
UploaderUserID: 1,
|
|
Name: "test.png",
|
|
FileSize: 12345,
|
|
Md5: "d41d8cd98f00b204e9800998ecf8427e",
|
|
Key: "uploads/test.png",
|
|
}
|
|
require.Equal(t, want, got)
|
|
}
|
|
|
|
func testUploadImagesGetByQuestionID(t *testing.T, ctx context.Context, db *uploadImages) {
|
|
_, err := db.Create(ctx, CreateUploadImageOptions{
|
|
Type: UploadImageQuestionTypeAsk,
|
|
QuestionID: 1,
|
|
UploaderUserID: 1,
|
|
Name: "test.png",
|
|
FileSize: 12345,
|
|
Md5: "d41d8cd98f00b204e9800998ecf8427e",
|
|
Key: "uploads/test.png",
|
|
IsDeletingPrevious: false,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
t.Run("normal", func(t *testing.T) {
|
|
got, err := db.GetByQuestionID(ctx, 1)
|
|
require.Nil(t, err)
|
|
|
|
want := []*UploadImage{{
|
|
Model: got[0].Model,
|
|
UploaderUserID: 1,
|
|
Name: "test.png",
|
|
FileSize: 12345,
|
|
Md5: "d41d8cd98f00b204e9800998ecf8427e",
|
|
Key: "uploads/test.png",
|
|
}}
|
|
require.Equal(t, got, want)
|
|
})
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
got, err := db.GetByQuestionID(ctx, 2)
|
|
require.Nil(t, err)
|
|
require.Equal(t, 0, len(got))
|
|
})
|
|
}
|
|
|
|
func testUploadImagesGetByTypeQuestionID(t *testing.T, ctx context.Context, db *uploadImages) {
|
|
_, err := db.Create(ctx, CreateUploadImageOptions{
|
|
Type: UploadImageQuestionTypeAsk,
|
|
QuestionID: 1,
|
|
UploaderUserID: 1,
|
|
Name: "test.png",
|
|
FileSize: 12345,
|
|
Md5: "d41d8cd98f00b204e9800998ecf8427e",
|
|
Key: "uploads/test.png",
|
|
IsDeletingPrevious: false,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
t.Run("normal", func(t *testing.T) {
|
|
got, err := db.GetByTypeQuestionID(ctx, UploadImageQuestionTypeAsk, 1)
|
|
require.Nil(t, err)
|
|
|
|
want := []*UploadImage{{
|
|
Model: got[0].Model,
|
|
UploaderUserID: 1,
|
|
Name: "test.png",
|
|
FileSize: 12345,
|
|
Md5: "d41d8cd98f00b204e9800998ecf8427e",
|
|
Key: "uploads/test.png",
|
|
}}
|
|
require.Equal(t, got, want)
|
|
})
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
got, err := db.GetByTypeQuestionID(ctx, UploadImageQuestionTypeAsk, 2)
|
|
require.Nil(t, err)
|
|
require.Equal(t, 0, len(got))
|
|
})
|
|
}
|