mirror of
https://github.com/planetmint/planetmint-go.git
synced 2025-03-30 15:08:28 +00:00
2534828fc6
* feature: add ante handler to block non validators from sending restricted txs * fix: add missing msg types and linter errors * fix: staticcheck error * refactor: use fallthrough on ante handle switch * fix: set val address on machine attest test --------- Signed-off-by: Lorenz Herzberger <lorenzherzberger@gmail.com>
107 lines
3.2 KiB
Go
107 lines
3.2 KiB
Go
package machine
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/planetmint/planetmint-go/config"
|
|
"github.com/planetmint/planetmint-go/testutil"
|
|
"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/assert"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
clitestutil "github.com/planetmint/planetmint-go/testutil/cli"
|
|
e2etestutil "github.com/planetmint/planetmint-go/testutil/e2e"
|
|
)
|
|
|
|
// RestE2ETestSuite struct definition of machine suite
|
|
type RestE2ETestSuite struct {
|
|
suite.Suite
|
|
|
|
cfg network.Config
|
|
network *network.Network
|
|
}
|
|
|
|
// NewRestE2ETestSuite returns configured machine RestE2ETestSuite
|
|
func NewRestE2ETestSuite(cfg network.Config) *RestE2ETestSuite {
|
|
return &RestE2ETestSuite{cfg: cfg}
|
|
}
|
|
|
|
// SetupSuite initializes machine E2ETestSuite
|
|
func (s *RestE2ETestSuite) SetupSuite() {
|
|
conf := config.GetConfig()
|
|
conf.FeeDenom = "stake"
|
|
|
|
s.T().Log("setting up e2e test suite")
|
|
|
|
s.network = network.New(s.T())
|
|
// create machine account for attestation
|
|
account, err := e2etestutil.CreateAccount(s.network, sample.Name, sample.Mnemonic)
|
|
s.Require().NoError(err)
|
|
err = e2etestutil.FundAccount(s.network, account)
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
// TearDownSuite clean up after testing
|
|
func (s *RestE2ETestSuite) TearDownSuite() {
|
|
s.T().Log("tearing down e2e test suite")
|
|
}
|
|
|
|
func (s *RestE2ETestSuite) TestAttestMachineREST() {
|
|
val := s.network.Validators[0]
|
|
baseURL := val.APIAddress
|
|
|
|
// Query Sequence Number
|
|
k, err := val.ClientCtx.Keyring.Key(sample.Name)
|
|
s.Require().NoError(err)
|
|
|
|
addr, err := k.GetAddress()
|
|
s.Require().NoError(err)
|
|
|
|
prvKey, pubKey := sample.KeyPair()
|
|
|
|
// Register TA
|
|
ta := sample.TrustAnchor(pubKey)
|
|
taMsg := machinetypes.MsgRegisterTrustAnchor{
|
|
Creator: val.Address.String(),
|
|
TrustAnchor: &ta,
|
|
}
|
|
out, err := e2etestutil.BuildSignBroadcastTx(s.T(), val.Address, &taMsg)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NoError(s.network.WaitForNextBlock())
|
|
rawLog, err := clitestutil.GetRawLogFromTxOut(val, out)
|
|
s.Require().NoError(err)
|
|
|
|
assert.Contains(s.T(), rawLog, "planetmintgo.machine.MsgRegisterTrustAnchor")
|
|
|
|
// Create Attest Machine TX
|
|
machine := sample.Machine(sample.Name, pubKey, prvKey, addr.String())
|
|
msg := machinetypes.MsgAttestMachine{
|
|
Creator: addr.String(),
|
|
Machine: &machine,
|
|
}
|
|
out, err = e2etestutil.BuildSignBroadcastTx(s.T(), addr, &msg)
|
|
s.Require().NoError(err)
|
|
|
|
// give machine attestation some time to issue the liquid asset
|
|
s.Require().NoError(s.network.WaitForNextBlock())
|
|
s.Require().NoError(s.network.WaitForNextBlock())
|
|
s.Require().NoError(s.network.WaitForNextBlock())
|
|
|
|
rawLog, err = clitestutil.GetRawLogFromTxOut(val, out)
|
|
s.Require().NoError(err)
|
|
|
|
assert.Contains(s.T(), rawLog, "planetmintgo.machine.MsgAttestMachine")
|
|
|
|
queryMachineURL := fmt.Sprintf("%s/planetmint/machine/get_machine_by_public_key/%s", baseURL, pubKey)
|
|
queryMachineRes, err := testutil.GetRequest(queryMachineURL)
|
|
s.Require().NoError(err)
|
|
|
|
var qmRes machinetypes.QueryGetMachineByPublicKeyResponse
|
|
err = val.ClientCtx.Codec.UnmarshalJSON(queryMachineRes, &qmRes)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(&machine, qmRes.Machine)
|
|
}
|