Long expectedDraftBoxCount =
messageApi.findUserMessage(senderId, MessageState.draft_box, null).getTotalElements() + 1;
Long expectedTrashBoxCount =
messageApi.findUserMessage(senderId, MessageState.trash_box, null).getTotalElements() + 1;
Message message = new Message();
message.setSenderId(senderId);
message.setReceiverId(receiverId);
message.setTitle("ded");
MessageContent content = new MessageContent();
content.setContent("abcde");
message.setContent(content);
message.setSenderState(MessageState.draft_box);
messageService.save(message);
Long actualDraftBoxCount = messageApi.findUserMessage(senderId, MessageState.draft_box, null).getTotalElements();
Assert.assertEquals(expectedDraftBoxCount, actualDraftBoxCount);