mirror of
https://github.com/status-im/status-go.git
synced 2026-08-31 09:01:16 +00:00
Igor Sirotin
f3e4363808
Part of the Go project layout migration, item 27. Pure move plus import-path rewrite across 502 files. No API or behaviour change. `internal/` keeps the messaging application logic unimportable from outside the module, which is what the issue asks for -- status-go is consumed through the C-bindings in mobile/, not as a Go library. Things that had to follow the move, beyond the Go imports: - tools/generate-handlers/template.txt. messenger_handlers.go is generated, and the template hard-codes the imports it emits, so the generated file kept importing protocol/common and failed typecheck. - .gitignore. The ignore rule for that generated file was pinned to the old path; without moving it, a 1486-line generated file starts being tracked. - Makefile: the logosstorage and torrent test targets (both the archive packages and ./protocol itself), the archive README, migration-protocol. - scripts/run_unit_tests.sh, which names the protocol package explicitly to shard its tests. - scripts/cleanup_generated_files.sh and .golangci.yml. scripts/migration_check.sh also needed a fix that is not specific to this move: it validated every file the branch touched under a migration dir against the timestamp naming rule, and a directory rename makes every migration in it look newly added. It now excludes renames, so moving a migration is not mistaken for adding one. refs #7067
97 lines
2.7 KiB
Go
97 lines
2.7 KiB
Go
package pushnotificationserver
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/status-im/status-go/internal/crypto"
|
|
"github.com/status-im/status-go/internal/db/appdatabase"
|
|
"github.com/status-im/status-go/internal/protocol/common"
|
|
"github.com/status-im/status-go/internal/protocol/protobuf"
|
|
"github.com/status-im/status-go/internal/protocol/sqlite"
|
|
"github.com/status-im/status-go/internal/testutils"
|
|
)
|
|
|
|
func TestSQLitePersistenceSuite(t *testing.T) {
|
|
suite.Run(t, new(SQLitePersistenceSuite))
|
|
}
|
|
|
|
type SQLitePersistenceSuite struct {
|
|
suite.Suite
|
|
persistence Persistence
|
|
}
|
|
|
|
func (s *SQLitePersistenceSuite) SetupTest() {
|
|
db, err := testutils.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
|
|
s.Require().NoError(err)
|
|
err = sqlite.Migrate(db)
|
|
s.Require().NoError(err)
|
|
|
|
s.persistence = NewSQLitePersistence(db)
|
|
}
|
|
|
|
func (s *SQLitePersistenceSuite) TestSaveAndRetrieve() {
|
|
key, err := crypto.GenerateKey()
|
|
s.Require().NoError(err)
|
|
installationID := "54242d02-bb92-11ea-b3de-0242ac130004"
|
|
|
|
registration := &protobuf.PushNotificationRegistration{
|
|
InstallationId: installationID,
|
|
Version: 5,
|
|
}
|
|
|
|
s.Require().NoError(s.persistence.SavePushNotificationRegistration(common.HashPublicKey(&key.PublicKey), registration))
|
|
|
|
retrievedRegistration, err := s.persistence.GetPushNotificationRegistrationByPublicKeyAndInstallationID(common.HashPublicKey(&key.PublicKey), installationID)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().True(proto.Equal(registration, retrievedRegistration))
|
|
}
|
|
|
|
func (s *SQLitePersistenceSuite) TestSaveAndRetrieveIdentity() {
|
|
retrievedKey, err := s.persistence.GetIdentity()
|
|
s.Require().NoError(err)
|
|
s.Require().Nil(retrievedKey)
|
|
|
|
key, err := crypto.GenerateKey()
|
|
s.Require().NoError(err)
|
|
s.Require().NoError(s.persistence.SaveIdentity(key))
|
|
|
|
retrievedKey, err = s.persistence.GetIdentity()
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Equal(key, retrievedKey)
|
|
}
|
|
|
|
func (s *SQLitePersistenceSuite) TestSaveDifferentIdenities() {
|
|
key1, err := crypto.GenerateKey()
|
|
s.Require().NoError(err)
|
|
key2, err := crypto.GenerateKey()
|
|
s.Require().NoError(err)
|
|
|
|
// First one should be successul, second should fail
|
|
s.Require().NoError(s.persistence.SaveIdentity(key1))
|
|
s.Require().Error(s.persistence.SaveIdentity(key2))
|
|
}
|
|
|
|
func (s *SQLitePersistenceSuite) TestExists() {
|
|
messageID1 := []byte("1")
|
|
messageID2 := []byte("2")
|
|
|
|
result, err := s.persistence.PushNotificationExists(messageID1)
|
|
s.Require().NoError(err)
|
|
s.Require().False(result)
|
|
|
|
result, err = s.persistence.PushNotificationExists(messageID1)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require()
|
|
s.Require().True(result)
|
|
|
|
result, err = s.persistence.PushNotificationExists(messageID2)
|
|
s.Require().NoError(err)
|
|
s.Require().False(result)
|
|
}
|