152 lines
3.8 KiB
Go
152 lines
3.8 KiB
Go
package verification
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/status-im/status-go/protocol/sqlite"
|
|
)
|
|
|
|
func TestPersistenceSuite(t *testing.T) {
|
|
suite.Run(t, new(PersistenceSuite))
|
|
}
|
|
|
|
type PersistenceSuite struct {
|
|
suite.Suite
|
|
|
|
db *Persistence
|
|
}
|
|
|
|
func (s *PersistenceSuite) SetupTest() {
|
|
s.db = nil
|
|
|
|
dbPath, err := ioutil.TempFile("", "")
|
|
s.NoError(err, "creating temp file for db")
|
|
|
|
db, err := sqlite.Open(dbPath.Name(), "", sqlite.ReducedKDFIterationsNumber)
|
|
s.NoError(err, "creating sqlite db instance")
|
|
|
|
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.False(success)
|
|
|
|
trustStatus, err = s.db.GetTrustStatus("0x03")
|
|
s.NoError(err)
|
|
s.Equal(TrustStatusUNKNOWN, trustStatus)
|
|
}
|