mirror of
https://github.com/usememos/memos.git
synced 2025-06-05 22:09:59 +02:00
refactor: memo store (#1741)
This commit is contained in:
@@ -5,37 +5,40 @@ import (
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/usememos/memos/api"
|
||||
"github.com/usememos/memos/store"
|
||||
)
|
||||
|
||||
func TestMemoStore(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
store := NewTestingStore(ctx, t)
|
||||
user, err := createTestingHostUser(ctx, store)
|
||||
ts := NewTestingStore(ctx, t)
|
||||
user, err := createTestingHostUser(ctx, ts)
|
||||
require.NoError(t, err)
|
||||
memoCreate := &api.MemoCreate{
|
||||
memoCreate := &store.MemoMessage{
|
||||
CreatorID: user.ID,
|
||||
Content: "test_content",
|
||||
Visibility: api.Public,
|
||||
Visibility: store.Public,
|
||||
}
|
||||
memo, err := store.CreateMemo(ctx, memoCreate)
|
||||
memo, err := ts.CreateMemo(ctx, memoCreate)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, memoCreate.Content, memo.Content)
|
||||
memoPatchContent := "test_content_2"
|
||||
memoPatch := &api.MemoPatch{
|
||||
memoPatch := &store.UpdateMemoMessage{
|
||||
ID: memo.ID,
|
||||
Content: &memoPatchContent,
|
||||
}
|
||||
memo, err = store.PatchMemo(ctx, memoPatch)
|
||||
err = ts.UpdateMemo(ctx, memoPatch)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, memoPatchContent, memo.Content)
|
||||
memoList, err := store.FindMemoList(ctx, &api.MemoFind{
|
||||
memo, err = ts.GetMemo(ctx, &store.FindMemoMessage{
|
||||
ID: &memo.ID,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
memoList, err := ts.ListMemos(ctx, &store.FindMemoMessage{
|
||||
CreatorID: &user.ID,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, 2, len(memoList))
|
||||
require.Equal(t, memo, memoList[1])
|
||||
err = store.DeleteMemo(ctx, &api.MemoDelete{
|
||||
err = ts.DeleteMemo(ctx, &store.DeleteMemoMessage{
|
||||
ID: memo.ID,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
Reference in New Issue
Block a user