193 lines
6.4 KiB
Go
193 lines
6.4 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
gethbridge "github.com/status-im/status-go/eth-node/bridge/geth"
|
|
"github.com/status-im/status-go/eth-node/types"
|
|
"github.com/status-im/status-go/protocol/common"
|
|
"github.com/status-im/status-go/protocol/communities"
|
|
"github.com/status-im/status-go/protocol/protobuf"
|
|
"github.com/status-im/status-go/protocol/requests"
|
|
"github.com/status-im/status-go/protocol/tt"
|
|
"github.com/status-im/status-go/waku"
|
|
)
|
|
|
|
func TestMessengerDeleteMessageForEveryoneSuite(t *testing.T) {
|
|
suite.Run(t, new(MessengerDeleteMessageForEveryoneSuite))
|
|
}
|
|
|
|
type MessengerDeleteMessageForEveryoneSuite struct {
|
|
suite.Suite
|
|
admin *Messenger
|
|
moderator *Messenger
|
|
bob *Messenger
|
|
shh types.Waku
|
|
logger *zap.Logger
|
|
}
|
|
|
|
func (s *MessengerDeleteMessageForEveryoneSuite) SetupTest() {
|
|
s.logger = tt.MustCreateTestLogger()
|
|
|
|
config := waku.DefaultConfig
|
|
config.MinimumAcceptedPoW = 0
|
|
shh := waku.New(&config, s.logger)
|
|
s.shh = gethbridge.NewGethWakuWrapper(shh)
|
|
s.Require().NoError(shh.Start())
|
|
|
|
s.admin = s.newMessenger()
|
|
s.bob = s.newMessenger()
|
|
s.moderator = s.newMessenger()
|
|
_, err := s.admin.Start()
|
|
s.Require().NoError(err)
|
|
_, err = s.bob.Start()
|
|
s.Require().NoError(err)
|
|
_, err = s.moderator.Start()
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
func (s *MessengerDeleteMessageForEveryoneSuite) TearDownTest() {
|
|
s.Require().NoError(s.admin.Shutdown())
|
|
s.Require().NoError(s.bob.Shutdown())
|
|
s.Require().NoError(s.moderator.Shutdown())
|
|
_ = s.logger.Sync()
|
|
}
|
|
|
|
func (s *MessengerDeleteMessageForEveryoneSuite) newMessenger() *Messenger {
|
|
privateKey, err := crypto.GenerateKey()
|
|
s.Require().NoError(err)
|
|
|
|
messenger, err := newMessengerWithKey(s.shh, privateKey, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
return messenger
|
|
}
|
|
|
|
func (s *MessengerDeleteMessageForEveryoneSuite) TestDeleteMessageForEveryone() {
|
|
community := s.createCommunity()
|
|
communityChat := s.createCommunityChat(community)
|
|
s.inviteAndJoin(community, s.moderator)
|
|
s.inviteAndJoin(community, s.bob)
|
|
|
|
response, err := s.admin.AddRoleToMember(&requests.AddRoleToMember{
|
|
CommunityID: community.ID(),
|
|
User: common.PubkeyToHexBytes(s.moderator.IdentityPublicKey()),
|
|
Role: protobuf.CommunityMember_ROLE_MODERATE_CONTENT,
|
|
})
|
|
s.Require().NoError(err)
|
|
s.Require().Len(response.Communities(), 1)
|
|
|
|
_, err = WaitOnMessengerResponse(s.moderator, func(response *MessengerResponse) bool {
|
|
return len(response.Communities()) > 0
|
|
}, "community description changed message not received")
|
|
s.Require().NoError(err)
|
|
_, err = WaitOnMessengerResponse(s.bob, func(response *MessengerResponse) bool {
|
|
return len(response.Communities()) > 0
|
|
}, "community description changed message not received")
|
|
s.Require().NoError(err)
|
|
|
|
ctx := context.Background()
|
|
inputMessage := &common.Message{}
|
|
inputMessage.ChatId = communityChat.ID
|
|
inputMessage.ContentType = protobuf.ChatMessage_TEXT_PLAIN
|
|
inputMessage.Text = "some text"
|
|
_, err = s.bob.SendChatMessage(ctx, inputMessage)
|
|
s.Require().NoError(err)
|
|
|
|
response, err = WaitOnMessengerResponse(s.moderator, func(response *MessengerResponse) bool {
|
|
return len(response.Messages()) > 0
|
|
}, "messages not received")
|
|
s.Require().NoError(err)
|
|
message := response.Messages()[0]
|
|
s.Require().Equal(inputMessage.Text, message.Text)
|
|
|
|
deleteMessageResponse, err := s.moderator.DeleteMessageAndSend(ctx, message.ID)
|
|
s.Require().NoError(err)
|
|
|
|
response, err = WaitOnMessengerResponse(s.bob, func(response *MessengerResponse) bool {
|
|
return len(response.RemovedMessages()) > 0
|
|
}, "removed messages not received")
|
|
s.Require().Equal(deleteMessageResponse.RemovedMessages()[0].DeletedBy, contactIDFromPublicKey(s.moderator.IdentityPublicKey()))
|
|
|
|
s.Require().NoError(err)
|
|
s.Require().Len(response.ActivityCenterNotifications(), 1)
|
|
s.Require().True(response.ActivityCenterNotifications()[0].Deleted)
|
|
|
|
message, err = s.bob.MessageByID(message.ID)
|
|
s.Require().NoError(err)
|
|
s.Require().True(message.Deleted)
|
|
}
|
|
|
|
func (s *MessengerDeleteMessageForEveryoneSuite) createCommunity() *communities.Community {
|
|
description := &requests.CreateCommunity{
|
|
Membership: protobuf.CommunityPermissions_NO_MEMBERSHIP,
|
|
Name: "status",
|
|
Color: "#ffffff",
|
|
Description: "status community description",
|
|
}
|
|
response, err := s.admin.CreateCommunity(description, false)
|
|
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(response)
|
|
s.Require().Len(response.Communities(), 1)
|
|
s.Require().Len(response.Chats(), 0)
|
|
|
|
return response.Communities()[0]
|
|
}
|
|
|
|
func (s *MessengerDeleteMessageForEveryoneSuite) createCommunityChat(community *communities.Community) *Chat {
|
|
orgChat := &protobuf.CommunityChat{
|
|
Permissions: &protobuf.CommunityPermissions{
|
|
Access: protobuf.CommunityPermissions_NO_MEMBERSHIP,
|
|
},
|
|
Identity: &protobuf.ChatIdentity{
|
|
DisplayName: "status-core",
|
|
Emoji: "",
|
|
Description: "status-core community chatToModerator",
|
|
},
|
|
}
|
|
|
|
response, err := s.admin.CreateCommunityChat(community.ID(), orgChat)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(response)
|
|
s.Require().Len(response.Communities(), 1)
|
|
s.Require().Len(response.Chats(), 1)
|
|
return response.Chats()[0]
|
|
}
|
|
|
|
func (s *MessengerDeleteMessageForEveryoneSuite) inviteAndJoin(community *communities.Community, target *Messenger) {
|
|
response, err := s.admin.InviteUsersToCommunity(&requests.InviteUsersToCommunity{
|
|
CommunityID: community.ID(),
|
|
Users: []types.HexBytes{common.PubkeyToHexBytes(&target.identity.PublicKey)},
|
|
})
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(response)
|
|
s.Require().Len(response.Communities(), 1)
|
|
|
|
community = response.Communities()[0]
|
|
s.Require().True(community.HasMember(&target.identity.PublicKey))
|
|
|
|
_, err = WaitOnMessengerResponse(target, func(response *MessengerResponse) bool {
|
|
return len(response.Communities()) > 0
|
|
}, "community not received")
|
|
s.Require().NoError(err)
|
|
|
|
response, err = target.JoinCommunity(context.Background(), community.ID(), false)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(response)
|
|
s.Require().Len(response.Communities(), 1)
|
|
s.Require().True(response.Communities()[0].Joined())
|
|
s.Require().Len(response.Chats(), 1)
|
|
|
|
s.Require().NoError(target.SaveChat(response.Chats()[0]))
|
|
|
|
_, err = WaitOnMessengerResponse(target, func(response *MessengerResponse) bool {
|
|
return len(response.Messages()) > 0
|
|
}, "message 'You have been invited to community' not received")
|
|
s.Require().NoError(err)
|
|
}
|