YourSelfHosted/memos
1
0
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-06-05 22:09:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

chore: update store tests

Browse Source
This commit is contained in:
Steven
2023-10-03 00:47:34 +08:00
parent 91f7839b31
commit 607fecf437
2 changed files with 59 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
test/store/memo_relation_test.go
View File

@@ -16,42 +16,76 @@ func TestMemoRelationStore(t *testing.T) {
require.NoError(t, err)
memoCreate := &store.Memo{
CreatorID: user.ID,
Content: "test_content",
Content: "main memo content",
Visibility: store.Public,
}
memo, err := ts.CreateMemo(ctx, memoCreate)
require.NoError(t, err)
require.Equal(t, memoCreate.Content, memo.Content)
memo2Create := &store.Memo{
relatedMemoCreate := &store.Memo{
CreatorID: user.ID,
Content: "test_content_2",
Content: "related memo content",
Visibility: store.Public,
}
memo2, err := ts.CreateMemo(ctx, memo2Create)
relatedMemo, err := ts.CreateMemo(ctx, relatedMemoCreate)
require.NoError(t, err)
require.Equal(t, memo2Create.Content, memo2.Content)
memoRelationMessage := &store.MemoRelation{
require.Equal(t, relatedMemoCreate.Content, relatedMemo.Content)
commentMemoCreate := &store.Memo{
CreatorID: user.ID,
Content: "comment memo content",
Visibility: store.Public,
}
commentMemo, err := ts.CreateMemo(ctx, commentMemoCreate)
require.NoError(t, err)
require.Equal(t, commentMemoCreate.Content, commentMemo.Content)
// Reference relation.
referenceRelation := &store.MemoRelation{
MemoID: memo.ID,
RelatedMemoID: memo2.ID,
RelatedMemoID: relatedMemo.ID,
Type: store.MemoRelationReference,
}
_, err = ts.UpsertMemoRelation(ctx, memoRelationMessage)
_, err = ts.UpsertMemoRelation(ctx, referenceRelation)
require.NoError(t, err)
// Comment relation.
commentRelation := &store.MemoRelation{
MemoID: memo.ID,
RelatedMemoID: commentMemo.ID,
Type: store.MemoRelationComment,
}
_, err = ts.UpsertMemoRelation(ctx, commentRelation)
require.NoError(t, err)
memo, err = ts.GetMemo(ctx, &store.FindMemo{
ID: &memo.ID,
})
require.NoError(t, err)
require.Equal(t, 2, len(memo.RelationList))
require.Equal(t, referenceRelation, memo.RelationList[0])
require.Equal(t, commentRelation, memo.RelationList[1])
relatedMemo, err = ts.GetMemo(ctx, &store.FindMemo{
ID: &relatedMemo.ID,
})
require.NoError(t, err)
require.Equal(t, 1, len(relatedMemo.RelationList))
require.Equal(t, referenceRelation, relatedMemo.RelationList[0])
commentMemo, err = ts.GetMemo(ctx, &store.FindMemo{
ID: &commentMemo.ID,
})
require.NoError(t, err)
require.Equal(t, 1, len(commentMemo.RelationList))
require.Equal(t, commentRelation, commentMemo.RelationList[0])
err = ts.DeleteMemo(ctx, &store.DeleteMemo{
ID: relatedMemo.ID,
})
require.NoError(t, err)
err = ts.DeleteMemo(ctx, &store.DeleteMemo{
ID: commentMemo.ID,
})
require.NoError(t, err)
memoRelation, err := ts.ListMemoRelations(ctx, &store.FindMemoRelation{
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, &store.DeleteMemo{
ID: memo2.ID,
})
require.NoError(t, err)
memoRelation, err = ts.ListMemoRelations(ctx, &store.FindMemoRelation{
MemoID: &memo.ID,
})
require.NoError(t, err)
require.Equal(t, 0, len(memoRelation))
}