status-go/protocol/ens/persistence_test.go

90 lines
2.5 KiB
Go
Raw Normal View History

2021-01-11 10:32:51 +00:00
package ens
import (
"testing"
"github.com/stretchr/testify/require"
"github.com/status-im/status-go/appdatabase"
2021-01-11 10:32:51 +00:00
"github.com/status-im/status-go/protocol/sqlite"
"github.com/status-im/status-go/t/helpers"
2021-01-11 10:32:51 +00:00
)
func TestGetENSToBeVerified(t *testing.T) {
pk := "1"
name := "test.eth"
updatedName := "test2.eth"
db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
2021-01-11 10:32:51 +00:00
require.NoError(t, err)
err = sqlite.Migrate(db)
require.NoError(t, err)
persistence := NewPersistence(db)
require.NotNil(t, persistence)
record := VerificationRecord{Name: name, PublicKey: pk, Clock: 2}
// We add a record, it should be nil
response, err := persistence.AddRecord(record)
require.NoError(t, err)
require.Nil(t, response)
// We add a record again, it should return the same record
response, err = persistence.AddRecord(record)
require.NoError(t, err)
require.NotNil(t, response)
require.False(t, response.Verified)
require.Equal(t, record.Name, response.Name)
require.Equal(t, record.PublicKey, response.PublicKey)
// We add a record again, with a different clock value
record.Clock++
response, err = persistence.AddRecord(record)
require.NoError(t, err)
require.NotNil(t, response)
require.False(t, response.Verified)
require.Equal(t, record.Name, response.Name)
require.Equal(t, record.PublicKey, response.PublicKey)
// We add a record again, with a different name, but lower clock value
record.Clock--
record.Name = updatedName
response, err = persistence.AddRecord(record)
require.NoError(t, err)
require.NotNil(t, response)
require.False(t, response.Verified)
require.Equal(t, name, response.Name)
require.Equal(t, record.PublicKey, response.PublicKey)
// We add a record again, with a different name and higher clock value
record.Clock += 2
record.Name = updatedName
response, err = persistence.AddRecord(record)
require.NoError(t, err)
require.Nil(t, response)
// update the record
record.Verified = false
record.VerificationRetries = 10
record.NextRetry = 20
record.VerifiedAt = 30
err = persistence.UpdateRecords([]*VerificationRecord{&record})
require.NoError(t, err)
toBeVerified, err := persistence.GetENSToBeVerified(20)
require.NoError(t, err)
require.Len(t, toBeVerified, 1)
require.False(t, toBeVerified[0].Verified)
require.Equal(t, uint64(10), toBeVerified[0].VerificationRetries)
require.Equal(t, uint64(20), toBeVerified[0].NextRetry)
require.Equal(t, uint64(30), toBeVerified[0].VerifiedAt)
require.Equal(t, updatedName, toBeVerified[0].Name)
require.Equal(t, pk, toBeVerified[0].PublicKey)
}