feat: implement memo relation server (#1618)

test/server/memo_relation_test.go

@@ -0,0 +1,97 @@
+package testserver
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"testing"
+
+	"github.com/pkg/errors"
+	"github.com/stretchr/testify/require"
+	"github.com/usememos/memos/api"
+)
+
+func TestMemoRelationServer(t *testing.T) {
+	ctx := context.Background()
+	s, err := NewTestingServer(ctx, t)
+	require.NoError(t, err)
+	defer s.Shutdown(ctx)
+
+	signup := &api.SignUp{
+		Username: "testuser",
+		Password: "testpassword",
+	}
+	user, err := s.postAuthSignup(signup)
+	require.NoError(t, err)
+	require.Equal(t, signup.Username, user.Username)
+	memoList, err := s.getMemoList()
+	require.NoError(t, err)
+	require.Len(t, memoList, 0)
+	memo, err := s.postMemoCreate(&api.MemoCreate{
+		Content: "test memo",
+	})
+	require.NoError(t, err)
+	require.Equal(t, "test memo", memo.Content)
+	memo2, err := s.postMemoCreate(&api.MemoCreate{
+		Content: "test memo2",
+		MemoRelationList: []*api.MemoRelationUpsert{
+			{
+				RelatedMemoID: memo.ID,
+				Type:          api.MemoRelationReference,
+			},
+		},
+	})
+	require.NoError(t, err)
+	require.Equal(t, "test memo2", memo2.Content)
+	memoList, err = s.getMemoList()
+	require.NoError(t, err)
+	require.Len(t, memoList, 2)
+	require.Len(t, memo2.RelationList, 1)
+	err = s.deleteMemoRelation(memo2.ID, memo.ID, api.MemoRelationReference)
+	require.NoError(t, err)
+	memo2, err = s.getMemo(memo2.ID)
+	require.NoError(t, err)
+	require.Len(t, memo2.RelationList, 0)
+	memoRelation, err := s.postMemoRelationUpsert(memo2.ID, &api.MemoRelationUpsert{
+		RelatedMemoID: memo.ID,
+		Type:          api.MemoRelationReference,
+	})
+	require.NoError(t, err)
+	require.Equal(t, memo.ID, memoRelation.RelatedMemoID)
+	memo2, err = s.getMemo(memo2.ID)
+	require.NoError(t, err)
+	require.Len(t, memo2.RelationList, 1)
+}
+
+func (s *TestingServer) postMemoRelationUpsert(memoID int, memoRelationUpsert *api.MemoRelationUpsert) (*api.MemoRelation, error) {
+	rawData, err := json.Marshal(&memoRelationUpsert)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to marshal memo relation upsert")
+	}
+	reader := bytes.NewReader(rawData)
+	body, err := s.post(fmt.Sprintf("/api/memo/%d/relation", memoID), reader, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	buf := &bytes.Buffer{}
+	_, err = buf.ReadFrom(body)
+	if err != nil {
+		return nil, errors.Wrap(err, "fail to read response body")
+	}
+
+	type MemoCreateResponse struct {
+		Data *api.MemoRelation `json:"data"`
+	}
+	res := new(MemoCreateResponse)
+	if err = json.Unmarshal(buf.Bytes(), res); err != nil {
+		return nil, errors.Wrap(err, "fail to unmarshal post memo relation upsert response")
+	}
+	return res.Data, nil
+}
+
+func (s *TestingServer) deleteMemoRelation(memoID int, relatedMemoID int, relationType api.MemoRelationType) error {
+	_, err := s.delete(fmt.Sprintf("/api/memo/%d/relation/%d/type/%s", memoID, relatedMemoID, relationType), nil)
+	return err
+}

test/server/memo_test.go

@@ -37,13 +37,13 @@ func TestMemoServer(t *testing.T) {
 	require.NoError(t, err)
 	require.Len(t, memoList, 1)
 	updatedContent := "updated memo"
-	memo, err = s.patchMemoPatch(&api.MemoPatch{
+	memo, err = s.patchMemo(&api.MemoPatch{
 		ID:      memo.ID,
 		Content: &updatedContent,
 	})
 	require.NoError(t, err)
 	require.Equal(t, updatedContent, memo.Content)
-	err = s.postMemoDelete(&api.MemoDelete{
+	err = s.deleteMemo(&api.MemoDelete{
 		ID: memo.ID,
 	})
 	require.NoError(t, err)
@@ -52,6 +52,28 @@ func TestMemoServer(t *testing.T) {
 	require.Len(t, memoList, 0)
 }
 
+func (s *TestingServer) getMemo(memoID int) (*api.Memo, error) {
+	body, err := s.get(fmt.Sprintf("/api/memo/%d", memoID), nil)
+	if err != nil {
+		return nil, err
+	}
+
+	buf := &bytes.Buffer{}
+	_, err = buf.ReadFrom(body)
+	if err != nil {
+		return nil, errors.Wrap(err, "fail to read response body")
+	}
+
+	type MemoCreateResponse struct {
+		Data *api.Memo `json:"data"`
+	}
+	res := new(MemoCreateResponse)
+	if err = json.Unmarshal(buf.Bytes(), res); err != nil {
+		return nil, errors.Wrap(err, "fail to unmarshal get memo response")
+	}
+	return res.Data, nil
+}
+
 func (s *TestingServer) getMemoList() ([]*api.Memo, error) {
 	body, err := s.get("/api/memo", nil)
 	if err != nil {
@@ -101,7 +123,7 @@ func (s *TestingServer) postMemoCreate(memoCreate *api.MemoCreate) (*api.Memo, e
 	return res.Data, nil
 }
 
-func (s *TestingServer) patchMemoPatch(memoPatch *api.MemoPatch) (*api.Memo, error) {
+func (s *TestingServer) patchMemo(memoPatch *api.MemoPatch) (*api.Memo, error) {
 	rawData, err := json.Marshal(&memoPatch)
 	if err != nil {
 		return nil, errors.Wrap(err, "failed to marshal memo patch")
@@ -128,7 +150,7 @@ func (s *TestingServer) patchMemoPatch(memoPatch *api.MemoPatch) (*api.Memo, err
 	return res.Data, nil
 }
 
-func (s *TestingServer) postMemoDelete(memoDelete *api.MemoDelete) error {
+func (s *TestingServer) deleteMemo(memoDelete *api.MemoDelete) error {
 	_, err := s.delete(fmt.Sprintf("/api/memo/%d", memoDelete.ID), nil)
 	return err
 }