package app import ( "database/sql" "testing" "github.com/golang/mock/gomock" "github.com/stretchr/testify/require" mmModel "github.com/mattermost/mattermost-server/v6/model" "github.com/mattermost/focalboard/server/model" ) type blockError struct { msg string } func (be blockError) Error() string { return be.msg } func TestInsertBlock(t *testing.T) { th, tearDown := SetupTestHelper(t) defer tearDown() t.Run("success scenerio", func(t *testing.T) { boardID := testBoardID block := model.Block{BoardID: boardID} board := &model.Board{ID: boardID} th.Store.EXPECT().GetBoard(boardID).Return(board, nil) th.Store.EXPECT().InsertBlock(&block, "user-id-1").Return(nil) th.Store.EXPECT().GetMembersForBoard(boardID).Return([]*model.BoardMember{}, nil) err := th.App.InsertBlock(block, "user-id-1") require.NoError(t, err) }) t.Run("error scenerio", func(t *testing.T) { boardID := testBoardID block := model.Block{BoardID: boardID} board := &model.Board{ID: boardID} th.Store.EXPECT().GetBoard(boardID).Return(board, nil) th.Store.EXPECT().InsertBlock(&block, "user-id-1").Return(blockError{"error"}) err := th.App.InsertBlock(block, "user-id-1") require.Error(t, err, "error") }) } func TestPatchBlocks(t *testing.T) { th, tearDown := SetupTestHelper(t) defer tearDown() t.Run("patchBlocks success scenerio", func(t *testing.T) { blockPatches := model.BlockPatchBatch{ BlockIDs: []string{"block1"}, BlockPatches: []model.BlockPatch{ {Title: mmModel.NewString("new title")}, }, } block1 := model.Block{ID: "block1"} th.Store.EXPECT().GetBlocksByIDs([]string{"block1"}).Return([]model.Block{block1}, nil) th.Store.EXPECT().PatchBlocks(gomock.Eq(&blockPatches), gomock.Eq("user-id-1")).Return(nil) th.Store.EXPECT().GetBlock("block1").Return(&block1, nil) // this call comes from the WS server notification th.Store.EXPECT().GetMembersForBoard(gomock.Any()).Times(1) err := th.App.PatchBlocks("team-id", &blockPatches, "user-id-1") require.NoError(t, err) }) t.Run("patchBlocks error scenerio", func(t *testing.T) { blockPatches := model.BlockPatchBatch{BlockIDs: []string{}} th.Store.EXPECT().GetBlocksByIDs([]string{}).Return(nil, sql.ErrNoRows) err := th.App.PatchBlocks("team-id", &blockPatches, "user-id-1") require.ErrorIs(t, err, sql.ErrNoRows) }) t.Run("cloud limit error scenario", func(t *testing.T) { th.App.SetCardLimit(5) fakeLicense := &mmModel.License{ Features: &mmModel.Features{Cloud: mmModel.NewBool(true)}, } blockPatches := model.BlockPatchBatch{ BlockIDs: []string{"block1"}, BlockPatches: []model.BlockPatch{ {Title: mmModel.NewString("new title")}, }, } block1 := model.Block{ ID: "block1", Type: model.TypeCard, ParentID: "board-id", BoardID: "board-id", UpdateAt: 100, } board1 := &model.Board{ ID: "board-id", Type: model.BoardTypeOpen, } th.Store.EXPECT().GetBlocksByIDs([]string{"block1"}).Return([]model.Block{block1}, nil) th.Store.EXPECT().GetBoard("board-id").Return(board1, nil) th.Store.EXPECT().GetLicense().Return(fakeLicense) th.Store.EXPECT().GetCardLimitTimestamp().Return(int64(150), nil) err := th.App.PatchBlocks("team-id", &blockPatches, "user-id-1") require.ErrorIs(t, err, ErrPatchUpdatesLimitedCards) }) } func TestDeleteBlock(t *testing.T) { th, tearDown := SetupTestHelper(t) defer tearDown() t.Run("success scenerio", func(t *testing.T) { boardID := testBoardID board := &model.Board{ID: boardID} block := model.Block{ ID: "block-id", BoardID: board.ID, } th.Store.EXPECT().GetBlock(gomock.Eq("block-id")).Return(&block, nil) th.Store.EXPECT().DeleteBlock(gomock.Eq("block-id"), gomock.Eq("user-id-1")).Return(nil) th.Store.EXPECT().GetBoard(gomock.Eq(testBoardID)).Return(board, nil) th.Store.EXPECT().GetMembersForBoard(boardID).Return([]*model.BoardMember{}, nil) err := th.App.DeleteBlock("block-id", "user-id-1") require.NoError(t, err) }) t.Run("error scenerio", func(t *testing.T) { boardID := testBoardID board := &model.Board{ID: boardID} block := model.Block{ ID: "block-id", BoardID: board.ID, } th.Store.EXPECT().GetBlock(gomock.Eq("block-id")).Return(&block, nil) th.Store.EXPECT().DeleteBlock(gomock.Eq("block-id"), gomock.Eq("user-id-1")).Return(blockError{"error"}) th.Store.EXPECT().GetBoard(gomock.Eq(testBoardID)).Return(board, nil) err := th.App.DeleteBlock("block-id", "user-id-1") require.Error(t, err, "error") }) } func TestUndeleteBlock(t *testing.T) { th, tearDown := SetupTestHelper(t) defer tearDown() t.Run("success scenerio", func(t *testing.T) { boardID := testBoardID board := &model.Board{ID: boardID} block := model.Block{ ID: "block-id", BoardID: board.ID, } th.Store.EXPECT().GetBlockHistory( gomock.Eq("block-id"), gomock.Eq(model.QueryBlockHistoryOptions{Limit: 1, Descending: true}), ).Return([]model.Block{block}, nil) th.Store.EXPECT().UndeleteBlock(gomock.Eq("block-id"), gomock.Eq("user-id-1")).Return(nil) th.Store.EXPECT().GetBlock(gomock.Eq("block-id")).Return(&block, nil) th.Store.EXPECT().GetBoard(boardID).Return(board, nil) th.Store.EXPECT().GetMembersForBoard(boardID).Return([]*model.BoardMember{}, nil) _, err := th.App.UndeleteBlock("block-id", "user-id-1") require.NoError(t, err) }) t.Run("error scenerio", func(t *testing.T) { block := model.Block{ ID: "block-id", } th.Store.EXPECT().GetBlockHistory( gomock.Eq("block-id"), gomock.Eq(model.QueryBlockHistoryOptions{Limit: 1, Descending: true}), ).Return([]model.Block{block}, nil) th.Store.EXPECT().UndeleteBlock(gomock.Eq("block-id"), gomock.Eq("user-id-1")).Return(blockError{"error"}) th.Store.EXPECT().GetBlock(gomock.Eq("block-id")).Return(&block, nil) _, err := th.App.UndeleteBlock("block-id", "user-id-1") require.Error(t, err, "error") }) }