numscript/internal/interpreter/balances_test.go

package interpreter

import (
	"math/big"
	"testing"

	"github.com/gkampitakis/go-snaps/snaps"
	"github.com/stretchr/testify/require"
)

func TestFilterQuery(t *testing.T) {
	fullBalance := Balances{
		"alice": AccountBalance{
			"EUR/2": big.NewInt(1),
			"USD/2": big.NewInt(2),
		},
		"bob": AccountBalance{
			"BTC": big.NewInt(3),
		},
	}

	filteredQuery := fullBalance.filterQuery(BalanceQuery{
		"alice":   []string{"GBP/2", "YEN", "EUR/2"},
		"bob":     []string{"BTC"},
		"charlie": []string{"ETH"},
	})

	require.Equal(t, BalanceQuery{
		"alice":   []string{"GBP/2", "YEN"},
		"charlie": []string{"ETH"},
	}, filteredQuery)
}

func TestCloneBalances(t *testing.T) {
	fullBalance := Balances{
		"alice": AccountBalance{
			"EUR/2": big.NewInt(1),
			"USD/2": big.NewInt(2),
		},
		"bob": AccountBalance{
			"BTC": big.NewInt(3),
		},
	}

	cloned := fullBalance.DeepClone()

	fullBalance["alice"]["USD/2"].Set(big.NewInt(42))

	require.Equal(t, big.NewInt(2), cloned["alice"]["USD/2"])
}

func TestPrettyPrintBalance(t *testing.T) {
	fullBalance := Balances{
		"alice": AccountBalance{
			"EUR/2":    big.NewInt(1),
			"USD/1234": big.NewInt(999999),
		},
		"bob": AccountBalance{
			"BTC": big.NewInt(3),
		},
	}

	snaps.MatchSnapshot(t, fullBalance.PrettyPrint())
}

func TestCmpMaps(t *testing.T) {

	b1 := Balances{
		"alice": AccountBalance{
			"EUR": big.NewInt(100),
		},
	}

	b2 := Balances{
		"alice": AccountBalance{
			"EUR": big.NewInt(42),
		},
	}

	require.Equal(t, false, CompareBalances(b1, b2))
}

func TestCmpMapsIncluding(t *testing.T) {

	t.Run("including (subset)", func(t *testing.T) {
		b2 := Balances{
			"alice": AccountBalance{
				"EUR": big.NewInt(100),
			},
			"bob": AccountBalance{
				"EUR": big.NewInt(100),
			},
		}

		b1 := Balances{
			"alice": AccountBalance{
				"EUR": big.NewInt(100),
			},
		}

		require.Equal(t, true, CompareBalancesIncluding(b1, b2))
	})

	t.Run("different value", func(t *testing.T) {
		b2 := Balances{
			"alice": AccountBalance{
				"EUR": big.NewInt(100),
			},
			"bob": AccountBalance{
				"EUR": big.NewInt(100),
			},
		}

		b1 := Balances{
			"alice": AccountBalance{
				"EUR": big.NewInt(0),
			},
		}

		require.Equal(t, false, CompareBalancesIncluding(b1, b2))
	})

	t.Run("extra value", func(t *testing.T) {
		b2 := Balances{
			"alice": AccountBalance{
				"EUR": big.NewInt(100),
			},
			"bob": AccountBalance{
				"EUR": big.NewInt(100),
			},
		}

		b1 := Balances{
			"alice": AccountBalance{
				"EUR": big.NewInt(100),
			},

			"extra-value": AccountBalance{
				"EUR": big.NewInt(100),
			},
		}

		require.Equal(t, false, CompareBalancesIncluding(b1, b2))
	})
}