Skip to content

Commit

Permalink
tmp: e2e test requirements
Browse files Browse the repository at this point in the history
  • Loading branch information
scorpioborn committed Apr 25, 2024
1 parent a109ef6 commit 52dcee6
Show file tree
Hide file tree
Showing 5 changed files with 362 additions and 3 deletions.
14 changes: 14 additions & 0 deletions tests/reward/cli_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
package client

Check failure on line 1 in tests/reward/cli_test.go

View workflow job for this annotation

GitHub Actions / Run golangci-lint

package should be `client_test` instead of `client` (testpackage)

import (
"testing"

"github.com/sge-network/sge/testutil/network"
"github.com/stretchr/testify/suite"
)

func TestE2ETestSuite(t *testing.T) {
cfg := network.DefaultConfig()
cfg.NumValidators = 1
suite.Run(t, NewE2ETestSuite(cfg))
}
1 change: 1 addition & 0 deletions tests/reward/grpc.go
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
package client
204 changes: 204 additions & 0 deletions tests/reward/suite.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,204 @@
package client

import (
"fmt"
"time"

sdkmath "cosmossdk.io/math"
"github.com/cosmos/cosmos-sdk/client/flags"
clitestutil "github.com/cosmos/cosmos-sdk/testutil/cli"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cosmos/gogoproto/proto"
"github.com/golang-jwt/jwt"
"github.com/google/uuid"
"github.com/spf13/cast"
"github.com/stretchr/testify/require"
"github.com/stretchr/testify/suite"

"github.com/sge-network/sge/testutil/network"
"github.com/sge-network/sge/testutil/simapp"
"github.com/sge-network/sge/x/reward/client/cli"
"github.com/sge-network/sge/x/reward/types"
)

var (
genesis types.GenesisState
promoterUID = uuid.NewString()
campaignStartDate = uint64(time.Now().Unix())
campaignEndDate = uint64(time.Now().Add(5 * time.Minute).Unix())
)

type E2ETestSuite struct {
suite.Suite

cfg network.Config
network *network.Network
}

func NewE2ETestSuite(cfg network.Config) *E2ETestSuite {
return &E2ETestSuite{cfg: cfg}
}

func (s *E2ETestSuite) SetupSuite() {
s.T().Log("setting up e2e test suite")

genesisState := s.cfg.GenesisState
s.Require().NoError(s.cfg.Codec.UnmarshalJSON(genesisState[types.ModuleName], &genesis))

// promoterAddr = simapp.TestParamUsers["user1"].Address.String()

// genesis.PromoterList = []types.Promoter{
// {
// Creator: promoterAddr,
// UID: uuid.NewString(),
// Addresses: []string{promoterAddr},
// Conf: types.PromoterConf{
// CategoryCap: []types.CategoryCap{{
// Category: types.RewardCategory_REWARD_CATEGORY_SIGNUP,
// CapPerAcc: 1,
// }},
// },
// },
// }

// genesis.PromoterByAddressList = []types.PromoterByAddress{
// {
// PromoterUID: genesis.PromoterList[0].UID,
// Address: promoterAddr,
// },
// }

// genesis.CampaignList = []types.Campaign{
// {
// Creator: promoterAddr,
// UID: uuid.NewString(),
// Promoter: promoterAddr,
// StartTS: campaignStartDate,
// EndTS: campaignEndDate,
// RewardCategory: types.RewardCategory_REWARD_CATEGORY_SIGNUP,
// RewardType: types.RewardType_REWARD_TYPE_SIGNUP,
// RewardAmountType: types.RewardAmountType_REWARD_AMOUNT_TYPE_FIXED,
// RewardAmount: &types.RewardAmount{
// SubaccountAmount: sdkmath.NewInt(100),
// },
// Pool: ,
// },
// }

genesisBz, err := s.cfg.Codec.MarshalJSON(&genesis)
s.Require().NoError(err)
genesisState[types.ModuleName] = genesisBz
s.cfg.GenesisState = genesisState

s.network = network.New(s.T(), s.cfg)
s.Require().NoError(s.network.WaitForNextBlock())
}

// func (s *E2ETestSuite) TearDownSuite() {
// s.T().Log("tearing down e2e test suite")
// s.network.Cleanup()
// }

func (s *E2ETestSuite) TestNewCampaignTxCmd() {
val := s.network.Validators[0]

{
clientCtx := val.ClientCtx

ticket, err := simapp.CreateJwtTicket(jwt.MapClaims{
"exp": time.Now().Add(time.Minute * 5).Unix(),
"iat": time.Now().Unix(),
"uid": promoterUID,
"conf": types.PromoterConf{
CategoryCap: []types.CategoryCap{{
Category: types.RewardCategory_REWARD_CATEGORY_SIGNUP,
CapPerAcc: 1,
}},
},
})
require.Nil(s.T(), err)

args := []string{
ticket,
fmt.Sprintf("--%s=%s", flags.FlagFrom, val.Address),
fmt.Sprintf("--%s=true", flags.FlagSkipConfirmation),
fmt.Sprintf("--%s=%s", flags.FlagBroadcastMode, flags.BroadcastSync),
fmt.Sprintf("--%s=%s", flags.FlagFees, sdk.NewCoins(sdk.NewCoin(s.cfg.BondDenom, sdkmath.NewInt(10))).String()),
}
bz, err := clitestutil.ExecTestCLICmd(clientCtx, cli.CmdCreatePromoter(), args)
s.Require().NoError(err)
respType := sdk.TxResponse{}
s.Require().NoError(clientCtx.Codec.UnmarshalJSON(bz.Bytes(), &respType), bz.String())
s.Require().Equal(uint32(0), respType.Code)
}

testCases := []struct {
name string
uid string
totalFunds sdkmath.Int
ticketClaims jwt.MapClaims
args []string
expectErr bool
expectedCode uint32
respType proto.Message
}{
{
"valid transaction",
uuid.NewString(),
sdkmath.NewInt(100),
jwt.MapClaims{
"exp": time.Now().Add(time.Minute * 5).Unix(),
"iat": time.Now().Unix(),
"promoter": val.Address,
"start_ts": cast.ToString(campaignStartDate),
"end_ts": cast.ToString(campaignEndDate),
"category": types.RewardCategory_REWARD_CATEGORY_SIGNUP,
"reward_type": types.RewardType_REWARD_TYPE_SIGNUP,
"reward_amount_type": types.RewardAmountType_REWARD_AMOUNT_TYPE_FIXED,
"reward_amount": types.RewardAmount{
SubaccountAmount: sdkmath.NewInt(100),
UnlockPeriod: uint64(1000),
},
"is_active": true,
"meta": "sample signup campaign",
"cap_count": 1,
},
[]string{
fmt.Sprintf("--%s=%s", flags.FlagFrom, val.Address),
fmt.Sprintf("--%s=true", flags.FlagSkipConfirmation),
fmt.Sprintf("--%s=%s", flags.FlagBroadcastMode, flags.BroadcastSync),
fmt.Sprintf("--%s=%s", flags.FlagFees, sdk.NewCoins(sdk.NewCoin(s.cfg.BondDenom, sdkmath.NewInt(10))).String()),
},
false, 0, &sdk.TxResponse{},
},
}

for _, tc := range testCases {
tc := tc

s.Require().NoError(s.network.WaitForNextBlock())
s.Run(tc.name, func() {
clientCtx := val.ClientCtx

ticket, err := simapp.CreateJwtTicket(tc.ticketClaims)
require.Nil(s.T(), err)

tc.args = append([]string{
tc.uid,
tc.totalFunds.String(),
ticket,
}, tc.args...)
bz, err := clitestutil.ExecTestCLICmd(clientCtx, cli.CmdCreateCampaign(), tc.args)

if tc.expectErr {
s.Require().Error(err)
} else {
s.Require().NoError(err)

s.Require().NoError(clientCtx.Codec.UnmarshalJSON(bz.Bytes(), tc.respType), bz.String())
txResp := tc.respType.(*sdk.TxResponse)

Check failure on line 199 in tests/reward/suite.go

View workflow job for this annotation

GitHub Actions / Run golangci-lint

type assertion must be checked (forcetypeassert)
s.Require().Equal(tc.expectedCode, txResp.Code)
}
})
}
}
130 changes: 127 additions & 3 deletions testutil/network/network.go
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,12 @@ import (
simtestutil "github.com/cosmos/cosmos-sdk/testutil/sims"
sdk "github.com/cosmos/cosmos-sdk/types"
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"

"github.com/sge-network/sge/app"
"github.com/sge-network/sge/app/params"
"github.com/sge-network/sge/testutil/simapp"
ovmtypes "github.com/sge-network/sge/x/ovm/types"
)

type (
Expand All @@ -39,6 +43,39 @@ func New(t *testing.T, configs ...network.Config) *network.Network {
} else {
cfg = configs[0]
}
// baseDir := t.TempDir()

// nodeDirName := fmt.Sprintf("node%d", 0)
// node0Dir := filepath.Join(baseDir, nodeDirName, "simcli")

// buf := bufio.NewReader(os.Stdin)
// kb, err := keyring.New(sdk.KeyringServiceName(), keyring.BackendTest, node0Dir, buf, cfg.Codec, cfg.KeyringOptions...)
// if err != nil {
// panic(err)
// }

// keyringAlgos, _ := kb.SupportedAlgorithms()
// algo, err := keyring.NewSigningAlgoFromString(cfg.SigningAlgo, keyringAlgos)
// if err != nil {
// panic(err)
// }

// addr, secret, err := testutil.GenerateSaveCoinKey(kb, nodeDirName, "", true, algo)
// if err != nil {
// panic(err)
// }

// info := map[string]string{"secret": secret}
// infoBz, err := json.Marshal(info)
// if err != nil {
// panic(err)
// }

// // save private key seed words
// err = simapp.WriteKeyringFile(fmt.Sprintf("%v.json", "key_seed"), node0Dir, infoBz)
// if err != nil {
// panic(err)
// }

net, err := network.New(t, t.TempDir(), cfg)
if err != nil {
Expand All @@ -56,6 +93,93 @@ func DefaultConfig() network.Config {
encoding = app.MakeEncodingConfig()
chainID = "chain-" + tmrand.NewRand().Str(6)
)

// simapp.GenerateSimappUsers()

// Initialize test app by genesis account
// genAccs := simapp.GenerateSimappGenesisAccounts()

// Create testapp instance
// balances := simapp.GenerateSimappUserBalances()

defGen := app.ModuleBasics.DefaultGenesis(encoding.Marshaler)
{
// modify the staking denom in the genesis
stakingGenState := defGen[stakingtypes.ModuleName]
var newStakingGenState stakingtypes.GenesisState

if err := encoding.Marshaler.UnmarshalJSON(stakingGenState, &newStakingGenState); err != nil {
panic(err)
}

// change to default bond denom
newStakingGenState.Params.BondDenom = params.DefaultBondDenom

var err error
defGen[stakingtypes.ModuleName], err = encoding.Marshaler.MarshalJSON(&newStakingGenState)
if err != nil {
panic(err)
}
}
// {
// authGenState := defGen[authtypes.ModuleName]
// var newAuthGenState authtypes.GenesisState

// if err := encoding.Marshaler.UnmarshalJSON(authGenState, &newAuthGenState); err != nil {
// panic(err)
// }

// for _, v := range genAccs {
// anyVal, err := codectypes.NewAnyWithValue(v)
// if err != nil {
// panic(err)
// }
// newAuthGenState.Accounts = append(newAuthGenState.Accounts, anyVal)
// }

// var err error
// defGen[authtypes.ModuleName], err = encoding.Marshaler.MarshalJSON(&newAuthGenState)
// if err != nil {
// panic(err)
// }
// }
// {
// bankGenState := defGen[banktypes.ModuleName]
// var newBankGenState banktypes.GenesisState

// if err := encoding.Marshaler.UnmarshalJSON(bankGenState, &newBankGenState); err != nil {
// panic(err)
// }

// newBankGenState.Balances = append(newBankGenState.Balances, balances...)

// // totalSupply := sdk.NewCoins()
// // for _, b := range balances {
// // totalSupply = totalSupply.Add(b.Coins...)
// // }
// // newBankGenState.Supply = totalSupply

// var err error
// defGen[banktypes.ModuleName], err = encoding.Marshaler.MarshalJSON(&newBankGenState)
// if err != nil {
// panic(err)
// }

// // stateBytes, err := json.MarshalIndent(genesisState, "", " ")
// // if err != nil {
// // panic(err)
// // }
// }

{
ovmGenesisState := &ovmtypes.GenesisState{
KeyVault: ovmtypes.KeyVault{
PublicKeys: simapp.GenerateOvmPublicKeys(ovmtypes.MinPubKeysCount),
},
}
defGen[ovmtypes.ModuleName] = encoding.Marshaler.MustMarshalJSON(ovmGenesisState)
}

return network.Config{
Codec: encoding.Marshaler,
TxConfig: encoding.TxConfig,
Expand All @@ -78,12 +202,12 @@ func DefaultConfig() network.Config {
baseapp.SetChainID(chainID),
)
},
GenesisState: app.ModuleBasics.DefaultGenesis(encoding.Marshaler),
GenesisState: defGen,
TimeoutCommit: 2 * time.Second,
ChainID: chainID,
NumValidators: 1,
BondDenom: sdk.DefaultBondDenom,
MinGasPrices: fmt.Sprintf("0.000006%s", sdk.DefaultBondDenom),
BondDenom: params.DefaultBondDenom,
MinGasPrices: fmt.Sprintf("0.000006%s", params.DefaultBondDenom),
AccountTokens: sdk.TokensFromConsensusPower(1000, sdk.DefaultPowerReduction),
StakingTokens: sdk.TokensFromConsensusPower(500, sdk.DefaultPowerReduction),
BondedTokens: sdk.TokensFromConsensusPower(100, sdk.DefaultPowerReduction),
Expand Down
Loading

0 comments on commit 52dcee6

Please sign in to comment.