fix: setTokenExiprationTime bug fix

Signed-off-by: Andrei Kuzmiankov <andrei@launchbadge.com>
hashgraph · Jun 9, 2021 · c85349e · c85349e
1 parent c881918
commit c85349e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 10 deletions.
diff --git a/token_create_transaction.go b/token_create_transaction.go
@@ -248,10 +248,7 @@ func (transaction *TokenCreateTransaction) GetFreezeDefault() bool {
 func (transaction *TokenCreateTransaction) SetExpirationTime(expirationTime time.Time) *TokenCreateTransaction {
 	transaction.requireNotFrozen()
 	transaction.pb.AutoRenewPeriod = nil
-	transaction.pb.Expiry = &proto.Timestamp{
-		Seconds: expirationTime.Unix(),
-		Nanos:   int32(expirationTime.UnixNano()),
-	}
+	transaction.pb.Expiry = timeToProtobuf(expirationTime)
 
 	return transaction
 }
@@ -278,7 +275,7 @@ func (transaction *TokenCreateTransaction) GetAutoRenewAccount() AccountID {
 // The interval at which the auto-renew account will be charged to extend the token's expiry
 func (transaction *TokenCreateTransaction) SetAutoRenewPeriod(autoRenewPeriod time.Duration) *TokenCreateTransaction {
 	transaction.requireNotFrozen()
-	transaction.pb.AutoRenewPeriod = &proto.Duration{Seconds: int64(autoRenewPeriod.Seconds())}
+	transaction.pb.AutoRenewPeriod = durationToProtobuf(autoRenewPeriod)
 	return transaction
 }
 

diff --git a/token_update_transaction.go b/token_update_transaction.go
@@ -184,7 +184,7 @@ func (transaction *TokenUpdateTransaction) GetAutoRenewAccount() AccountID {
 // The new interval at which the auto-renew account will be charged to extend the token's expiry.
 func (transaction *TokenUpdateTransaction) SetAutoRenewPeriod(autoRenewPeriod time.Duration) *TokenUpdateTransaction {
 	transaction.requireNotFrozen()
-	transaction.pb.AutoRenewPeriod = &proto.Duration{Seconds: int64(autoRenewPeriod.Seconds())}
+	transaction.pb.AutoRenewPeriod = durationToProtobuf(autoRenewPeriod)
 	return transaction
 }
 
@@ -201,10 +201,7 @@ func (transaction *TokenUpdateTransaction) GetAutoRenewPeriod() time.Duration {
 // INVALID_EXPIRATION_TIME
 func (transaction *TokenUpdateTransaction) SetExpirationTime(expirationTime time.Time) *TokenUpdateTransaction {
 	transaction.requireNotFrozen()
-	transaction.pb.Expiry = &proto.Timestamp{
-		Seconds: expirationTime.Unix(),
-		Nanos:   int32(expirationTime.UnixNano()),
-	}
+	transaction.pb.Expiry = timeToProtobuf(expirationTime)
 	return transaction
 }