status-go/protocol/ens/persistence_test.go

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)
}