package ens import ( "testing" "github.com/stretchr/testify/require" "github.com/status-im/status-go/appdatabase" "github.com/status-im/status-go/protocol/sqlite" "github.com/status-im/status-go/t/helpers" ) func TestGetENSToBeVerified(t *testing.T) { pk := "1" name := "test.eth" updatedName := "test2.eth" db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{}) 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) }