From 99cd8c80fbe42381abf747277532c66a2c819bb5 Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Sun, 6 Jun 2021 20:34:45 +0800
Subject: [PATCH] Add tests for big.Int usage (#1917)

Reviewed-on: https://gitea.com/xorm/xorm/pulls/1917
Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com>
Co-committed-by: Lunny Xiao <xiaolunwen@gmail.com>
---
 integrations/types_test.go | 48 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/integrations/types_test.go b/integrations/types_test.go
index d0357d6b..4fc123ec 100644
--- a/integrations/types_test.go
+++ b/integrations/types_test.go
@@ -7,6 +7,8 @@ package integrations
 import (
 	"errors"
 	"fmt"
+	"math/big"
+	"strconv"
 	"testing"
 
 	"xorm.io/xorm"
@@ -402,3 +404,49 @@ func TestUnsigned(t *testing.T) {
 		assert.False(t, true, "Unsigned is not implemented")
 	}
 }
+
+type MyDecimal big.Int
+
+func (d *MyDecimal) FromDB(data []byte) error {
+	i, _ := strconv.ParseInt(string(data), 10, 64)
+	if d == nil {
+		d = (*MyDecimal)(big.NewInt(i))
+	} else {
+		(*big.Int)(d).SetInt64(i)
+	}
+	return nil
+}
+
+func (d *MyDecimal) ToDB() ([]byte, error) {
+	return []byte(fmt.Sprintf("%d", (*big.Int)(d).Int64())), nil
+}
+
+func (d *MyDecimal) AsBigInt() *big.Int {
+	return (*big.Int)(d)
+}
+
+func (d *MyDecimal) AsInt64() int64 {
+	return d.AsBigInt().Int64()
+}
+
+func TestDecimal(t *testing.T) {
+	type MyMoney struct {
+		Id      int64
+		Account *MyDecimal
+	}
+
+	assert.NoError(t, PrepareEngine())
+	assertSync(t, new(MyMoney))
+
+	_, err := testEngine.Insert(&MyMoney{
+		Account: (*MyDecimal)(big.NewInt(10000000000000000)),
+	})
+	assert.NoError(t, err)
+
+	var m MyMoney
+	has, err := testEngine.Get(&m)
+	assert.NoError(t, err)
+	assert.True(t, has)
+	assert.NotNil(t, m.Account)
+	assert.EqualValues(t, 10000000000000000, m.Account.AsInt64())
+}