status-go/protocol/persistence_profile_showcase_test.go

package protocol

import (
	"testing"

	"github.com/stretchr/testify/suite"

	"github.com/status-im/status-go/protocol/identity"
)

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 := &identity.ProfileShowcasePreferences{
		Communities: []*identity.ProfileShowcaseCommunityPreference{
			&identity.ProfileShowcaseCommunityPreference{
				CommunityID:        "0x32433445133424",
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityEveryone,
				Order:              0,
			},
		},
		Accounts: []*identity.ProfileShowcaseAccountPreference{
			&identity.ProfileShowcaseAccountPreference{
				Address:            "0x32433445133424",
				Name:               "Status Account",
				ColorID:            "blue",
				Emoji:              "-_-",
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityEveryone,
				Order:              0,
			},
			&identity.ProfileShowcaseAccountPreference{
				Address:            "0x3845354643324",
				Name:               "Money Box",
				ColorID:            "red",
				Emoji:              ":o)",
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityContacts,
				Order:              1,
			},
		},
		Collectibles: []*identity.ProfileShowcaseCollectiblePreference{
			&identity.ProfileShowcaseCollectiblePreference{
				ContractAddress:    "0x12378534257568678487683576",
				ChainID:            3,
				TokenID:            "0x12321389592999f903",
				CommunityID:        "0x01312357798976535",
				AccountAddress:     "0x32433445133424",
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityEveryone,
				Order:              0,
			},
		},
		VerifiedTokens: []*identity.ProfileShowcaseVerifiedTokenPreference{
			&identity.ProfileShowcaseVerifiedTokenPreference{
				Symbol:             "ETH",
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityEveryone,
				Order:              1,
			},
			&identity.ProfileShowcaseVerifiedTokenPreference{
				Symbol:             "DAI",
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityIDVerifiedContacts,
				Order:              2,
			},
			&identity.ProfileShowcaseVerifiedTokenPreference{
				Symbol:             "SNT",
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityNoOne,
				Order:              3,
			},
		},
		UnverifiedTokens: []*identity.ProfileShowcaseUnverifiedTokenPreference{
			&identity.ProfileShowcaseUnverifiedTokenPreference{
				ContractAddress:    "0x454525452023452",
				ChainID:            1,
				CommunityID:        "0x32433445133424",
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityEveryone,
				Order:              0,
			},
			&identity.ProfileShowcaseUnverifiedTokenPreference{
				ContractAddress:    "0x12312323323233",
				ChainID:            2,
				CommunityID:        "",
				ShowcaseVisibility: identity.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 := &identity.ProfileShowcase{
		ContactID: "contact_1",
		Communities: []*identity.ProfileShowcaseCommunity{
			&identity.ProfileShowcaseCommunity{
				CommunityID: "0x012312234234234",
				Order:       6,
			},
			&identity.ProfileShowcaseCommunity{
				CommunityID: "0x04523233466753",
				Order:       7,
			},
		},
		Accounts: []*identity.ProfileShowcaseAccount{
			&identity.ProfileShowcaseAccount{
				ContactID: "contact_1",
				Address:   "0x32433445133424",
				Name:      "Status Account",
				ColorID:   "blue",
				Emoji:     "-_-",
				Order:     0,
			},
			&identity.ProfileShowcaseAccount{
				ContactID: "contact_1",
				Address:   "0x3845354643324",
				Name:      "Money Box",
				ColorID:   "red",
				Emoji:     ":o)",
				Order:     1,
			},
		},
		Collectibles: []*identity.ProfileShowcaseCollectible{
			&identity.ProfileShowcaseCollectible{
				ContractAddress: "0x12378534257568678487683576",
				ChainID:         2,
				TokenID:         "0x12321389592999f903",
				CommunityID:     "0x01312357798976535",
				Order:           0,
			},
		},
		VerifiedTokens: []*identity.ProfileShowcaseVerifiedToken{
			&identity.ProfileShowcaseVerifiedToken{
				Symbol: "ETH",
				Order:  1,
			},
			&identity.ProfileShowcaseVerifiedToken{
				Symbol: "DAI",
				Order:  2,
			},
			&identity.ProfileShowcaseVerifiedToken{
				Symbol: "SNT",
				Order:  3,
			},
		},
		UnverifiedTokens: []*identity.ProfileShowcaseUnverifiedToken{
			&identity.ProfileShowcaseUnverifiedToken{
				ContractAddress: "0x454525452023452",
				ChainID:         1,
				CommunityID:     "",
				Order:           0,
			},
			&identity.ProfileShowcaseUnverifiedToken{
				ContractAddress: "0x12312323323233",
				ChainID:         2,
				CommunityID:     "0x32433445133424",
				Order:           1,
			},
		},
	}
	err = persistence.SaveProfileShowcaseForContact(showcase1)
	s.Require().NoError(err)

	showcase2 := &identity.ProfileShowcase{
		ContactID: "contact_2",
		Communities: []*identity.ProfileShowcaseCommunity{
			&identity.ProfileShowcaseCommunity{
				CommunityID: "0x012312234234234", // same id to check query
				Order:       3,
			},
			&identity.ProfileShowcaseCommunity{
				CommunityID: "0x096783478384593",
				Order:       7,
			},
		},
		Collectibles: []*identity.ProfileShowcaseCollectible{
			&identity.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 := &identity.ProfileShowcase{
		ContactID: "contact_1",
		Accounts: []*identity.ProfileShowcaseAccount{
			&identity.ProfileShowcaseAccount{
				ContactID: "contact_1",
				Address:   "0x0000000000000000000000000000000000000001",
				Name:      "Contact1-Account1",
				ColorID:   "blue",
				Emoji:     "-_-",
				Order:     0,
			},
			&identity.ProfileShowcaseAccount{
				ContactID: "contact_1",
				Address:   "0x0000000000000000000000000000000000000002",
				Name:      "Contact1-Account2",
				ColorID:   "blue",
				Emoji:     "-_-",
				Order:     1,
			},
		},
	}
	showcase2 := &identity.ProfileShowcase{
		ContactID: "contact_2",
		Accounts: []*identity.ProfileShowcaseAccount{
			&identity.ProfileShowcaseAccount{
				ContactID: "contact_2",
				Address:   "0x0000000000000000000000000000000000000001",
				Name:      "Contact2-Account1",
				ColorID:   "blue",
				Emoji:     "-_-",
				Order:     0,
			},
			&identity.ProfileShowcaseAccount{
				ContactID: "contact_2",
				Address:   "0x0000000000000000000000000000000000000002",
				Name:      "Contact2-Account2",
				ColorID:   "blue",
				Emoji:     "-_-",
				Order:     1,
			},
		},
	}
	showcase3 := &identity.ProfileShowcase{
		ContactID: "contact_3",
		Accounts: []*identity.ProfileShowcaseAccount{
			&identity.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")
		}
	}
}

func (s *TestProfileShowcasePersistence) TestUpdateProfileShowcaseAccountOnWalletAccountChange() {
	db, err := openTestDB()
	s.Require().NoError(err)
	persistence := newSQLitePersistence(db)

	deleteAccountAddress := "0x3243344513424"
	updateAccountAddress := "0x3845354643324"

	preferences := &identity.ProfileShowcasePreferences{
		Accounts: []*identity.ProfileShowcaseAccountPreference{
			&identity.ProfileShowcaseAccountPreference{
				Address:            deleteAccountAddress,
				Name:               "Status Account",
				ColorID:            "blue",
				Emoji:              "-_-",
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityEveryone,
				Order:              0,
			},
			&identity.ProfileShowcaseAccountPreference{
				Address:            updateAccountAddress,
				Name:               "Money Box",
				ColorID:            "red",
				Emoji:              ":o)",
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityContacts,
				Order:              1,
			},
		},
	}

	err = persistence.SaveProfileShowcasePreferences(preferences)
	s.Require().NoError(err)

	account, err := persistence.GetProfileShowcaseAccountPreference(updateAccountAddress)
	s.Require().NoError(err)
	s.Require().NotNil(account)
	s.Require().Equal(*account, *preferences.Accounts[1])

	account.Name = "Music Box"
	account.ColorID = "green"
	account.Emoji = ">:-]"
	account.ShowcaseVisibility = identity.ProfileShowcaseVisibilityIDVerifiedContacts
	account.Order = 7

	err = persistence.SaveProfileShowcaseAccountPreference(account)
	s.Require().NoError(err)

	deleted, err := persistence.DeleteProfileShowcaseAccountPreference(deleteAccountAddress)
	s.Require().NoError(err)
	s.Require().True(deleted)

	// One more time to check correct error handling
	deleted, err = persistence.DeleteProfileShowcaseAccountPreference(deleteAccountAddress)
	s.Require().NoError(err)
	s.Require().False(deleted)

	preferencesBack, err := persistence.GetProfileShowcasePreferences()
	s.Require().NoError(err)

	s.Require().Len(preferencesBack.Accounts, 1)
	s.Require().Equal(*preferencesBack.Accounts[0], *account)
}

func (s *TestProfileShowcasePersistence) TestUpdateProfileShowcaseCommunityOnChange() {
	db, err := openTestDB()
	s.Require().NoError(err)
	persistence := newSQLitePersistence(db)

	deleteCommunityID := "0x3243344513424"

	preferences := &identity.ProfileShowcasePreferences{
		Communities: []*identity.ProfileShowcaseCommunityPreference{
			&identity.ProfileShowcaseCommunityPreference{
				CommunityID:        "0x32433445133424",
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityEveryone,
				Order:              0,
			},
			&identity.ProfileShowcaseCommunityPreference{
				CommunityID:        deleteCommunityID,
				ShowcaseVisibility: identity.ProfileShowcaseVisibilityContacts,
				Order:              1,
			},
		},
	}

	err = persistence.SaveProfileShowcasePreferences(preferences)
	s.Require().NoError(err)

	deleted, err := persistence.DeleteProfileShowcaseCommunityPreference(deleteCommunityID)
	s.Require().NoError(err)
	s.Require().True(deleted)

	// One more time to check correct error handling
	deleted, err = persistence.DeleteProfileShowcaseCommunityPreference(deleteCommunityID)
	s.Require().NoError(err)
	s.Require().False(deleted)

	preferencesBack, err := persistence.GetProfileShowcasePreferences()
	s.Require().NoError(err)

	s.Require().Len(preferencesBack.Communities, 1)
	s.Require().Equal(*preferencesBack.Communities[0], *preferences.Communities[0])
}