mirror of
https://github.com/status-im/status-go.git
synced 2026-08-31 17:11:23 +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
154 lines
3.9 KiB
Go
154 lines
3.9 KiB
Go
package verification
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/status-im/status-go/internal/db/appdatabase"
|
|
"github.com/status-im/status-go/internal/protocol/sqlite"
|
|
"github.com/status-im/status-go/internal/testutils"
|
|
)
|
|
|
|
func TestPersistenceSuite(t *testing.T) {
|
|
suite.Run(t, new(PersistenceSuite))
|
|
}
|
|
|
|
type PersistenceSuite struct {
|
|
suite.Suite
|
|
|
|
db *Persistence
|
|
}
|
|
|
|
func (s *PersistenceSuite) SetupTest() {
|
|
db, err := testutils.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
|
|
s.Require().NoError(err)
|
|
err = sqlite.Migrate(db)
|
|
s.Require().NoError(err)
|
|
|
|
s.db = &Persistence{db: db}
|
|
}
|
|
|
|
func (s *PersistenceSuite) TestVerificationRequests() {
|
|
request := &Request{
|
|
ID: "0xabc",
|
|
From: "0x01",
|
|
To: "0x02",
|
|
Challenge: "ABC",
|
|
Response: "",
|
|
RequestedAt: uint64(time.Unix(1000, 0).Unix()),
|
|
RepliedAt: uint64(time.Unix(0, 0).Unix()),
|
|
RequestStatus: RequestStatusPENDING,
|
|
}
|
|
|
|
// Test Insert
|
|
err := s.db.SaveVerificationRequest(request)
|
|
s.Require().NoError(err)
|
|
|
|
// Test Found
|
|
dbRequest, err := s.db.GetVerificationRequest("0xabc")
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(request, dbRequest)
|
|
|
|
// Test Not Found
|
|
dbRequest2, err := s.db.GetVerificationRequest("0xdef")
|
|
s.Require().NoError(err)
|
|
s.Require().Nil(dbRequest2)
|
|
|
|
// Test Accept
|
|
err = s.db.AcceptContactVerificationRequest("0xabc", "XYZ")
|
|
s.Require().NoError(err)
|
|
|
|
dbRequest, err = s.db.GetVerificationRequest("0xabc")
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(dbRequest)
|
|
s.Require().Equal(RequestStatusACCEPTED, dbRequest.RequestStatus)
|
|
s.Require().Equal("XYZ", dbRequest.Response)
|
|
s.Require().NotEqual(time.Unix(0, 0), dbRequest.RepliedAt)
|
|
|
|
// Test Decline
|
|
request = &Request{
|
|
ID: "0x01",
|
|
From: "0x03",
|
|
To: "0x02",
|
|
Challenge: "ABC",
|
|
Response: "",
|
|
RequestedAt: uint64(time.Unix(1000, 0).Unix()),
|
|
RepliedAt: uint64(time.Unix(0, 0).Unix()),
|
|
RequestStatus: RequestStatusPENDING,
|
|
}
|
|
|
|
err = s.db.SaveVerificationRequest(request)
|
|
s.NoError(err)
|
|
|
|
err = s.db.DeclineContactVerificationRequest("0x01")
|
|
s.NoError(err)
|
|
|
|
dbRequest, err = s.db.GetVerificationRequest("0x01")
|
|
s.NoError(err)
|
|
s.Equal(RequestStatusDECLINED, dbRequest.RequestStatus)
|
|
s.NotEqual(time.Unix(0, 0), dbRequest.RepliedAt)
|
|
|
|
}
|
|
|
|
func (s *PersistenceSuite) TestTrustStatus() {
|
|
|
|
err := s.db.SetTrustStatus("0x01", TrustStatusTRUSTED, 1000)
|
|
s.NoError(err)
|
|
|
|
err = s.db.SetTrustStatus("0x02", TrustStatusUNKNOWN, 1001)
|
|
s.NoError(err)
|
|
|
|
trustStatus, err := s.db.GetTrustStatus("0x01")
|
|
s.NoError(err)
|
|
s.Equal(TrustStatusTRUSTED, trustStatus)
|
|
|
|
err = s.db.SetTrustStatus("0x01", TrustStatusUNTRUSTWORTHY, 1000)
|
|
s.NoError(err)
|
|
|
|
trustStatus, err = s.db.GetTrustStatus("0x01")
|
|
s.NoError(err)
|
|
s.Equal(TrustStatusUNTRUSTWORTHY, trustStatus)
|
|
|
|
trustStatus, err = s.db.GetTrustStatus("0x03")
|
|
s.NoError(err)
|
|
s.Equal(TrustStatusUNKNOWN, trustStatus)
|
|
|
|
statuses, err := s.db.GetAllTrustStatus()
|
|
s.NoError(err)
|
|
|
|
s.Len(statuses, 2)
|
|
s.Equal(TrustStatusUNTRUSTWORTHY, statuses["0x01"])
|
|
s.Equal(TrustStatusUNKNOWN, statuses["0x02"])
|
|
|
|
// Upsert
|
|
success, err := s.db.UpsertTrustStatus("0x03", TrustStatusTRUSTED, 1000)
|
|
s.NoError(err)
|
|
s.True(success)
|
|
|
|
trustStatus, err = s.db.GetTrustStatus("0x03")
|
|
s.NoError(err)
|
|
s.Equal(TrustStatusTRUSTED, trustStatus)
|
|
|
|
success, err = s.db.UpsertTrustStatus("0x03", TrustStatusUNKNOWN, 500) // should not be successful without error, because of being older than latest value in the DB
|
|
s.NoError(err)
|
|
s.False(success)
|
|
|
|
trustStatus, err = s.db.GetTrustStatus("0x03")
|
|
s.NoError(err)
|
|
s.Equal(TrustStatusTRUSTED, trustStatus)
|
|
|
|
success, err = s.db.UpsertTrustStatus("0x03", TrustStatusUNKNOWN, 1500)
|
|
s.NoError(err)
|
|
s.True(success)
|
|
|
|
trustStatus, err = s.db.GetTrustStatus("0x03")
|
|
s.NoError(err)
|
|
s.Equal(TrustStatusUNKNOWN, trustStatus)
|
|
|
|
success, err = s.db.UpsertTrustStatus("0x03", TrustStatusUNKNOWN, 1500) // same trust status, should not be successful
|
|
s.NoError(err)
|
|
s.False(success)
|
|
}
|