mirror of
https://github.com/QuilibriumNetwork/ceremonyclient.git
synced 2026-02-21 10:27:26 +08:00
Refactor split command to allow testing of split operations
Closes #338
This commit is contained in:
littleblackcloud
Tyler Sturos
parent
c1c4327c15
commit
90a38817a1
232
client/cmd/split_test.go
Normal file
232
client/cmd/split_test.go
Normal file
@ -0,0 +1,232 @@
|
||||
package cmd_test
|
||||
|
||||
import (
|
||||
"encoding/hex"
|
||||
"math/big"
|
||||
"reflect"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/shopspring/decimal"
|
||||
"source.quilibrium.com/quilibrium/monorepo/client/cmd"
|
||||
)
|
||||
|
||||
func TestSplit(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
args []string
|
||||
totalAmount string
|
||||
amounts [][]byte
|
||||
payload []byte
|
||||
expectError bool
|
||||
}{
|
||||
{
|
||||
name: "Valid split - specified amounts",
|
||||
args: []string{"0x1234", "0.5", "0.25", "0.25"},
|
||||
totalAmount: "1.0",
|
||||
amounts: [][]byte{
|
||||
{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 238, 107, 40, 0},
|
||||
{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 119, 53, 148, 0},
|
||||
{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 119, 53, 148, 0},
|
||||
},
|
||||
payload: []byte{
|
||||
115, 112, 108, 105, 116,
|
||||
18, 52,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 238, 107, 40, 0,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 119, 53, 148, 0,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 119, 53, 148, 0,
|
||||
},
|
||||
expectError: false,
|
||||
},
|
||||
{
|
||||
name: "Invalid split - amounts do not sum to the total amount of the coin",
|
||||
args: []string{"0x1234", "0.5", "0.25"},
|
||||
totalAmount: "1.0",
|
||||
amounts: [][]byte{},
|
||||
payload: []byte{},
|
||||
expectError: true,
|
||||
},
|
||||
{
|
||||
name: "Invalid split - amounts exceed total amount of the coin",
|
||||
args: []string{"0x1234", "0.5", "1"},
|
||||
totalAmount: "1.0",
|
||||
amounts: [][]byte{},
|
||||
payload: []byte{},
|
||||
expectError: true,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range tests {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
payload := []byte("split")
|
||||
coinaddrHex, _ := strings.CutPrefix(tc.args[0], "0x")
|
||||
coinaddr, err := hex.DecodeString(coinaddrHex)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
payload = append(payload, coinaddr...)
|
||||
|
||||
conversionFactor, _ := new(big.Int).SetString("1DCD65000", 16)
|
||||
totalAmount, _ := decimal.NewFromString(tc.totalAmount)
|
||||
totalAmount = totalAmount.Mul(decimal.NewFromBigInt(conversionFactor, 0))
|
||||
|
||||
amounts := [][]byte{}
|
||||
|
||||
if tc.expectError {
|
||||
_, _, err = cmd.Split(tc.args[1:], amounts, payload, totalAmount.BigInt())
|
||||
if err == nil {
|
||||
t.Errorf("want error for invalid split, got nil")
|
||||
}
|
||||
} else {
|
||||
amounts, payload, err = cmd.Split(tc.args[1:], amounts, payload, totalAmount.BigInt())
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if !reflect.DeepEqual(tc.amounts, amounts) {
|
||||
t.Errorf("expected amounts: %v, got: %v", tc.amounts, amounts)
|
||||
}
|
||||
if !reflect.DeepEqual(tc.payload, payload) {
|
||||
t.Errorf("expected payloads: %v, got: %v", tc.payload, payload)
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestSplitParts(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
args []string
|
||||
parts int
|
||||
totalAmount string
|
||||
amounts [][]byte
|
||||
payload []byte
|
||||
}{
|
||||
{
|
||||
name: "Valid split - into parts",
|
||||
args: []string{"0x1234"},
|
||||
parts: 3,
|
||||
totalAmount: "1.0",
|
||||
amounts: [][]byte{
|
||||
{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 158, 242, 26, 170},
|
||||
{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 158, 242, 26, 170},
|
||||
{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 158, 242, 26, 170},
|
||||
{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2},
|
||||
},
|
||||
payload: []byte{
|
||||
115, 112, 108, 105, 116,
|
||||
18, 52,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 158, 242, 26, 170,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 158, 242, 26, 170,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 158, 242, 26, 170,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range tests {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
payload := []byte("split")
|
||||
coinaddrHex, _ := strings.CutPrefix(tc.args[0], "0x")
|
||||
coinaddr, err := hex.DecodeString(coinaddrHex)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
payload = append(payload, coinaddr...)
|
||||
|
||||
conversionFactor, _ := new(big.Int).SetString("1DCD65000", 16)
|
||||
totalAmount, _ := decimal.NewFromString(tc.totalAmount)
|
||||
totalAmount = totalAmount.Mul(decimal.NewFromBigInt(conversionFactor, 0))
|
||||
|
||||
amounts := [][]byte{}
|
||||
|
||||
amounts, payload = cmd.SplitIntoParts(amounts, payload, totalAmount.BigInt(), tc.parts)
|
||||
if !reflect.DeepEqual(tc.amounts, amounts) {
|
||||
t.Errorf("expected amounts: %v, got: %v", tc.amounts, amounts)
|
||||
}
|
||||
if !reflect.DeepEqual(tc.payload, payload) {
|
||||
t.Errorf("expected payloads: %v, got: %v", tc.payload, payload)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestSplitIntoPartsAmount(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
args []string
|
||||
parts int
|
||||
partAmount string
|
||||
totalAmount string
|
||||
amounts [][]byte
|
||||
payload []byte
|
||||
expectError bool
|
||||
}{
|
||||
{
|
||||
name: "Valid split - into parts of specified amount",
|
||||
args: []string{"0x1234"},
|
||||
parts: 2,
|
||||
partAmount: "0.35",
|
||||
totalAmount: "1.0",
|
||||
amounts: [][]byte{
|
||||
{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 166, 228, 156, 0},
|
||||
{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 166, 228, 156, 0},
|
||||
{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 143, 13, 24, 0},
|
||||
},
|
||||
payload: []byte{
|
||||
115, 112, 108, 105, 116,
|
||||
18, 52,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 166, 228, 156, 0,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 166, 228, 156, 0,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 143, 13, 24, 0,
|
||||
},
|
||||
expectError: false,
|
||||
},
|
||||
{
|
||||
name: "Invalid split - amounts exceed total amount of the coin",
|
||||
args: []string{"0x1234"},
|
||||
parts: 3,
|
||||
partAmount: "0.5",
|
||||
totalAmount: "1.0",
|
||||
amounts: [][]byte{},
|
||||
payload: []byte{},
|
||||
expectError: true,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range tests {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
payload := []byte("split")
|
||||
coinaddrHex, _ := strings.CutPrefix(tc.args[0], "0x")
|
||||
coinaddr, err := hex.DecodeString(coinaddrHex)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
payload = append(payload, coinaddr...)
|
||||
|
||||
conversionFactor, _ := new(big.Int).SetString("1DCD65000", 16)
|
||||
totalAmount, _ := decimal.NewFromString(tc.totalAmount)
|
||||
totalAmount = totalAmount.Mul(decimal.NewFromBigInt(conversionFactor, 0))
|
||||
|
||||
amounts := [][]byte{}
|
||||
|
||||
if tc.expectError {
|
||||
_, _, err = cmd.SplitIntoPartsAmount(amounts, payload, totalAmount.BigInt(), tc.parts, tc.partAmount)
|
||||
if err == nil {
|
||||
t.Errorf("want error for invalid split, got nil")
|
||||
}
|
||||
} else {
|
||||
amounts, payload, err = cmd.SplitIntoPartsAmount(amounts, payload, totalAmount.BigInt(), tc.parts, tc.partAmount)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if !reflect.DeepEqual(tc.amounts, amounts) {
|
||||
t.Errorf("expected amounts: %v, got: %v", tc.amounts, amounts)
|
||||
}
|
||||
if !reflect.DeepEqual(tc.payload, payload) {
|
||||
t.Errorf("expected payloads: %v, got: %v", tc.payload, payload)
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user