0904

2026-09-08 10:10:44 +08:00
commit 2ed27dd55e
288 changed files with 67061 additions and 0 deletions
@@ -0,0 +1,133 @@
+// 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))
+	})
+}