status-go/protocol/verification/persistence_test.go

package verification

import (
	"testing"
	"time"

	"github.com/stretchr/testify/suite"

	"github.com/status-im/status-go/appdatabase"
	"github.com/status-im/status-go/protocol/sqlite"
	"github.com/status-im/status-go/t/helpers"
)

func TestPersistenceSuite(t *testing.T) {
	suite.Run(t, new(PersistenceSuite))
}

type PersistenceSuite struct {
	suite.Suite

	db *Persistence
}

func (s *PersistenceSuite) SetupTest() {
	db, err := helpers.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.False(success)

	trustStatus, err = s.db.GetTrustStatus("0x03")
	s.NoError(err)
	s.Equal(TrustStatusUNKNOWN, trustStatus)
}