2023-07-04 13:48:52 +00:00
|
|
|
package protocol
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"fmt"
|
|
|
|
"strings"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
|
|
|
|
"github.com/status-im/status-go/eth-node/crypto"
|
|
|
|
"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/urls"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestMessengerShareUrlsSuite(t *testing.T) {
|
|
|
|
suite.Run(t, new(MessengerShareUrlsSuite))
|
|
|
|
}
|
|
|
|
|
|
|
|
type MessengerShareUrlsSuite struct {
|
2023-07-13 10:28:34 +00:00
|
|
|
MessengerBaseTestSuite
|
2023-07-04 13:48:52 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) createCommunity() *communities.Community {
|
|
|
|
description := &requests.CreateCommunity{
|
|
|
|
Membership: protobuf.CommunityPermissions_NO_MEMBERSHIP,
|
|
|
|
Name: "status",
|
|
|
|
Color: "#ffffff",
|
|
|
|
Description: "status community description",
|
|
|
|
}
|
|
|
|
|
|
|
|
// Create an community chat
|
|
|
|
response, err := s.m.CreateCommunity(description, false)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NotNil(response)
|
|
|
|
return response.Communities()[0]
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) createContact() (*Messenger, *Contact) {
|
|
|
|
theirMessenger := s.newMessenger()
|
|
|
|
_, err := theirMessenger.Start()
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))
|
|
|
|
ensName := "blah.stateofus.eth"
|
|
|
|
|
|
|
|
s.Require().NoError(s.m.ENSVerified(contactID, ensName))
|
|
|
|
|
|
|
|
response, err := s.m.AddContact(context.Background(), &requests.AddContact{ID: contactID})
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NotNil(response)
|
|
|
|
s.Require().Len(response.Contacts, 1)
|
|
|
|
|
|
|
|
contact := response.Contacts[0]
|
|
|
|
s.Require().Equal(ensName, contact.EnsName)
|
|
|
|
s.Require().True(contact.ENSVerified)
|
|
|
|
|
|
|
|
return theirMessenger, contact
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) createCommunityWithChannel() (*communities.Community, *protobuf.CommunityChat, string) {
|
|
|
|
community := s.createCommunity()
|
|
|
|
|
|
|
|
chat := &protobuf.CommunityChat{
|
|
|
|
Permissions: &protobuf.CommunityPermissions{
|
|
|
|
Access: protobuf.CommunityPermissions_NO_MEMBERSHIP,
|
|
|
|
},
|
|
|
|
Identity: &protobuf.ChatIdentity{
|
|
|
|
DisplayName: "status-core",
|
|
|
|
Emoji: "😎",
|
|
|
|
Description: "status-core community chat",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
response, err := s.m.CreateCommunityChat(community.ID(), chat)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NotNil(response)
|
|
|
|
s.Require().Len(response.Communities(), 1)
|
|
|
|
s.Require().Len(response.Chats(), 1)
|
|
|
|
|
|
|
|
community = response.Communities()[0]
|
|
|
|
s.Require().Len(community.Chats(), 1)
|
|
|
|
|
|
|
|
var channelID string
|
|
|
|
var channel *protobuf.CommunityChat
|
|
|
|
|
|
|
|
for key, value := range community.Chats() {
|
|
|
|
channelID = key
|
|
|
|
channel = value
|
|
|
|
break
|
|
|
|
}
|
|
|
|
s.Require().NotNil(channel)
|
|
|
|
return community, channel, channelID
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestDecodeEncodeDataURL() {
|
|
|
|
ts := [][]byte{
|
|
|
|
[]byte("test data 123"),
|
|
|
|
[]byte("test data 123test data 123test data 123test data 123test data 123"),
|
|
|
|
}
|
|
|
|
|
|
|
|
for i := range ts {
|
|
|
|
encodedData, err := urls.EncodeDataURL(ts[i])
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
decodedData, err := urls.DecodeDataURL(encodedData)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Equal(ts[i], decodedData)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestSerializePublicKey() {
|
|
|
|
key, err := crypto.GenerateKey()
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
serializedKey, err := s.m.SerializePublicKey(crypto.CompressPubkey(&key.PublicKey))
|
|
|
|
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Len(serializedKey, 49)
|
|
|
|
s.Require().True(strings.HasPrefix(serializedKey, "zQ3sh"))
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestDeserializePublicKey() {
|
|
|
|
serializedKey := "zQ3shPyZJnxZK4Bwyx9QsaksNKDYTPmpwPvGSjMYVHoXHeEgB"
|
|
|
|
|
|
|
|
shortKey, err := s.m.DeserializePublicKey(serializedKey)
|
|
|
|
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Len(shortKey, 33)
|
|
|
|
s.Require().True(strings.HasPrefix(shortKey.String(), "0x"))
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestParseWrongUrls() {
|
|
|
|
urls := map[string]string{
|
|
|
|
"https://status.appc/#zQ3shYSHp7GoiXaauJMnDcjwU2yNjdzpXLosAWapPS4CFxc11": "unhandled shared url",
|
|
|
|
"https://status.app/cc#zQ3shYSHp7GoiXaauJMnDcjwU2yNjdzpXLosAWapPS4CFxc11": "unhandled shared url",
|
|
|
|
"https://status.app/a#zQ3shYSHp7GoiXaauJMnDcjwU2yNjdzpXLosAWapPS4CFxc11": "unhandled shared url",
|
|
|
|
"https://status.app/u/": "url should contain at least one `#` separator",
|
|
|
|
"https://status.im/u#zQ3shYSHp7GoiXaauJMnDcjwU2yNjdzpXLosAWapPS4CFxc11": "url should start with 'https://status.app'",
|
|
|
|
}
|
|
|
|
|
|
|
|
for url, expectedError := range urls {
|
|
|
|
urlData, err := s.m.ParseSharedURL(url)
|
|
|
|
s.Require().Error(err)
|
|
|
|
|
|
|
|
s.Require().True(strings.HasPrefix(err.Error(), expectedError))
|
|
|
|
s.Require().Nil(urlData)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestShareCommunityURLWithChatKey() {
|
|
|
|
community := s.createCommunity()
|
|
|
|
|
|
|
|
url, err := s.m.ShareCommunityURLWithChatKey(community.ID())
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
shortKey, err := s.m.SerializePublicKey(community.ID())
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
expectedURL := fmt.Sprintf("%s/c#%s", baseShareURL, shortKey)
|
|
|
|
s.Require().Equal(expectedURL, url)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestParseCommunityURLWithChatKey() {
|
|
|
|
community := s.createCommunity()
|
|
|
|
|
|
|
|
shortKey, err := s.m.SerializePublicKey(community.ID())
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
url := fmt.Sprintf("%s/c#%s", baseShareURL, shortKey)
|
|
|
|
|
|
|
|
urlData, err := s.m.ParseSharedURL(url)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NotNil(urlData)
|
|
|
|
|
|
|
|
s.Require().NotNil(urlData.Community)
|
|
|
|
s.Require().Equal(community.Identity().DisplayName, urlData.Community.DisplayName)
|
|
|
|
s.Require().Equal(community.DescriptionText(), urlData.Community.Description)
|
|
|
|
s.Require().Equal(uint32(community.MembersCount()), urlData.Community.MembersCount)
|
|
|
|
s.Require().Equal(community.Identity().GetColor(), urlData.Community.Color)
|
|
|
|
s.Require().Equal(community.TagsIndices(), urlData.Community.TagIndices)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestShareCommunityURLWithData() {
|
|
|
|
community := s.createCommunity()
|
|
|
|
|
|
|
|
url, err := s.m.ShareCommunityURLWithData(community.ID())
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
communityData, signature, err := s.m.prepareEncodedCommunityData(community)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
expectedURL := fmt.Sprintf("%s/c/%s#%s", baseShareURL, communityData, signature)
|
|
|
|
s.Require().Equal(expectedURL, url)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestParseCommunityURLWithData() {
|
|
|
|
community := s.createCommunity()
|
|
|
|
|
|
|
|
communityData, signature, err := s.m.prepareEncodedCommunityData(community)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
url := fmt.Sprintf("%s/c/%s#%s", baseShareURL, communityData, signature)
|
|
|
|
|
|
|
|
urlData, err := s.m.ParseSharedURL(url)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NotNil(urlData)
|
|
|
|
|
|
|
|
s.Require().NotNil(urlData.Community)
|
|
|
|
s.Require().Equal(community.Identity().DisplayName, urlData.Community.DisplayName)
|
|
|
|
s.Require().Equal(community.DescriptionText(), urlData.Community.Description)
|
|
|
|
s.Require().Equal(uint32(community.MembersCount()), urlData.Community.MembersCount)
|
|
|
|
s.Require().Equal(community.Identity().GetColor(), urlData.Community.Color)
|
|
|
|
s.Require().Equal(community.TagsIndices(), urlData.Community.TagIndices)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestShareCommunityChannelURLWithChatKey() {
|
|
|
|
community := s.createCommunity()
|
|
|
|
channelID := "003cdcd5-e065-48f9-b166-b1a94ac75a11"
|
|
|
|
|
|
|
|
request := &requests.CommunityChannelShareURL{
|
|
|
|
CommunityID: community.ID(),
|
|
|
|
ChannelID: channelID,
|
|
|
|
}
|
|
|
|
url, err := s.m.ShareCommunityChannelURLWithChatKey(request)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
shortKey, err := s.m.SerializePublicKey(community.ID())
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
expectedURL := fmt.Sprintf("%s/cc/%s#%s", baseShareURL, channelID, shortKey)
|
|
|
|
s.Require().Equal(expectedURL, url)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestParseCommunityChannelURLWithChatKey() {
|
|
|
|
community, channel, channelID := s.createCommunityWithChannel()
|
|
|
|
|
|
|
|
shortKey, err := s.m.SerializePublicKey(community.ID())
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
url := fmt.Sprintf("%s/cc/%s#%s", baseShareURL, channelID, shortKey)
|
|
|
|
|
|
|
|
urlData, err := s.m.ParseSharedURL(url)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NotNil(urlData)
|
|
|
|
|
|
|
|
s.Require().NotNil(urlData.Community)
|
|
|
|
s.Require().Equal(community.Identity().DisplayName, urlData.Community.DisplayName)
|
|
|
|
s.Require().Equal(community.DescriptionText(), urlData.Community.Description)
|
|
|
|
s.Require().Equal(uint32(community.MembersCount()), urlData.Community.MembersCount)
|
|
|
|
s.Require().Equal(community.Identity().GetColor(), urlData.Community.Color)
|
|
|
|
s.Require().Equal(community.TagsIndices(), urlData.Community.TagIndices)
|
|
|
|
|
|
|
|
s.Require().NotNil(urlData.Channel)
|
|
|
|
s.Require().Equal(channel.Identity.Emoji, urlData.Channel.Emoji)
|
|
|
|
s.Require().Equal(channel.Identity.DisplayName, urlData.Channel.DisplayName)
|
|
|
|
s.Require().Equal(channel.Identity.Color, urlData.Channel.Color)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestShareCommunityChannelURLWithData() {
|
|
|
|
community, channel, channelID := s.createCommunityWithChannel()
|
|
|
|
|
|
|
|
request := &requests.CommunityChannelShareURL{
|
|
|
|
CommunityID: community.ID(),
|
|
|
|
ChannelID: channelID,
|
|
|
|
}
|
|
|
|
url, err := s.m.ShareCommunityChannelURLWithData(request)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
communityData, signature, err := s.m.prepareEncodedCommunityChannelData(community, channel, channelID)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
expectedURL := fmt.Sprintf("%s/cc/%s#%s", baseShareURL, communityData, signature)
|
|
|
|
s.Require().Equal(expectedURL, url)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestParseCommunityChannelURLWithData() {
|
|
|
|
community, channel, channelID := s.createCommunityWithChannel()
|
|
|
|
|
|
|
|
communityChannelData, signature, err := s.m.prepareEncodedCommunityChannelData(community, channel, channelID)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
url := fmt.Sprintf("%s/cc/%s#%s", baseShareURL, communityChannelData, signature)
|
|
|
|
|
|
|
|
urlData, err := s.m.ParseSharedURL(url)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NotNil(urlData)
|
|
|
|
|
|
|
|
s.Require().NotNil(urlData.Community)
|
|
|
|
s.Require().Equal(community.Identity().DisplayName, urlData.Community.DisplayName)
|
|
|
|
s.Require().Equal(community.DescriptionText(), urlData.Community.Description)
|
|
|
|
s.Require().Equal(uint32(community.MembersCount()), urlData.Community.MembersCount)
|
|
|
|
s.Require().Equal(community.Identity().GetColor(), urlData.Community.Color)
|
|
|
|
s.Require().Equal(community.TagsIndices(), urlData.Community.TagIndices)
|
|
|
|
|
|
|
|
s.Require().NotNil(urlData.Channel)
|
|
|
|
s.Require().Equal(channel.Identity.Emoji, urlData.Channel.Emoji)
|
|
|
|
s.Require().Equal(channel.Identity.DisplayName, urlData.Channel.DisplayName)
|
|
|
|
s.Require().Equal(channel.Identity.Color, urlData.Channel.Color)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestShareUserURLWithChatKey() {
|
|
|
|
_, contact := s.createContact()
|
|
|
|
|
|
|
|
url, err := s.m.ShareUserURLWithChatKey(contact.ID)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
publicKey, err := common.HexToPubkey(contact.ID)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
shortKey, err := s.m.SerializePublicKey(crypto.CompressPubkey(publicKey))
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
expectedURL := fmt.Sprintf("%s/u#%s", baseShareURL, shortKey)
|
|
|
|
s.Require().Equal(expectedURL, url)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestParseUserURLWithChatKey() {
|
|
|
|
_, contact := s.createContact()
|
|
|
|
|
|
|
|
publicKey, err := common.HexToPubkey(contact.ID)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
shortKey, err := s.m.SerializePublicKey(crypto.CompressPubkey(publicKey))
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
url := fmt.Sprintf("%s/u#%s", baseShareURL, shortKey)
|
|
|
|
|
|
|
|
urlData, err := s.m.ParseSharedURL(url)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NotNil(urlData)
|
|
|
|
|
|
|
|
s.Require().NotNil(urlData.Contact)
|
|
|
|
s.Require().Equal(contact.DisplayName, urlData.Contact.DisplayName)
|
|
|
|
s.Require().Equal(contact.Bio, urlData.Contact.DisplayName)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestShareUserURLWithENS() {
|
|
|
|
_, contact := s.createContact()
|
|
|
|
|
|
|
|
url, err := s.m.ShareUserURLWithENS(contact.ID)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
expectedURL := fmt.Sprintf("%s/u#%s", baseShareURL, contact.EnsName)
|
|
|
|
s.Require().Equal(expectedURL, url)
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: ens in the next ticket
|
|
|
|
// func (s *MessengerShareUrlsSuite) TestParseUserURLWithENS() {
|
|
|
|
// _, contact := s.createContact()
|
|
|
|
|
|
|
|
// url := fmt.Sprintf("%s/u#%s", baseShareURL, contact.EnsName)
|
|
|
|
|
|
|
|
// urlData, err := s.m.ParseSharedURL(url)
|
|
|
|
// s.Require().NoError(err)
|
|
|
|
// s.Require().NotNil(urlData)
|
|
|
|
|
|
|
|
// s.Require().NotNil(urlData.Contact)
|
|
|
|
// s.Require().Equal(contact.DisplayName, urlData.Contact.DisplayName)
|
|
|
|
// s.Require().Equal(contact.Bio, urlData.Contact.DisplayName)
|
|
|
|
// }
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestShareUserURLWithData() {
|
|
|
|
_, contact := s.createContact()
|
|
|
|
|
|
|
|
url, err := s.m.ShareUserURLWithData(contact.ID)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
userData, signature, err := s.m.prepareEncodedUserData(contact)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
expectedURL := fmt.Sprintf("%s/u/%s#%s", baseShareURL, userData, signature)
|
|
|
|
s.Require().Equal(expectedURL, url)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *MessengerShareUrlsSuite) TestParseUserURLWithData() {
|
|
|
|
_, contact := s.createContact()
|
|
|
|
|
|
|
|
userData, signature, err := s.m.prepareEncodedUserData(contact)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
|
|
|
|
url := fmt.Sprintf("%s/u/%s#%s", baseShareURL, userData, signature)
|
|
|
|
|
|
|
|
urlData, err := s.m.ParseSharedURL(url)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NotNil(urlData)
|
|
|
|
|
|
|
|
s.Require().NotNil(urlData.Contact)
|
|
|
|
s.Require().Equal(contact.DisplayName, urlData.Contact.DisplayName)
|
|
|
|
s.Require().Equal(contact.Bio, urlData.Contact.Description)
|
|
|
|
}
|