planetmint-go/x/dao/keeper/distribution_test.go
Jürgen Eckel bfbe9584a1
296 change distribution (#306)
* updated planetmint-go dependency in lib/tx
* added two additional distribution addresses. splitted up the previous investor pool of 31% into three pools:
* early investors 19%
* investors 10%
* strategic 2%

* adjusted distribution request message
* adjusted distribution result
* adjusted distribution order
* refactored reissuance and distribution msg methods to be more readable

* fixed token distribution test case

Signed-off-by: Jürgen Eckel <juergen@riddleandcode.com>
2024-01-29 14:33:40 +01:00

93 lines
3.3 KiB
Go

package keeper_test
import (
"fmt"
"strconv"
"testing"
keepertest "github.com/planetmint/planetmint-go/testutil/keeper"
"github.com/stretchr/testify/assert"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/planetmint/planetmint-go/x/dao/keeper"
"github.com/planetmint/planetmint-go/x/dao/types"
)
func createNDistributionOrder(keeper *keeper.Keeper, ctx sdk.Context, n int) []types.DistributionOrder {
items := make([]types.DistributionOrder, n)
for i := range items {
amount := 10000
items[i].FirstPop = int64(i)*1000 + 1
items[i].LastPop = int64(i+1) * 1000
items[i].DaoAddr = fmt.Sprintf("DAO%v", i)
items[i].DaoAmount = fmt.Sprintf("%v", float64(amount)*types.PercentageDao)
items[i].InvestorAddr = fmt.Sprintf("INVESTOR%v", i)
items[i].InvestorAmount = strconv.FormatInt(int64(float64(amount)*types.PercentageInvestor), 10)
items[i].PopAddr = fmt.Sprintf("POP%v", i)
items[i].PopAmount = fmt.Sprintf("%v", float64(amount)*types.PercentagePop)
keeper.StoreDistributionOrder(ctx, items[i])
}
return items
}
func TestDistributionOrder(t *testing.T) {
t.Parallel()
keeper, ctx := keepertest.DaoKeeper(t)
items := createNDistributionOrder(keeper, ctx, 10)
for _, item := range items {
challenge, found := keeper.LookupDistributionOrder(ctx, item.LastPop)
assert.True(t, found)
assert.Equal(t, item, challenge)
}
lastDistribution, found := keeper.GetLastDistributionOrder(ctx)
assert.True(t, found)
assert.Equal(t, items[9], lastDistribution)
}
func TestTokenDistribution(t *testing.T) {
t.Parallel()
k, ctx := keepertest.DaoKeeper(t)
reissuanceValue := 998.85844748
reissuances := 1000
Amount1stBatch := 781.0
Amount2ndBatch := 219.0
_ = createNReissuances(k, ctx, reissuances)
distribution, err := k.GetDistributionForReissuedTokens(ctx, 780)
assert.Nil(t, err)
amount1, err1 := strconv.ParseFloat(distribution.DaoAmount, 64)
amount2, err2 := strconv.ParseFloat(distribution.InvestorAmount, 64)
amount3, err3 := strconv.ParseFloat(distribution.PopAmount, 64)
amount4, err4 := strconv.ParseFloat(distribution.EarlyInvAmount, 64)
amount5, err5 := strconv.ParseFloat(distribution.StrategicAmount, 64)
assert.Nil(t, err1)
assert.Nil(t, err2)
assert.Nil(t, err3)
assert.Nil(t, err4)
assert.Nil(t, err5)
sum := amount1 + amount2 + amount3 + amount4 + amount5
expSum := reissuanceValue * Amount1stBatch // add the [0] of the
assert.True(t, expSum-sum < 0.000001)
var lastDistribution types.DistributionOrder
lastDistribution.LastPop = 780
k.StoreDistributionOrder(ctx, lastDistribution)
lastDistribution, err0 := k.GetDistributionForReissuedTokens(ctx, 999)
assert.Nil(t, err0)
amount1, err1 = strconv.ParseFloat(lastDistribution.DaoAmount, 64)
amount2, err2 = strconv.ParseFloat(lastDistribution.InvestorAmount, 64)
amount3, err3 = strconv.ParseFloat(lastDistribution.PopAmount, 64)
amount4, err4 = strconv.ParseFloat(lastDistribution.EarlyInvAmount, 64)
amount5, err5 = strconv.ParseFloat(lastDistribution.StrategicAmount, 64)
assert.Nil(t, err1)
assert.Nil(t, err2)
assert.Nil(t, err3)
assert.Nil(t, err4)
assert.Nil(t, err5)
sum = amount1 + amount2 + amount3 + amount4 + amount5
expSum = reissuanceValue * Amount2ndBatch // add the [0] of the
assert.True(t, expSum-sum < 0.000001)
assert.Equal(t, float64(reissuances), Amount1stBatch+Amount2ndBatch)
}