package multidevice import ( "database/sql" "os" "testing" "github.com/stretchr/testify/suite" "github.com/status-im/status-go/protocol/sqlite" ) const ( dbPath = "/tmp/status-key-store.db" ) func TestSQLLitePersistenceTestSuite(t *testing.T) { suite.Run(t, new(SQLLitePersistenceTestSuite)) } type SQLLitePersistenceTestSuite struct { suite.Suite // nolint: structcheck, megacheck db *sql.DB service *sqlitePersistence } func (s *SQLLitePersistenceTestSuite) SetupTest() { os.Remove(dbPath) db, err := sqlite.Open(dbPath, "test-key", sqlite.ReducedKDFIterationsNumber) s.Require().NoError(err) s.service = newSQLitePersistence(db) } func (s *SQLLitePersistenceTestSuite) TestAddInstallations() { identity := []byte("alice") installations := []*Installation{ {ID: "alice-1", Version: 1, Enabled: true}, {ID: "alice-2", Version: 2, Enabled: true}, } addedInstallations, err := s.service.AddInstallations( identity, 1, installations, true, ) s.Require().NoError(err) enabledInstallations, err := s.service.GetActiveInstallations(5, identity) s.Require().NoError(err) s.Require().Equal(installations, enabledInstallations) s.Require().Equal(installations, addedInstallations) } func (s *SQLLitePersistenceTestSuite) TestAddInstallationVersions() { identity := []byte("alice") installations := []*Installation{ {ID: "alice-1", Version: 1, Enabled: true}, } _, err := s.service.AddInstallations( identity, 1, installations, true, ) s.Require().NoError(err) enabledInstallations, err := s.service.GetActiveInstallations(5, identity) s.Require().NoError(err) s.Require().Equal(installations, enabledInstallations) installationsWithDowngradedVersion := []*Installation{ {ID: "alice-1", Version: 0}, } _, err = s.service.AddInstallations( identity, 3, installationsWithDowngradedVersion, true, ) s.Require().NoError(err) enabledInstallations, err = s.service.GetActiveInstallations(5, identity) s.Require().NoError(err) s.Require().Equal(installations, enabledInstallations) } func (s *SQLLitePersistenceTestSuite) TestAddInstallationsLimit() { identity := []byte("alice") installations := []*Installation{ {ID: "alice-1", Version: 1}, {ID: "alice-2", Version: 2}, } _, err := s.service.AddInstallations( identity, 1, installations, true, ) s.Require().NoError(err) installations = []*Installation{ {ID: "alice-1", Version: 1}, {ID: "alice-3", Version: 3}, } _, err = s.service.AddInstallations( identity, 2, installations, true, ) s.Require().NoError(err) installations = []*Installation{ {ID: "alice-2", Version: 2, Enabled: true}, {ID: "alice-3", Version: 3, Enabled: true}, {ID: "alice-4", Version: 4, Enabled: true}, } _, err = s.service.AddInstallations( identity, 3, installations, true, ) s.Require().NoError(err) enabledInstallations, err := s.service.GetActiveInstallations(3, identity) s.Require().NoError(err) s.Require().Equal(installations, enabledInstallations) } func (s *SQLLitePersistenceTestSuite) TestAddInstallationsDisabled() { identity := []byte("alice") installations := []*Installation{ {ID: "alice-1", Version: 1}, {ID: "alice-2", Version: 2}, } _, err := s.service.AddInstallations( identity, 1, installations, false, ) s.Require().NoError(err) actualInstallations, err := s.service.GetActiveInstallations(3, identity) s.Require().NoError(err) s.Require().Nil(actualInstallations) } func (s *SQLLitePersistenceTestSuite) TestDisableInstallation() { identity := []byte("alice") installations := []*Installation{ {ID: "alice-1", Version: 1}, {ID: "alice-2", Version: 2}, } _, err := s.service.AddInstallations( identity, 1, installations, true, ) s.Require().NoError(err) err = s.service.DisableInstallation(identity, "alice-1") s.Require().NoError(err) // We add the installations again installations = []*Installation{ {ID: "alice-1", Version: 1}, {ID: "alice-2", Version: 2}, } addedInstallations, err := s.service.AddInstallations( identity, 1, installations, true, ) s.Require().NoError(err) s.Require().Equal(0, len(addedInstallations)) actualInstallations, err := s.service.GetActiveInstallations(3, identity) s.Require().NoError(err) expected := []*Installation{{ID: "alice-2", Version: 2, Enabled: true}} s.Require().Equal(expected, actualInstallations) } func (s *SQLLitePersistenceTestSuite) TestEnableInstallation() { identity := []byte("alice") installations := []*Installation{ {ID: "alice-1", Version: 1}, {ID: "alice-2", Version: 2}, } _, err := s.service.AddInstallations( identity, 1, installations, true, ) s.Require().NoError(err) err = s.service.DisableInstallation(identity, "alice-1") s.Require().NoError(err) actualInstallations, err := s.service.GetActiveInstallations(3, identity) s.Require().NoError(err) expected := []*Installation{{ID: "alice-2", Version: 2, Enabled: true}} s.Require().Equal(expected, actualInstallations) err = s.service.EnableInstallation(identity, "alice-1") s.Require().NoError(err) actualInstallations, err = s.service.GetActiveInstallations(3, identity) s.Require().NoError(err) expected = []*Installation{ {ID: "alice-1", Version: 1, Enabled: true}, {ID: "alice-2", Version: 2, Enabled: true}, } s.Require().Equal(expected, actualInstallations) } func (s *SQLLitePersistenceTestSuite) TestGetInstallations() { identity := []byte("alice") installations := []*Installation{ {ID: "alice-1", Version: 1}, {ID: "alice-2", Version: 2}, } _, err := s.service.AddInstallations( identity, 1, installations, true, ) s.Require().NoError(err) err = s.service.DisableInstallation(identity, "alice-1") s.Require().NoError(err) actualInstallations, err := s.service.GetInstallations(identity) s.Require().NoError(err) emptyMetadata := &InstallationMetadata{} expected := []*Installation{ {ID: "alice-1", Version: 1, Timestamp: 1, Enabled: false, InstallationMetadata: emptyMetadata}, {ID: "alice-2", Version: 2, Timestamp: 1, Enabled: true, InstallationMetadata: emptyMetadata}, } s.Require().Equal(2, len(actualInstallations)) s.Require().ElementsMatch(expected, actualInstallations) } func (s *SQLLitePersistenceTestSuite) TestSetMetadata() { identity := []byte("alice") installations := []*Installation{ {ID: "alice-1", Version: 1}, {ID: "alice-2", Version: 2}, } _, err := s.service.AddInstallations( identity, 1, installations, true, ) s.Require().NoError(err) err = s.service.DisableInstallation(identity, "alice-1") s.Require().NoError(err) emptyMetadata := &InstallationMetadata{} setMetadata := &InstallationMetadata{ Name: "a", FCMToken: "b", DeviceType: "c", } err = s.service.SetInstallationMetadata(identity, "alice-2", setMetadata) s.Require().NoError(err) actualInstallations, err := s.service.GetInstallations(identity) s.Require().NoError(err) expected := []*Installation{ {ID: "alice-1", Version: 1, Timestamp: 1, Enabled: false, InstallationMetadata: emptyMetadata}, {ID: "alice-2", Version: 2, Timestamp: 1, Enabled: true, InstallationMetadata: setMetadata}, } s.Require().ElementsMatch(expected, actualInstallations) }