From ef2d43d5b029c565c7f5d9724adee14fc5435b2d Mon Sep 17 00:00:00 2001
From: Christopher Goes <cwgoes@protonmail.ch>
Date: Wed, 4 Apr 2018 19:48:56 +0200
Subject: [PATCH] Add tests for bondedRatio, bondedShareExRate,
 unbondedShareExRate

---
 x/stake/pool_test.go | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/x/stake/pool_test.go b/x/stake/pool_test.go
index 3427c9268..d52e4316e 100644
--- a/x/stake/pool_test.go
+++ b/x/stake/pool_test.go
@@ -11,6 +11,42 @@ import (
 	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
+func TestBondedRatio(t *testing.T) {
+	ctx, _, keeper := createTestInput(t, nil, false, 0)
+	pool := keeper.GetPool(ctx)
+	pool.TotalSupply = 3
+	pool.BondedPool = 2
+	// bonded pool / total supply
+	require.Equal(t, pool.bondedRatio(), sdk.NewRat(2).Quo(sdk.NewRat(3)))
+	pool.TotalSupply = 0
+	// avoids divide-by-zero
+	require.Equal(t, pool.bondedRatio(), sdk.ZeroRat)
+}
+
+func TestBondedShareExRate(t *testing.T) {
+	ctx, _, keeper := createTestInput(t, nil, false, 0)
+	pool := keeper.GetPool(ctx)
+	pool.BondedPool = 3
+	pool.BondedShares = sdk.NewRat(10)
+	// bonded pool / bonded shares
+	require.Equal(t, pool.bondedShareExRate(), sdk.NewRat(3).Quo(sdk.NewRat(10)))
+	pool.BondedShares = sdk.ZeroRat
+	// avoids divide-by-zero
+	require.Equal(t, pool.bondedShareExRate(), sdk.OneRat)
+}
+
+func TestUnbondedShareExRate(t *testing.T) {
+	ctx, _, keeper := createTestInput(t, nil, false, 0)
+	pool := keeper.GetPool(ctx)
+	pool.UnbondedPool = 3
+	pool.UnbondedShares = sdk.NewRat(10)
+	// unbonded pool / unbonded shares
+	require.Equal(t, pool.unbondedShareExRate(), sdk.NewRat(3).Quo(sdk.NewRat(10)))
+	pool.UnbondedShares = sdk.ZeroRat
+	// avoids divide-by-zero
+	require.Equal(t, pool.unbondedShareExRate(), sdk.OneRat)
+}
+
 func TestBondedToUnbondedPool(t *testing.T) {
 	ctx, _, keeper := createTestInput(t, nil, false, 0)