tamabox/internal/db/questions_test.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},
	} {
		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",