add TopLevelDomain.ExtendExpirationDate

x/registry/types/top_level_domain.go

@@ -1,6 +1,8 @@
 package types
 
 import (
+	"time"
+
 	errorsmod "cosmossdk.io/errors"
 	"cosmossdk.io/math"
 	sdk "github.com/cosmos/cosmos-sdk/types"
@@ -50,3 +52,11 @@ func (topLevelDomain TopLevelDomain) IsEditable(sender string) (isEditable bool,
 	isEditable = topLevelDomain.AccessControl[sender] == DomainRole_EDITOR || topLevelDomain.AccessControl[sender] == DomainRole_OWNER
 	return isEditable, err
 }
+
+func (topLevelDomain *TopLevelDomain) ExtendExpirationDate(from time.Time, extensionPeriodInYear uint64) (expirationDateInUnixNano int64) {
+	newExpirationDate := from.AddDate(0, 0, params.OneYearInDays*int(extensionPeriodInYear))
+	expirationDateInUnixNano = newExpirationDate.UnixNano()
+	topLevelDomain.ExpirationDate = expirationDateInUnixNano
+
+	return expirationDateInUnixNano
+}

x/registry/types/top_level_domain_test.go

@@ -2,17 +2,12 @@ package types
 
 import (
 	"testing"
+	"time"
 
 	errorsmod "cosmossdk.io/errors"
-	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/stretchr/testify/require"
 )
 
-type TopLevelDomainTest struct {
-	Domain      TopLevelDomain
-	DomainPrice sdk.Coins
-}
-
 func TestTopLevelDomainValidate(t *testing.T) {
 	testCases := []struct {
 		domain TopLevelDomain
@@ -49,3 +44,31 @@ func TestTopLevelDomainValidate(t *testing.T) {
 		}
 	}
 }
+
+func TestExtendExpirationDate(t *testing.T) {
+	testCases := []struct {
+		from                   time.Time
+		extensionPeriodInYear  uint64
+		expectedExpirationDate time.Time
+	}{
+		{
+			from:                   time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC),
+			extensionPeriodInYear:  1,
+			expectedExpirationDate: time.Date(2020, 12, 31, 0, 0, 0, 0, time.UTC),
+		},
+		{
+			from:                   time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC),
+			extensionPeriodInYear:  2,
+			expectedExpirationDate: time.Date(2021, 12, 31, 0, 0, 0, 0, time.UTC),
+		},
+	}
+
+	for _, tc := range testCases {
+		domain := TopLevelDomain{
+			Name: "myc",
+		}
+		extendExpirationDateInUnixNano := domain.ExtendExpirationDate(tc.from, tc.extensionPeriodInYear)
+		require.Equal(t, tc.expectedExpirationDate.UnixNano(), domain.ExpirationDate)
+		require.Equal(t, tc.expectedExpirationDate.UnixNano(), extendExpirationDateInUnixNano)
+	}
+}