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,7 +5,6 @@ import (
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/usememos/memos/api"
|
||||
"github.com/usememos/memos/store"
|
||||
)
|
||||
|
||||
@@ -14,22 +13,22 @@ func TestMemoRelationStore(t *testing.T) {
|
||||
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 := ts.CreateMemo(ctx, memoCreate)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, memoCreate.Content, memo.Content)
|
||||
memoCreate = &api.MemoCreate{
|
||||
memo2Create := &store.MemoMessage{
|
||||
CreatorID: user.ID,
|
||||
Content: "test_content_2",
|
||||
Visibility: api.Public,
|
||||
Visibility: store.Public,
|
||||
}
|
||||
memo2, err := ts.CreateMemo(ctx, memoCreate)
|
||||
memo2, err := ts.CreateMemo(ctx, memo2Create)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, memoCreate.Content, memo2.Content)
|
||||
require.Equal(t, memo2Create.Content, memo2.Content)
|
||||
memoRelationMessage := &store.MemoRelationMessage{
|
||||
MemoID: memo.ID,
|
||||
RelatedMemoID: memo2.ID,
|
||||
@@ -45,7 +44,7 @@ func TestMemoRelationStore(t *testing.T) {
|
||||
require.Equal(t, memo2.ID, memoRelation[0].RelatedMemoID)
|
||||
require.Equal(t, memo.ID, memoRelation[0].MemoID)
|
||||
require.Equal(t, store.MemoRelationReference, memoRelation[0].Type)
|
||||
err = ts.DeleteMemo(ctx, &api.MemoDelete{
|
||||
err = ts.DeleteMemo(ctx, &store.DeleteMemoMessage{
|
||||
ID: memo2.ID,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
Reference in New Issue
Block a user