549 lines
13 KiB
Go
549 lines
13 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"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/tamakyi/TamaBox/internal/dbutil"
|
|
)
|
|
|
|
func TestQuestions(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db, cleanup := newTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
questionsStore := NewQuestionsStore(db)
|
|
|
|
for _, tc := range []struct {
|
|
name string
|
|
test func(t *testing.T, ctx context.Context, db *questions)
|
|
}{
|
|
{"Create", testQuestionsCreate},
|
|
{"GetByID", testQuestionsGetByID},
|
|
{"GetByUserID", testQuestionsGetByUserID},
|
|
{"GetByAskUserID", testQuestionsGetByAskUserID},
|
|
{"AnswerByID", testQuestionsAnswerByID},
|
|
{"AnswerByIDPrivate", testQuestionsAnswerByIDPrivate},
|
|
{"DeleteByID", testQuestionsDeleteByID},
|
|
{"UpdateCensor", testQuestionsUpdateCensor},
|
|
{"Count", testQuestionsCount},
|
|
{"SetPrivate", testQuestionsSetPrivate},
|
|
{"SetPublic", testQuestionsSetPublic},
|
|
{"NextUnread", testQuestionsNextUnread},
|
|
} {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Cleanup(func() {
|
|
if err := cleanup("questions"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
tc.test(t, ctx, questionsStore.(*questions))
|
|
})
|
|
}
|
|
}
|
|
|
|
func testQuestionsCreate(t *testing.T, ctx context.Context, db *questions) {
|
|
t.Run("normal", func(t *testing.T) {
|
|
got, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Hello, world!",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 0,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
got.Token = ""
|
|
got.CreatedAt = time.Time{}
|
|
got.UpdatedAt = time.Time{}
|
|
|
|
want := &Question{
|
|
Model: dbutil.Model{
|
|
ID: 1,
|
|
},
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Hello, world!",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
}
|
|
require.Equal(t, got, want)
|
|
})
|
|
}
|
|
|
|
func testQuestionsGetByID(t *testing.T, ctx context.Context, db *questions) {
|
|
_, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Hello, world!",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 0,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
t.Run("normal", func(t *testing.T) {
|
|
got, err := db.GetByID(ctx, 1)
|
|
require.Nil(t, err)
|
|
|
|
got.Token = ""
|
|
got.CreatedAt = time.Time{}
|
|
got.UpdatedAt = time.Time{}
|
|
|
|
want := &Question{
|
|
Model: dbutil.Model{
|
|
ID: 1,
|
|
},
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Hello, world!",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
}
|
|
require.Equal(t, got, want)
|
|
})
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
_, err := db.GetByID(ctx, 404)
|
|
require.Equal(t, ErrQuestionNotExist, err)
|
|
})
|
|
}
|
|
|
|
func testQuestionsGetByUserID(t *testing.T, ctx context.Context, db *questions) {
|
|
_, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 0,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
_, err = db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 2,
|
|
Content: "Content - 2",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 0,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
_, err = db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 3",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 0,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
t.Run("all", func(t *testing.T) {
|
|
got, err := db.GetByUserID(ctx, 1, GetQuestionsByUserIDOptions{})
|
|
require.Nil(t, err)
|
|
|
|
for _, g := range got {
|
|
g.Token = ""
|
|
g.CreatedAt = time.Time{}
|
|
g.UpdatedAt = time.Time{}
|
|
}
|
|
|
|
want := []*Question{
|
|
{
|
|
Model: dbutil.Model{
|
|
ID: 3,
|
|
},
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 3",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
},
|
|
{
|
|
Model: dbutil.Model{
|
|
ID: 1,
|
|
},
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
},
|
|
}
|
|
require.Equal(t, want, got)
|
|
})
|
|
|
|
t.Run("cursor", func(t *testing.T) {
|
|
got, err := db.GetByUserID(ctx, 1, GetQuestionsByUserIDOptions{
|
|
Cursor: &dbutil.Cursor{
|
|
Value: 3,
|
|
PageSize: 10,
|
|
},
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
for _, g := range got {
|
|
g.Token = ""
|
|
g.CreatedAt = time.Time{}
|
|
g.UpdatedAt = time.Time{}
|
|
}
|
|
|
|
want := []*Question{
|
|
{
|
|
Model: dbutil.Model{
|
|
ID: 1,
|
|
},
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
},
|
|
}
|
|
require.Equal(t, want, got)
|
|
})
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
_, err := db.GetByID(ctx, 404)
|
|
require.Equal(t, ErrQuestionNotExist, err)
|
|
})
|
|
}
|
|
|
|
func testQuestionsGetByAskUserID(t *testing.T, ctx context.Context, db *questions) {
|
|
_, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 1,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
_, err = db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 2,
|
|
Content: "Content - 2",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 2,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
_, err = db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 2,
|
|
Content: "Content - 2",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 2,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
t.Run("all", func(t *testing.T) {
|
|
got, err := db.GetByAskUserID(ctx, 1, GetQuestionsByAskUserIDOptions{})
|
|
require.Nil(t, err)
|
|
|
|
for _, g := range got {
|
|
g.Token = ""
|
|
g.CreatedAt = time.Time{}
|
|
g.UpdatedAt = time.Time{}
|
|
}
|
|
|
|
want := []*Question{
|
|
{
|
|
Model: dbutil.Model{
|
|
ID: 1,
|
|
},
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 1,
|
|
},
|
|
}
|
|
require.Equal(t, want, got)
|
|
})
|
|
|
|
t.Run("cursor", func(t *testing.T) {
|
|
got, err := db.GetByAskUserID(ctx, 2, GetQuestionsByAskUserIDOptions{
|
|
Cursor: &dbutil.Cursor{
|
|
Value: 3,
|
|
PageSize: 10,
|
|
},
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
for _, g := range got {
|
|
g.Token = ""
|
|
g.CreatedAt = time.Time{}
|
|
g.UpdatedAt = time.Time{}
|
|
}
|
|
|
|
want := []*Question{
|
|
{
|
|
Model: dbutil.Model{
|
|
ID: 2,
|
|
},
|
|
FromIP: "114.5.1.4",
|
|
UserID: 2,
|
|
Content: "Content - 2",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 2,
|
|
},
|
|
}
|
|
require.Equal(t, want, got)
|
|
})
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
_, err := db.GetByID(ctx, 404)
|
|
require.Equal(t, ErrQuestionNotExist, err)
|
|
})
|
|
}
|
|
|
|
func testQuestionsAnswerByID(t *testing.T, ctx context.Context, db *questions) {
|
|
t.Run("normal", func(t *testing.T) {
|
|
_, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 1,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
err = db.AnswerByID(ctx, 1, "Answer - 1")
|
|
require.Nil(t, err)
|
|
|
|
got, err := db.GetByID(ctx, 1)
|
|
require.Nil(t, err)
|
|
require.Equal(t, "Answer - 1", got.Answer)
|
|
})
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
err := db.AnswerByID(ctx, 404, "Answer - 1")
|
|
require.Equal(t, ErrQuestionNotExist, err)
|
|
})
|
|
}
|
|
|
|
// testQuestionsAnswerByIDPrivate 覆盖 L2「收窄公开」的写入语义：
|
|
// 回答与 is_private 必须在同一次写入里生效，不能出现「公开 + 已作答」的中间态。
|
|
func testQuestionsAnswerByIDPrivate(t *testing.T, ctx context.Context, db *questions) {
|
|
t.Run("normal", func(t *testing.T) {
|
|
_, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
AskerUserID: 1,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
err = db.AnswerByIDPrivate(ctx, 1, "Answer - 1")
|
|
require.Nil(t, err)
|
|
|
|
got, err := db.GetByID(ctx, 1)
|
|
require.Nil(t, err)
|
|
require.Equal(t, "Answer - 1", got.Answer)
|
|
require.True(t, got.IsPrivate)
|
|
})
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
err := db.AnswerByIDPrivate(ctx, 404, "Answer - 1")
|
|
require.Equal(t, ErrQuestionNotExist, err)
|
|
})
|
|
}
|
|
|
|
func testQuestionsDeleteByID(t *testing.T, ctx context.Context, db *questions) {
|
|
t.Run("normal", func(t *testing.T) {
|
|
_, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 1,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
err = db.DeleteByID(ctx, 1)
|
|
require.Nil(t, err)
|
|
|
|
_, err = db.GetByID(ctx, 1)
|
|
require.Equal(t, ErrQuestionNotExist, err)
|
|
})
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
err := db.DeleteByID(ctx, 404)
|
|
require.Equal(t, ErrQuestionNotExist, err)
|
|
})
|
|
}
|
|
|
|
func testQuestionsUpdateCensor(t *testing.T, ctx context.Context, db *questions) {
|
|
t.Run("normal", func(t *testing.T) {
|
|
_, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 1,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
err = db.DeleteByID(ctx, 1)
|
|
require.Nil(t, err)
|
|
|
|
_, err = db.GetByID(ctx, 1)
|
|
require.Equal(t, ErrQuestionNotExist, err)
|
|
})
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
err := db.UpdateCensor(ctx, 404, UpdateQuestionCensorOptions{})
|
|
require.NotNil(t, err)
|
|
})
|
|
}
|
|
|
|
func testQuestionsCount(t *testing.T, ctx context.Context, db *questions) {
|
|
_, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 1,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
_, err = db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 2,
|
|
Content: "Content - 2",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 0,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
_, err = db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 3",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 0,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
err = db.AnswerByID(ctx, 3, "Answer - 3")
|
|
require.Nil(t, err)
|
|
|
|
t.Run("normal", func(t *testing.T) {
|
|
got, err := db.Count(ctx, 1, GetQuestionsCountOptions{})
|
|
require.Nil(t, err)
|
|
|
|
want := int64(2)
|
|
require.Equal(t, want, got)
|
|
})
|
|
|
|
t.Run("filter", func(t *testing.T) {
|
|
got, err := db.Count(ctx, 1, GetQuestionsCountOptions{FilterAnswered: true})
|
|
require.Nil(t, err)
|
|
|
|
want := int64(1)
|
|
require.Equal(t, want, got)
|
|
})
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
got, err := db.Count(ctx, 404, GetQuestionsCountOptions{})
|
|
require.Nil(t, err)
|
|
|
|
want := int64(0)
|
|
require.Equal(t, want, got)
|
|
})
|
|
}
|
|
|
|
func testQuestionsSetPrivate(t *testing.T, ctx context.Context, db *questions) {
|
|
_, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 1,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
t.Run("normal", func(t *testing.T) {
|
|
err = db.SetPrivate(ctx, 1)
|
|
require.Nil(t, err)
|
|
|
|
got, err := db.GetByID(ctx, 1)
|
|
require.Nil(t, err)
|
|
require.True(t, got.IsPrivate)
|
|
})
|
|
}
|
|
|
|
func testQuestionsSetPublic(t *testing.T, ctx context.Context, db *questions) {
|
|
_, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: 1,
|
|
Content: "Content - 1",
|
|
ReceiveReplyEmail: "[email protected]",
|
|
AskerUserID: 1,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
t.Run("normal", func(t *testing.T) {
|
|
err = db.SetPublic(ctx, 1)
|
|
require.Nil(t, err)
|
|
|
|
got, err := db.GetByID(ctx, 1)
|
|
require.Nil(t, err)
|
|
require.False(t, got.IsPrivate)
|
|
})
|
|
}
|
|
|
|
// NextUnread 必须满足三条：只取本人的提问、只取未读（is_read=false 且未回答）、按最早优先。
|
|
func testQuestionsNextUnread(t *testing.T, ctx context.Context, db *questions) {
|
|
create := func(userID uint, content string) *Question {
|
|
q, err := db.Create(ctx, CreateQuestionOptions{
|
|
FromIP: "114.5.1.4",
|
|
UserID: userID,
|
|
Content: content,
|
|
})
|
|
require.Nil(t, err)
|
|
return q
|
|
}
|
|
|
|
q1 := create(1, "Content - 1")
|
|
q2 := create(1, "Content - 2")
|
|
q3 := create(1, "Content - 3")
|
|
other := create(2, "Content - other")
|
|
|
|
t.Run("oldest first", func(t *testing.T) {
|
|
got, err := db.NextUnread(ctx, 1, 0)
|
|
require.Nil(t, err)
|
|
require.Equal(t, q1.ID, got.ID)
|
|
})
|
|
|
|
t.Run("exclude current", func(t *testing.T) {
|
|
got, err := db.NextUnread(ctx, 1, q1.ID)
|
|
require.Nil(t, err)
|
|
require.Equal(t, q2.ID, got.ID)
|
|
})
|
|
|
|
t.Run("only own questions", func(t *testing.T) {
|
|
got, err := db.NextUnread(ctx, 2, 0)
|
|
require.Nil(t, err)
|
|
require.Equal(t, other.ID, got.ID)
|
|
})
|
|
|
|
t.Run("skip read and answered", func(t *testing.T) {
|
|
require.Nil(t, db.SetRead(ctx, q1.ID))
|
|
require.Nil(t, db.SetRead(ctx, q2.ID))
|
|
require.Nil(t, db.AnswerByID(ctx, q3.ID, "Answer - 3"))
|
|
|
|
got, err := db.NextUnread(ctx, 1, 0)
|
|
require.Nil(t, err)
|
|
require.Nil(t, got, "已读与已回答都不算未读")
|
|
})
|
|
|
|
t.Run("empty", func(t *testing.T) {
|
|
got, err := db.NextUnread(ctx, 404, 0)
|
|
require.Nil(t, err)
|
|
require.Nil(t, got)
|
|
})
|
|
}
|