79 lines
2.7 KiB
Go
79 lines
2.7 KiB
Go
package pgutils
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_NumericToFloat(t *testing.T) {
|
|
tests := []struct {
|
|
given pgtype.Numeric
|
|
expected float64
|
|
}{
|
|
{given: pgtype.Numeric{Int: big.NewInt(0), Exp: 0, Valid: true}, expected: 0},
|
|
{given: pgtype.Numeric{Int: big.NewInt(5), Exp: 0, Valid: true}, expected: 5},
|
|
{given: pgtype.Numeric{Int: big.NewInt(10), Exp: 0, Valid: true}, expected: 10},
|
|
{given: pgtype.Numeric{Int: big.NewInt(1000), Exp: -2, Valid: true}, expected: 10.00},
|
|
{given: pgtype.Numeric{Int: big.NewInt(1000), Exp: -3, Valid: true}, expected: 1.000},
|
|
{given: pgtype.Numeric{Int: big.NewInt(1000), Exp: -4, Valid: true}, expected: 0.1000},
|
|
{given: pgtype.Numeric{Int: big.NewInt(2555), Exp: -2, Valid: true}, expected: 25.55},
|
|
{given: pgtype.Numeric{Int: big.NewInt(-1), Exp: -2, Valid: true}, expected: -0.01},
|
|
}
|
|
for _, test := range tests {
|
|
t.Run(fmt.Sprintf("given %v, expected %v", test.given, test.expected), func(t *testing.T) {
|
|
actual := NumericToFloat64(test.given)
|
|
assert.Equal(t, test.expected, actual)
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
func Test_FloatToNumeric(t *testing.T) {
|
|
tests := []struct {
|
|
given float64
|
|
expected pgtype.Numeric
|
|
precision int
|
|
}{
|
|
{given: 0.0, expected: pgtype.Numeric{Int: big.NewInt(0), Exp: 0, Valid: true}, precision: 0},
|
|
{given: 25.50, expected: pgtype.Numeric{Int: big.NewInt(2550), Exp: -2, Valid: true}, precision: 2},
|
|
{given: 100.0, expected: pgtype.Numeric{Int: big.NewInt(1), Exp: 2, Valid: true}, precision: 0},
|
|
{given: 0.0001, expected: pgtype.Numeric{Int: big.NewInt(0), Exp: -2, Valid: true}, precision: 2},
|
|
{given: 0.0001, expected: pgtype.Numeric{Int: big.NewInt(1), Exp: -4, Valid: true}, precision: 4},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(fmt.Sprintf("given %v, expected %v", test.given, test.expected), func(t *testing.T) {
|
|
actual := FloatToNumeric(test.given, test.precision)
|
|
assert.Equal(t, test.expected, actual)
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_AddNumeric(t *testing.T) {
|
|
valueA := pgtype.Numeric{Int: big.NewInt(1), Exp: -3, Valid: true}
|
|
valueB := pgtype.Numeric{Int: big.NewInt(2), Exp: -2, Valid: true}
|
|
|
|
slog.Info("valueA", "valueA", valueA)
|
|
slog.Info("valueB", "valueB", valueB)
|
|
|
|
actual := AddNumeric(valueA, valueB)
|
|
|
|
slog.Info("actual", "actual", actual)
|
|
assert.Equal(t, pgtype.Numeric{Int: big.NewInt(21), Exp: -3, Valid: true}, actual)
|
|
}
|
|
|
|
func Test_SubtractNumeric(t *testing.T) {
|
|
valueA := pgtype.Numeric{Int: big.NewInt(1), Exp: -3, Valid: true}
|
|
valueB := pgtype.Numeric{Int: big.NewInt(2), Exp: -2, Valid: true}
|
|
|
|
actual := SubtractNumeric(valueA, valueB)
|
|
|
|
slog.Info("actual", "actual", actual)
|
|
assert.Equal(t, pgtype.Numeric{Int: big.NewInt(-19), Exp: -3, Valid: true}, actual)
|
|
}
|