377 lines
12 KiB
Go
377 lines
12 KiB
Go
package protocol
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
func TestProfileShowcasePersistenceSuite(t *testing.T) {
|
|
suite.Run(t, new(TestProfileShowcasePersistence))
|
|
}
|
|
|
|
type TestProfileShowcasePersistence struct {
|
|
suite.Suite
|
|
}
|
|
|
|
func (s *TestProfileShowcasePersistence) TestProfileShowcasePreferences() {
|
|
db, err := openTestDB()
|
|
s.Require().NoError(err)
|
|
persistence := newSQLitePersistence(db)
|
|
|
|
preferences := &ProfileShowcasePreferences{
|
|
Communities: []*ProfileShowcaseCommunityPreference{
|
|
&ProfileShowcaseCommunityPreference{
|
|
CommunityID: "0x32433445133424",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityEveryone,
|
|
Order: 0,
|
|
},
|
|
},
|
|
Accounts: []*ProfileShowcaseAccountPreference{
|
|
&ProfileShowcaseAccountPreference{
|
|
Address: "0x32433445133424",
|
|
Name: "Status Account",
|
|
ColorID: "blue",
|
|
Emoji: "-_-",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityEveryone,
|
|
Order: 0,
|
|
},
|
|
&ProfileShowcaseAccountPreference{
|
|
Address: "0x3845354643324",
|
|
Name: "Money Box",
|
|
ColorID: "red",
|
|
Emoji: ":o)",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityContacts,
|
|
Order: 1,
|
|
},
|
|
},
|
|
Collectibles: []*ProfileShowcaseCollectiblePreference{
|
|
&ProfileShowcaseCollectiblePreference{
|
|
ContractAddress: "0x12378534257568678487683576",
|
|
ChainID: 3,
|
|
TokenID: "0x12321389592999f903",
|
|
CommunityID: "0x01312357798976535",
|
|
AccountAddress: "0x32433445133424",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityEveryone,
|
|
Order: 0,
|
|
},
|
|
},
|
|
VerifiedTokens: []*ProfileShowcaseVerifiedTokenPreference{
|
|
&ProfileShowcaseVerifiedTokenPreference{
|
|
Symbol: "ETH",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityEveryone,
|
|
Order: 1,
|
|
},
|
|
&ProfileShowcaseVerifiedTokenPreference{
|
|
Symbol: "DAI",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityIDVerifiedContacts,
|
|
Order: 2,
|
|
},
|
|
&ProfileShowcaseVerifiedTokenPreference{
|
|
Symbol: "SNT",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityNoOne,
|
|
Order: 3,
|
|
},
|
|
},
|
|
UnverifiedTokens: []*ProfileShowcaseUnverifiedTokenPreference{
|
|
&ProfileShowcaseUnverifiedTokenPreference{
|
|
ContractAddress: "0x454525452023452",
|
|
ChainID: 1,
|
|
CommunityID: "0x32433445133424",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityEveryone,
|
|
Order: 0,
|
|
},
|
|
&ProfileShowcaseUnverifiedTokenPreference{
|
|
ContractAddress: "0x12312323323233",
|
|
ChainID: 2,
|
|
CommunityID: "",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityContacts,
|
|
Order: 1,
|
|
},
|
|
},
|
|
}
|
|
|
|
err = persistence.SaveProfileShowcasePreferences(preferences)
|
|
s.Require().NoError(err)
|
|
|
|
preferencesBack, err := persistence.GetProfileShowcasePreferences()
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Equal(len(preferencesBack.Communities), len(preferences.Communities))
|
|
for i := 0; i < len(preferences.Communities); i++ {
|
|
s.Require().Equal(*preferences.Communities[i], *preferencesBack.Communities[i])
|
|
}
|
|
|
|
s.Require().Equal(len(preferencesBack.Accounts), len(preferences.Accounts))
|
|
for i := 0; i < len(preferences.Accounts); i++ {
|
|
s.Require().Equal(*preferences.Accounts[i], *preferencesBack.Accounts[i])
|
|
}
|
|
|
|
s.Require().Equal(len(preferencesBack.Collectibles), len(preferences.Collectibles))
|
|
for i := 0; i < len(preferences.Collectibles); i++ {
|
|
s.Require().Equal(*preferences.Collectibles[i], *preferencesBack.Collectibles[i])
|
|
}
|
|
|
|
s.Require().Equal(len(preferencesBack.VerifiedTokens), len(preferences.VerifiedTokens))
|
|
for i := 0; i < len(preferences.VerifiedTokens); i++ {
|
|
s.Require().Equal(*preferences.VerifiedTokens[i], *preferencesBack.VerifiedTokens[i])
|
|
}
|
|
|
|
s.Require().Equal(len(preferencesBack.UnverifiedTokens), len(preferences.UnverifiedTokens))
|
|
for i := 0; i < len(preferences.UnverifiedTokens); i++ {
|
|
s.Require().Equal(*preferences.UnverifiedTokens[i], *preferencesBack.UnverifiedTokens[i])
|
|
}
|
|
}
|
|
|
|
func (s *TestProfileShowcasePersistence) TestProfileShowcaseContacts() {
|
|
db, err := openTestDB()
|
|
s.Require().NoError(err)
|
|
persistence := newSQLitePersistence(db)
|
|
|
|
showcase1 := &ProfileShowcase{
|
|
ContactID: "contact_1",
|
|
Communities: []*ProfileShowcaseCommunity{
|
|
&ProfileShowcaseCommunity{
|
|
CommunityID: "0x012312234234234",
|
|
Order: 6,
|
|
},
|
|
&ProfileShowcaseCommunity{
|
|
CommunityID: "0x04523233466753",
|
|
Order: 7,
|
|
},
|
|
},
|
|
Accounts: []*ProfileShowcaseAccount{
|
|
&ProfileShowcaseAccount{
|
|
ContactID: "contact_1",
|
|
Address: "0x32433445133424",
|
|
Name: "Status Account",
|
|
ColorID: "blue",
|
|
Emoji: "-_-",
|
|
Order: 0,
|
|
},
|
|
&ProfileShowcaseAccount{
|
|
ContactID: "contact_1",
|
|
Address: "0x3845354643324",
|
|
Name: "Money Box",
|
|
ColorID: "red",
|
|
Emoji: ":o)",
|
|
Order: 1,
|
|
},
|
|
},
|
|
Collectibles: []*ProfileShowcaseCollectible{
|
|
&ProfileShowcaseCollectible{
|
|
ContractAddress: "0x12378534257568678487683576",
|
|
ChainID: 2,
|
|
TokenID: "0x12321389592999f903",
|
|
CommunityID: "0x01312357798976535",
|
|
Order: 0,
|
|
},
|
|
},
|
|
VerifiedTokens: []*ProfileShowcaseVerifiedToken{
|
|
&ProfileShowcaseVerifiedToken{
|
|
Symbol: "ETH",
|
|
Order: 1,
|
|
},
|
|
&ProfileShowcaseVerifiedToken{
|
|
Symbol: "DAI",
|
|
Order: 2,
|
|
},
|
|
&ProfileShowcaseVerifiedToken{
|
|
Symbol: "SNT",
|
|
Order: 3,
|
|
},
|
|
},
|
|
UnverifiedTokens: []*ProfileShowcaseUnverifiedToken{
|
|
&ProfileShowcaseUnverifiedToken{
|
|
ContractAddress: "0x454525452023452",
|
|
ChainID: 1,
|
|
CommunityID: "",
|
|
Order: 0,
|
|
},
|
|
&ProfileShowcaseUnverifiedToken{
|
|
ContractAddress: "0x12312323323233",
|
|
ChainID: 2,
|
|
CommunityID: "0x32433445133424",
|
|
Order: 1,
|
|
},
|
|
},
|
|
}
|
|
err = persistence.SaveProfileShowcaseForContact(showcase1)
|
|
s.Require().NoError(err)
|
|
|
|
showcase2 := &ProfileShowcase{
|
|
ContactID: "contact_2",
|
|
Communities: []*ProfileShowcaseCommunity{
|
|
&ProfileShowcaseCommunity{
|
|
CommunityID: "0x012312234234234", // same id to check query
|
|
Order: 3,
|
|
},
|
|
&ProfileShowcaseCommunity{
|
|
CommunityID: "0x096783478384593",
|
|
Order: 7,
|
|
},
|
|
},
|
|
Collectibles: []*ProfileShowcaseCollectible{
|
|
&ProfileShowcaseCollectible{
|
|
ContractAddress: "0x12378534257568678487683576",
|
|
ChainID: 2,
|
|
TokenID: "0x12321389592999f903",
|
|
CommunityID: "0x01312357798976535",
|
|
Order: 1,
|
|
},
|
|
},
|
|
}
|
|
err = persistence.SaveProfileShowcaseForContact(showcase2)
|
|
s.Require().NoError(err)
|
|
|
|
showcase1Back, err := persistence.GetProfileShowcaseForContact("contact_1")
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Equal(len(showcase1.Communities), len(showcase1Back.Communities))
|
|
for i := 0; i < len(showcase1.Communities); i++ {
|
|
s.Require().Equal(*showcase1.Communities[i], *showcase1Back.Communities[i])
|
|
}
|
|
s.Require().Equal(len(showcase1.Accounts), len(showcase1Back.Accounts))
|
|
for i := 0; i < len(showcase1.Accounts); i++ {
|
|
s.Require().Equal(*showcase1.Accounts[i], *showcase1Back.Accounts[i])
|
|
}
|
|
s.Require().Equal(len(showcase1.Collectibles), len(showcase1Back.Collectibles))
|
|
for i := 0; i < len(showcase1.Collectibles); i++ {
|
|
s.Require().Equal(*showcase1.Collectibles[i], *showcase1Back.Collectibles[i])
|
|
}
|
|
s.Require().Equal(len(showcase1.VerifiedTokens), len(showcase1Back.VerifiedTokens))
|
|
for i := 0; i < len(showcase1.VerifiedTokens); i++ {
|
|
s.Require().Equal(*showcase1.VerifiedTokens[i], *showcase1Back.VerifiedTokens[i])
|
|
}
|
|
s.Require().Equal(len(showcase1.UnverifiedTokens), len(showcase1Back.UnverifiedTokens))
|
|
for i := 0; i < len(showcase1.UnverifiedTokens); i++ {
|
|
s.Require().Equal(*showcase1.UnverifiedTokens[i], *showcase1Back.UnverifiedTokens[i])
|
|
}
|
|
|
|
showcase2Back, err := persistence.GetProfileShowcaseForContact("contact_2")
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Equal(len(showcase2.Communities), len(showcase2Back.Communities))
|
|
s.Require().Equal(*showcase2.Communities[0], *showcase2Back.Communities[0])
|
|
s.Require().Equal(*showcase2.Communities[1], *showcase2Back.Communities[1])
|
|
s.Require().Equal(len(showcase2.Collectibles), len(showcase2Back.Collectibles))
|
|
s.Require().Equal(*showcase2.Collectibles[0], *showcase2Back.Collectibles[0])
|
|
s.Require().Equal(0, len(showcase2Back.Accounts))
|
|
s.Require().Equal(0, len(showcase2Back.VerifiedTokens))
|
|
s.Require().Equal(0, len(showcase2Back.UnverifiedTokens))
|
|
}
|
|
|
|
func (s *TestProfileShowcasePersistence) TestFetchingProfileShowcaseAccountsByAddress() {
|
|
db, err := openTestDB()
|
|
s.Require().NoError(err)
|
|
persistence := newSQLitePersistence(db)
|
|
|
|
conatacts := []*Contact{
|
|
&Contact{
|
|
ID: "contact_1",
|
|
},
|
|
&Contact{
|
|
ID: "contact_2",
|
|
},
|
|
&Contact{
|
|
ID: "contact_3",
|
|
},
|
|
}
|
|
|
|
err = persistence.SaveContacts(conatacts)
|
|
s.Require().NoError(err)
|
|
|
|
showcase1 := &ProfileShowcase{
|
|
ContactID: "contact_1",
|
|
Accounts: []*ProfileShowcaseAccount{
|
|
&ProfileShowcaseAccount{
|
|
ContactID: "contact_1",
|
|
Address: "0x0000000000000000000000000000000000000001",
|
|
Name: "Contact1-Account1",
|
|
ColorID: "blue",
|
|
Emoji: "-_-",
|
|
Order: 0,
|
|
},
|
|
&ProfileShowcaseAccount{
|
|
ContactID: "contact_1",
|
|
Address: "0x0000000000000000000000000000000000000002",
|
|
Name: "Contact1-Account2",
|
|
ColorID: "blue",
|
|
Emoji: "-_-",
|
|
Order: 1,
|
|
},
|
|
},
|
|
}
|
|
showcase2 := &ProfileShowcase{
|
|
ContactID: "contact_2",
|
|
Accounts: []*ProfileShowcaseAccount{
|
|
&ProfileShowcaseAccount{
|
|
ContactID: "contact_2",
|
|
Address: "0x0000000000000000000000000000000000000001",
|
|
Name: "Contact2-Account1",
|
|
ColorID: "blue",
|
|
Emoji: "-_-",
|
|
Order: 0,
|
|
},
|
|
&ProfileShowcaseAccount{
|
|
ContactID: "contact_2",
|
|
Address: "0x0000000000000000000000000000000000000002",
|
|
Name: "Contact2-Account2",
|
|
ColorID: "blue",
|
|
Emoji: "-_-",
|
|
Order: 1,
|
|
},
|
|
},
|
|
}
|
|
showcase3 := &ProfileShowcase{
|
|
ContactID: "contact_3",
|
|
Accounts: []*ProfileShowcaseAccount{
|
|
&ProfileShowcaseAccount{
|
|
ContactID: "contact_3",
|
|
Address: "0x0000000000000000000000000000000000000001",
|
|
Name: "Contact3-Account1",
|
|
ColorID: "blue",
|
|
Emoji: "-_-",
|
|
Order: 0,
|
|
},
|
|
},
|
|
}
|
|
|
|
err = persistence.SaveProfileShowcaseForContact(showcase1)
|
|
s.Require().NoError(err)
|
|
err = persistence.SaveProfileShowcaseForContact(showcase2)
|
|
s.Require().NoError(err)
|
|
err = persistence.SaveProfileShowcaseForContact(showcase3)
|
|
s.Require().NoError(err)
|
|
|
|
showcaseAccounts, err := persistence.GetProfileShowcaseAccountsByAddress(showcase1.Accounts[0].Address)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Equal(3, len(showcaseAccounts))
|
|
for i := 0; i < len(showcaseAccounts); i++ {
|
|
if showcaseAccounts[i].ContactID == showcase1.ContactID {
|
|
s.Require().Equal(showcase1.Accounts[0].Address, showcase1.Accounts[0].Address)
|
|
} else if showcaseAccounts[i].ContactID == showcase2.ContactID {
|
|
s.Require().Equal(showcase2.Accounts[0].Address, showcase2.Accounts[0].Address)
|
|
} else if showcaseAccounts[i].ContactID == showcase3.ContactID {
|
|
s.Require().Equal(showcase3.Accounts[0].Address, showcase3.Accounts[0].Address)
|
|
} else {
|
|
s.Require().Fail("unexpected contact id")
|
|
}
|
|
}
|
|
|
|
showcaseAccounts, err = persistence.GetProfileShowcaseAccountsByAddress(showcase1.Accounts[1].Address)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Equal(2, len(showcaseAccounts))
|
|
for i := 0; i < len(showcaseAccounts); i++ {
|
|
if showcaseAccounts[i].ContactID == showcase1.ContactID {
|
|
s.Require().Equal(showcase1.Accounts[0].Address, showcase1.Accounts[0].Address)
|
|
} else if showcaseAccounts[i].ContactID == showcase2.ContactID {
|
|
s.Require().Equal(showcase2.Accounts[0].Address, showcase2.Accounts[0].Address)
|
|
} else {
|
|
s.Require().Fail("unexpected contact id")
|
|
}
|
|
}
|
|
}
|