mirror of
https://github.com/usememos/memos.git
synced 2024-11-15 19:26:54 +08:00
57 lines
1.6 KiB
Go
57 lines
1.6 KiB
Go
package teststore
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/usememos/memos/api"
|
|
"github.com/usememos/memos/store"
|
|
)
|
|
|
|
func TestMemoRelationStore(t *testing.T) {
|
|
ctx := context.Background()
|
|
ts := NewTestingStore(ctx, t)
|
|
user, err := createTestingHostUser(ctx, ts)
|
|
require.NoError(t, err)
|
|
memoCreate := &api.MemoCreate{
|
|
CreatorID: user.ID,
|
|
Content: "test_content",
|
|
Visibility: api.Public,
|
|
}
|
|
memo, err := ts.CreateMemo(ctx, memoCreate)
|
|
require.NoError(t, err)
|
|
require.Equal(t, memoCreate.Content, memo.Content)
|
|
memoCreate = &api.MemoCreate{
|
|
CreatorID: user.ID,
|
|
Content: "test_content_2",
|
|
Visibility: api.Public,
|
|
}
|
|
memo2, err := ts.CreateMemo(ctx, memoCreate)
|
|
require.NoError(t, err)
|
|
require.Equal(t, memoCreate.Content, memo2.Content)
|
|
memoRelationMessage := &store.MemoRelationMessage{
|
|
MemoID: memo.ID,
|
|
RelatedMemoID: memo2.ID,
|
|
Type: store.MemoRelationReference,
|
|
}
|
|
_, err = ts.UpsertMemoRelation(ctx, memoRelationMessage)
|
|
require.NoError(t, err)
|
|
memoRelation, err := ts.ListMemoRelations(ctx, &store.FindMemoRelationMessage{
|
|
MemoID: &memo.ID,
|
|
})
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(memoRelation))
|
|
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{
|
|
ID: memo2.ID,
|
|
})
|
|
require.NoError(t, err)
|
|
memoRelation, err = ts.ListMemoRelations(ctx, &store.FindMemoRelationMessage{
|
|
MemoID: &memo.ID,
|
|
})
|
|
require.NoError(t, err)
|
|
require.Equal(t, 0, len(memoRelation))
|
|
}
|