Mirroristas/planetmint-go
2
0
Fork 0
mirror of https://github.com/planetmint/planetmint-go.git synced 2025-03-30 15:08:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: add prosume e2e test suite

Browse Source 
Signed-off-by: Lorenz Herzberger <lorenzherzberger@gmail.com>
This commit is contained in:
Lorenz Herzberger 2024-08-06 12:18:02 +02:00
parent 20d5c24fe0
commit 26814fecd6
No known key found for this signature in database
GPG Key ID: FA5EE906EB55316A
2 changed files with 140 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/e2e/machine/cli_test.go
View File

@ -15,3 +15,9 @@ func TestE2EMachineTestSuite(t *testing.T) {
cfg.NumValidators = 3
suite.Run(t, NewE2ETestSuite(cfg))
}
func TestE2EProsumeTestSuite(t *testing.T) {
cfg := network.LoaderDefaultConfig()
cfg.NumValidators = 3
suite.Run(t, NewProsumeE2ETestSuite(cfg))
}

134
tests/e2e/machine/prosume_suite.go Normal file
View File

@ -0,0 +1,134 @@
package machine
import (
"fmt"
"strings"
sdktestutil "github.com/cosmos/cosmos-sdk/testutil"
sdk "github.com/cosmos/cosmos-sdk/types"
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
"github.com/planetmint/planetmint-go/lib"
e2etestutil "github.com/planetmint/planetmint-go/testutil/e2e"
"github.com/planetmint/planetmint-go/testutil/network"
"github.com/planetmint/planetmint-go/testutil/sample"
machinetypes "github.com/planetmint/planetmint-go/x/machine/types"
"github.com/stretchr/testify/suite"
)
var machines = []struct {
name string
mnemonic string
address string
}{
{
name: "R2D2",
mnemonic: "number judge garbage lock village slush business upset suspect green wrestle puzzle foil tragic drum stereo ticket teach upper bone inject monkey deny portion",
address: "plmnt1kp93kns6hs2066d8qw0uz84fw3vlthewt2ck6p",
},
{
name: "C3PO",
mnemonic: "letter plate husband impulse grid lake panel seminar try powder virtual run spice siege mutual enhance ripple country two boring have convince symptom fuel",
address: "plmnt15wrx9eqegjtlvvx80huau7rkn3f44rdj969xrx",
},
}
// ProsumeE2ETestSuite struct definition of prosume e2e test suite
type ProsumeE2ETestSuite struct {
suite.Suite
cfg network.Config
network *network.Network
feeDenom string
}
// NewProsumeE2ETestSuite returns configured prosume e2e test suite
func NewProsumeE2ETestSuite(cfg network.Config) *ProsumeE2ETestSuite {
return &ProsumeE2ETestSuite{cfg: cfg}
}
func (s *ProsumeE2ETestSuite) SetupSuite() {
s.T().Log("setting up prosume e2e test suite")
s.feeDenom = sample.FeeDenom
s.network = network.Load(s.T(), s.cfg)
// create machines
for i, machine := range machines {
s.Require().NoError(e2etestutil.AttestMachine(s.network, machine.name, machine.mnemonic, i, s.feeDenom))
}
}
func (s *ProsumeE2ETestSuite) TestProsume() {
val := s.network.Validators[0]
producerKeyring, err := val.ClientCtx.Keyring.Key(machines[0].name)
s.Require().NoError(err)
producerAddress, err := producerKeyring.GetAddress()
s.Require().NoError(err)
production := sdk.NewCoins(sdk.NewCoin("kwh", sdk.NewInt(1000)))
mintProdMsg := machinetypes.NewMsgMintProduction(machines[0].address, production)
out, err := e2etestutil.BuildSignBroadcastTx(s.T(), producerAddress, mintProdMsg)
s.Require().NoError(err)
txRes, err := lib.GetTxResponseFromOut(out)
s.Require().NoError(err)
s.Require().Equal(int(0), int(txRes.Code))
s.Require().NoError(s.network.WaitForNextBlock())
// check balance
var queryBankBalanceResp banktypes.QueryAllBalancesResponse
baseURL := strings.ReplaceAll(strings.ReplaceAll(val.APIAddress, "[", ""), "]", "")
url := fmt.Sprintf("%s/cosmos/bank/v1beta1/balances/%s", baseURL, machines[0].address)
resp, err := sdktestutil.GetRequest(url)
s.Require().NoError(err)
s.Require().NoError(val.ClientCtx.Codec.UnmarshalJSON(resp, &queryBankBalanceResp))
s.Require().Equal(queryBankBalanceResp.Balances.AmountOf("kwh"), sdk.NewInt(1000))
consumerKeyring, err := val.ClientCtx.Keyring.Key(machines[1].name)
s.Require().NoError(err)
consumerAddress, err := consumerKeyring.GetAddress()
s.Require().NoError(err)
sendMsg := banktypes.NewMsgSend(producerAddress, consumerAddress, production)
out, err = e2etestutil.BuildSignBroadcastTx(s.T(), producerAddress, sendMsg)
s.Require().NoError(err)
txRes, err = lib.GetTxResponseFromOut(out)
s.Require().NoError(err)
s.Require().Equal(int(0), int(txRes.Code))
s.Require().NoError(s.network.WaitForNextBlock())
// check balances
url = fmt.Sprintf("%s/cosmos/bank/v1beta1/balances/%s", baseURL, machines[0].address)
resp, err = sdktestutil.GetRequest(url)
s.Require().NoError(err)
s.Require().NoError(val.ClientCtx.Codec.UnmarshalJSON(resp, &queryBankBalanceResp))
s.Require().Equal(queryBankBalanceResp.Balances.AmountOf("kwh"), sdk.NewInt(0))
url = fmt.Sprintf("%s/cosmos/bank/v1beta1/balances/%s", baseURL, machines[1].address)
resp, err = sdktestutil.GetRequest(url)
s.Require().NoError(err)
s.Require().NoError(val.ClientCtx.Codec.UnmarshalJSON(resp, &queryBankBalanceResp))
s.Require().Equal(queryBankBalanceResp.Balances.AmountOf("kwh"), sdk.NewInt(1000))
consumption := sdk.NewCoins(sdk.NewCoin("kwh", sdk.NewInt(500)))
burnConsMsg := machinetypes.NewMsgBurnConsumption(machines[1].address, consumption)
out, err = e2etestutil.BuildSignBroadcastTx(s.T(), consumerAddress, burnConsMsg)
s.Require().NoError(err)
txRes, err = lib.GetTxResponseFromOut(out)
s.Require().NoError(err)
s.Require().Equal(int(0), int(txRes.Code))
s.Require().NoError(s.network.WaitForNextBlock())
// check balance
url = fmt.Sprintf("%s/cosmos/bank/v1beta1/balances/%s", baseURL, machines[1].address)
resp, err = sdktestutil.GetRequest(url)
s.Require().NoError(err)
s.Require().NoError(val.ClientCtx.Codec.UnmarshalJSON(resp, &queryBankBalanceResp))
s.Require().Equal(queryBankBalanceResp.Balances.AmountOf("kwh"), sdk.NewInt(500))
}