Skip to content

Commit

Permalink
fix: unit tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@shreyasbhat0
shreyasbhat0 committed Feb 13, 2025
1 parent 194c545 commit d9b16f6
Showing 1 changed file with 19 additions and 19 deletions.
38 changes: 19 additions & 19 deletions x/arkeo/keeper/msg_server_claim_contract_income_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,8 +54,8 @@ func TestValidate(t *testing.T) {
contract.Deposit = cosmos.NewInt(contract.Duration * contract.Rate.Amount.Int64())
contract.Id = 1
require.NoError(t, k.SetContract(ctx, contract))
require.NoError(t, k.MintToModule(ctx, types.ModuleName, getCoin(common.Tokens(10000*100*2))))
require.NoError(t, k.SendFromModuleToModule(ctx, types.ModuleName, types.ContractName, getCoins(1000*100)))
require.NoError(t, k.MintToModule(ctx, types.ReserveName, getCoin(common.Tokens(10000*100*2))))
require.NoError(t, k.SendFromModuleToModule(ctx, types.ReserveName, types.ContractName, getCoins(1000*100)))

// happy path

Expand Down Expand Up @@ -106,9 +106,9 @@ func TestHandlePayAsYouGo(t *testing.T) {
require.NoError(t, err)
client, err := common.NewPubKeyFromCrypto(pk)
require.NoError(t, err)
require.NoError(t, k.MintToModule(ctx, types.ModuleName, getCoin(common.Tokens(10000))))
require.NoError(t, k.SendFromModuleToModule(ctx, types.ModuleName, types.ContractName, getCoins(1000)))
initalModuleBalance := k.GetBalanceOfModule(ctx, types.ModuleName, configs.Denom).Int64()
require.NoError(t, k.MintToModule(ctx, types.ReserveName, getCoin(common.Tokens(10000))))
require.NoError(t, k.SendFromModuleToModule(ctx, types.ReserveName, types.ContractName, getCoins(1000)))
initalModuleBalance := k.GetBalanceOfModule(ctx, types.ReserveName, configs.Denom).Int64()
rate, err := cosmos.ParseCoin("10uarkeo")
require.NoError(t, err)

Expand All @@ -135,7 +135,7 @@ func TestHandlePayAsYouGo(t *testing.T) {

require.Equal(t, k.GetBalance(ctx, acc).AmountOf(configs.Denom).Int64(), int64(180))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ContractName, configs.Denom).Int64(), int64(800))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ModuleName, configs.Denom).Int64(), int64(999999999020))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ReserveName, configs.Denom).Int64(), int64(999999999020))

msg = types.MsgClaimContractIncome{
ContractId: contract.Id,
Expand All @@ -151,7 +151,7 @@ func TestHandlePayAsYouGo(t *testing.T) {
require.NoError(t, s.HandlerClaimContractIncome(ctx, &msg))
require.Equal(t, k.GetBalance(ctx, acc).AmountOf(configs.Denom).Int64(), int64(189))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ContractName, configs.Denom).Int64(), int64(790))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ModuleName, configs.Denom).Int64(), int64(999999999021))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ReserveName, configs.Denom).Int64(), int64(999999999021))

// increase the nonce and get slightly more funds for the provider
msg.Nonce = 25
Expand All @@ -166,7 +166,7 @@ func TestHandlePayAsYouGo(t *testing.T) {
require.Equal(t, acct, int64(225))
cname := k.GetBalanceOfModule(ctx, types.ContractName, configs.Denom).Int64()
require.Equal(t, cname, int64(750))
rname := k.GetBalanceOfModule(ctx, types.ModuleName, configs.Denom).Int64()
rname := k.GetBalanceOfModule(ctx, types.ReserveName, configs.Denom).Int64()
require.Equal(t, rname, int64(999999999025))
require.Equal(t, (rname+cname+acct)-initalModuleBalance, contract.Rate.Amount.Int64()*contract.Duration)

Expand All @@ -183,7 +183,7 @@ func TestHandlePayAsYouGo(t *testing.T) {
require.Equal(t, acct, int64(900))
cname = k.GetBalanceOfModule(ctx, types.ContractName, configs.Denom).Int64()
require.Equal(t, cname, int64(0))
rname = k.GetBalanceOfModule(ctx, types.ModuleName, configs.Denom).Int64()
rname = k.GetBalanceOfModule(ctx, types.ReserveName, configs.Denom).Int64()
require.Equal(t, rname, int64(999999999100))
require.Equal(t, (rname+cname+acct)-initalModuleBalance, contract.Rate.Amount.Int64()*contract.Duration)
}
Expand All @@ -200,9 +200,9 @@ func TestHandleSubscription(t *testing.T) {
require.NoError(t, err)
service := common.BTCService
client := types.GetRandomPubKey()
require.NoError(t, k.MintToModule(ctx, types.ModuleName, getCoin(common.Tokens(10000))))
require.NoError(t, k.SendFromModuleToModule(ctx, types.ModuleName, types.ContractName, getCoins(1000)))
initalModuleBalance := k.GetBalanceOfModule(ctx, types.ModuleName, configs.Denom).Int64()
require.NoError(t, k.MintToModule(ctx, types.ReserveName, getCoin(common.Tokens(10000))))
require.NoError(t, k.SendFromModuleToModule(ctx, types.ReserveName, types.ContractName, getCoins(1000)))
initalModuleBalance := k.GetBalanceOfModule(ctx, types.ReserveName, configs.Denom).Int64()
rate, err := cosmos.ParseCoin("10uarkeo")
require.NoError(t, err)

Expand All @@ -226,7 +226,7 @@ func TestHandleSubscription(t *testing.T) {

require.Equal(t, k.GetBalance(ctx, acc).AmountOf(configs.Denom).Int64(), int64(90))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ContractName, configs.Denom).Int64(), int64(900))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ModuleName, configs.Denom).Int64(), int64(999999999010))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ReserveName, configs.Denom).Int64(), int64(999999999010))

msg = types.MsgClaimContractIncome{
ContractId: contract.Id,
Expand All @@ -237,7 +237,7 @@ func TestHandleSubscription(t *testing.T) {
require.NoError(t, s.HandlerClaimContractIncome(ctx, &msg))
require.Equal(t, k.GetBalance(ctx, acc).AmountOf(configs.Denom).Int64(), int64(90))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ContractName, configs.Denom).Int64(), int64(900))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ModuleName, configs.Denom).Int64(), int64(999999999010))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ReserveName, configs.Denom).Int64(), int64(999999999010))

// increase the nonce and get slightly more funds for the provider
ctx = ctx.WithBlockHeight(30)
Expand All @@ -248,7 +248,7 @@ func TestHandleSubscription(t *testing.T) {
require.Equal(t, acct, int64(180))
cname := k.GetBalanceOfModule(ctx, types.ContractName, configs.Denom).Int64()
require.Equal(t, cname, int64(800))
rname := k.GetBalanceOfModule(ctx, types.ModuleName, configs.Denom).Int64()
rname := k.GetBalanceOfModule(ctx, types.ReserveName, configs.Denom).Int64()
require.Equal(t, rname, int64(999999999020))
require.Equal(t, (rname+cname+acct)-initalModuleBalance, contract.Rate.Amount.Int64()*contract.Duration)

Expand All @@ -260,7 +260,7 @@ func TestHandleSubscription(t *testing.T) {
require.Equal(t, acct, int64(900))
cname = k.GetBalanceOfModule(ctx, types.ContractName, configs.Denom).Int64()
require.Equal(t, cname, int64(0))
rname = k.GetBalanceOfModule(ctx, types.ModuleName, configs.Denom).Int64()
rname = k.GetBalanceOfModule(ctx, types.ReserveName, configs.Denom).Int64()
require.Equal(t, rname, int64(999999999100))
require.Equal(t, (rname+cname+acct)-initalModuleBalance, contract.Rate.Amount.Int64()*contract.Duration)
}
Expand Down Expand Up @@ -290,8 +290,8 @@ func TestClaimContractIncomeHandler(t *testing.T) {
require.NoError(t, err)
client, err := common.NewPubKeyFromCrypto(pk)
require.NoError(t, err)
require.NoError(t, k.MintToModule(ctx, types.ModuleName, getCoin(common.Tokens(10000))))
require.NoError(t, k.SendFromModuleToModule(ctx, types.ModuleName, types.ContractName, getCoins(1000)))
require.NoError(t, k.MintToModule(ctx, types.ReserveName, getCoin(common.Tokens(10000))))
require.NoError(t, k.SendFromModuleToModule(ctx, types.ReserveName, types.ContractName, getCoins(1000)))

rate, err := cosmos.ParseCoin("10uarkeo")
require.NoError(t, err)
Expand Down Expand Up @@ -319,7 +319,7 @@ func TestClaimContractIncomeHandler(t *testing.T) {

require.Equal(t, k.GetBalance(ctx, acc).AmountOf(configs.Denom).Int64(), int64(180))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ContractName, configs.Denom).Int64(), int64(800))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ModuleName, configs.Denom).Int64(), int64(999999999020))
require.Equal(t, k.GetBalanceOfModule(ctx, types.ReserveName, configs.Denom).Int64(), int64(999999999020))

// bad nonce
msg.Nonce = 0
Expand Down

0 comments on commit d9b16f6

Please sign in to comment.