From 52dcee60c9d1c5df6002aa2c06a5cee47b7e8f9d Mon Sep 17 00:00:00 2001 From: scorpioborn <97235353+scorpioborn@users.noreply.github.com> Date: Thu, 25 Apr 2024 09:32:30 +0300 Subject: [PATCH] tmp: e2e test requirements --- tests/reward/cli_test.go | 14 +++ tests/reward/grpc.go | 1 + tests/reward/suite.go | 204 ++++++++++++++++++++++++++++++++ testutil/network/network.go | 130 +++++++++++++++++++- testutil/simapp/test_helpers.go | 16 +++ 5 files changed, 362 insertions(+), 3 deletions(-) create mode 100644 tests/reward/cli_test.go create mode 100644 tests/reward/grpc.go create mode 100644 tests/reward/suite.go diff --git a/tests/reward/cli_test.go b/tests/reward/cli_test.go new file mode 100644 index 00000000..497bbde6 --- /dev/null +++ b/tests/reward/cli_test.go @@ -0,0 +1,14 @@ +package client + +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)) +} diff --git a/tests/reward/grpc.go b/tests/reward/grpc.go new file mode 100644 index 00000000..da13c8ef --- /dev/null +++ b/tests/reward/grpc.go @@ -0,0 +1 @@ +package client diff --git a/tests/reward/suite.go b/tests/reward/suite.go new file mode 100644 index 00000000..5ff511fd --- /dev/null +++ b/tests/reward/suite.go @@ -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) + s.Require().Equal(tc.expectedCode, txResp.Code) + } + }) + } +} diff --git a/testutil/network/network.go b/testutil/network/network.go index 291f0317..3524a46c 100644 --- a/testutil/network/network.go +++ b/testutil/network/network.go @@ -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 ( @@ -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 { @@ -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, @@ -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), diff --git a/testutil/simapp/test_helpers.go b/testutil/simapp/test_helpers.go index c2a6fd8e..b15bdd86 100644 --- a/testutil/simapp/test_helpers.go +++ b/testutil/simapp/test_helpers.go @@ -7,6 +7,8 @@ import ( "encoding/hex" "fmt" "math" + "os" + "path/filepath" "github.com/golang-jwt/jwt" "github.com/spf13/cast" @@ -111,3 +113,17 @@ func RandomString(length int) string { str := base64.RawURLEncoding.EncodeToString(buff) return str[:length] // strip 1 extra character we get from odd length results } + +func WriteKeyringFile(name string, dir string, contents []byte) error { + file := filepath.Join(dir, name) + + if err := os.MkdirAll(dir, 0o755); err != nil { + return fmt.Errorf("could not create directory %q: %w", dir, err) + } + + if err := os.WriteFile(file, contents, 0o644); err != nil { //nolint: gosec + return err + } + + return nil +}