BigIntMut() for legacyDec and test

cosmos · Nov 9, 2023 · 0c9ed37 · 0c9ed37
1 parent 19e57dd
commit 0c9ed37
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/math/dec.go b/math/dec.go
@@ -235,6 +235,15 @@ func (d LegacyDec) BigInt() *big.Int {
 	return cp.Set(d.i)
 }
 
+// BigIntMut converts LegacyDec to big.Int, mutative the input
+func (d LegacyDec) BigIntMut() *big.Int {
+	if d.IsNil() {
+		return nil
+	}
+
+	return d.i
+}
+
 func (d LegacyDec) ImmutOp(op func(LegacyDec, LegacyDec) LegacyDec, d2 LegacyDec) LegacyDec {
 	return op(d.Clone(), d2)
 }

diff --git a/math/dec_test.go b/math/dec_test.go
@@ -754,3 +754,23 @@ func TestNegativePrecisionPanic(t *testing.T) {
 		math.LegacyNewDecWithPrec(10, -1)
 	})
 }
+
+func (s *decimalTestSuite) TestConvertToBigIntMutativeForLegacyDec() {
+	r := big.NewInt(30)
+	i := math.LegacyNewDecFromBigInt(r)
+
+	// Compare value of BigInt & BigIntMut
+	s.Require().Equal(i.BigInt(), i.BigIntMut())
+
+	// Modify BigIntMut() pointer and ensure i.BigIntMut() & i.BigInt() change
+	p1 := i.BigIntMut()
+	p1.SetInt64(40)
+	s.Require().Equal(big.NewInt(40), i.BigIntMut())
+	s.Require().Equal(big.NewInt(40), i.BigInt())
+
+	// Modify big.Int() pointer and ensure i.BigIntMut() & i.BigInt() don't change
+	p2 := i.BigInt()
+	p2.SetInt64(50)
+	s.Require().NotEqual(big.NewInt(50), i.BigIntMut())
+	s.Require().NotEqual(big.NewInt(50), i.BigInt())
+}