chore: baseline snapshot before moderation-strictness feature
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.
This commit is contained in:
tamakyi
@@ -0,0 +1,467 @@
|
||||
// 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},
|
||||
{"DeleteByID", testQuestionsDeleteByID},
|
||||
{"UpdateCensor", testQuestionsUpdateCensor},
|
||||
{"Count", testQuestionsCount},
|
||||
{"SetPrivate", testQuestionsSetPrivate},
|
||||
{"SetPublic", testQuestionsSetPublic},
|
||||
} {
|
||||
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)
|
||||
})
|
||||
}
|
||||
|
||||
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)
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user