|
1 | 1 | package integration_tests |
2 | 2 |
|
3 | 3 | import ( |
| 4 | + "context" |
4 | 5 | "crypto/sha256" |
5 | 6 | "encoding/base64" |
6 | 7 | "encoding/hex" |
@@ -29,6 +30,9 @@ import ( |
29 | 30 | "github.com/stretchr/testify/require" |
30 | 31 |
|
31 | 32 | wasmkeeper "github.com/CosmWasm/wasmd/x/wasm/keeper" |
| 33 | + cometClient "github.com/cometbft/cometbft/rpc/client" |
| 34 | + rpchttp "github.com/cometbft/cometbft/rpc/client/http" |
| 35 | + cometRpcCoreTypes "github.com/cometbft/cometbft/rpc/core/types" |
32 | 36 | "github.com/cosmos/cosmos-sdk/crypto/keyring" |
33 | 37 | authsigning "github.com/cosmos/cosmos-sdk/x/auth/signing" |
34 | 38 | aatypes "github.com/larry0x/abstract-account/x/abstractaccount/types" |
@@ -580,7 +584,7 @@ func TestXionAbstractAccount(t *testing.T) { |
580 | 584 | aaContractAddr, |
581 | 585 | path.Join(xion.GetNode().HomeDir(), removeFilePath[len(removeFilePath)-1]), |
582 | 586 | "--chain-id", xion.Config().ChainID, |
583 | | - "--authenticator-id", "1", |
| 587 | + "--nuthenticator-id", "1", |
584 | 588 | ) |
585 | 589 | require.NoError(t, err) |
586 | 590 |
|
@@ -617,3 +621,287 @@ func GetAAContractAddress(t *testing.T, txDetails map[string]interface{}) string |
617 | 621 |
|
618 | 622 | return addr |
619 | 623 | } |
| 624 | + |
| 625 | +func TestXionClientEvent(t *testing.T) { |
| 626 | + if testing.Short() { |
| 627 | + t.Skip("skipping in short mode") |
| 628 | + } |
| 629 | + |
| 630 | + t.Parallel() |
| 631 | + td := BuildXionChain(t, "0.0uxion", ModifyInterChainGenesis(ModifyInterChainGenesisFn{ModifyGenesisShortProposals}, [][]string{{votingPeriod, maxDepositPeriod}})) |
| 632 | + xion, ctx := td.xionChain, td.ctx |
| 633 | + |
| 634 | + config := types.GetConfig() |
| 635 | + config.SetBech32PrefixForAccount("xion", "xionpub") |
| 636 | + |
| 637 | + // Create and Fund User Wallets |
| 638 | + t.Log("creating and funding user accounts") |
| 639 | + fundAmount := math.NewInt(10_000_000) |
| 640 | + xionUser, err := ibctest.GetAndFundTestUserWithMnemonic(ctx, "default", deployerMnemonic, fundAmount, xion) |
| 641 | + require.NoError(t, err) |
| 642 | + currentHeight, _ := xion.Height(ctx) |
| 643 | + err = testutil.WaitForBlocks(ctx, int(currentHeight)+8, xion) |
| 644 | + require.NoError(t, err) |
| 645 | + t.Logf("created xion user %s", xionUser.FormattedAddress()) |
| 646 | + |
| 647 | + xionUserBalInitial, err := xion.GetBalance(ctx, xionUser.FormattedAddress(), xion.Config().Denom) |
| 648 | + require.NoError(t, err) |
| 649 | + require.Equal(t, fundAmount, xionUserBalInitial) |
| 650 | + |
| 651 | + // Create a Secondary Key For Rotation |
| 652 | + recipientKeyName := "recipient-key" |
| 653 | + err = xion.CreateKey(ctx, recipientKeyName) |
| 654 | + require.NoError(t, err) |
| 655 | + receipientKeyAddressBytes, err := xion.GetAddress(ctx, recipientKeyName) |
| 656 | + require.NoError(t, err) |
| 657 | + recipientKeyAddress, err := types.Bech32ifyAddressBytes(xion.Config().Bech32Prefix, receipientKeyAddressBytes) |
| 658 | + require.NoError(t, err) |
| 659 | + |
| 660 | + // Get Public Key For Funded Account |
| 661 | + account, err := ExecBin(t, ctx, xion.GetNode(), |
| 662 | + "keys", "show", |
| 663 | + xionUser.KeyName(), |
| 664 | + "--keyring-backend", keyring.BackendTest, |
| 665 | + "-p", |
| 666 | + ) |
| 667 | + require.NoError(t, err) |
| 668 | + t.Log("Funded Account:") |
| 669 | + for k, v := range account { |
| 670 | + t.Logf("[%s]: %v", k, v) |
| 671 | + } |
| 672 | + |
| 673 | + fp, err := os.Getwd() |
| 674 | + require.NoError(t, err) |
| 675 | + |
| 676 | + // Store Wasm Contract |
| 677 | + codeID, err := xion.StoreContract(ctx, xionUser.FormattedAddress(), path.Join(fp, |
| 678 | + "integration_tests", "testdata", "contracts", "account-wasm-updatable-event-aarch64.wasm")) |
| 679 | + require.NoError(t, err) |
| 680 | + |
| 681 | + // retrieve the hash |
| 682 | + codeResp, err := ExecQuery(t, ctx, xion.GetNode(), |
| 683 | + "wasm", "code-info", codeID) |
| 684 | + require.NoError(t, err) |
| 685 | + t.Logf("code response: %s", codeResp) |
| 686 | + |
| 687 | + depositedFunds := fmt.Sprintf("%d%s", 100000, xion.Config().Denom) |
| 688 | + |
| 689 | + registeredTxHash, err := ExecTx(t, ctx, xion.GetNode(), |
| 690 | + xionUser.KeyName(), |
| 691 | + "xion", "register", |
| 692 | + codeID, |
| 693 | + xionUser.KeyName(), |
| 694 | + "--funds", depositedFunds, |
| 695 | + "--salt", "0", |
| 696 | + "--authenticator", "Secp256K1", |
| 697 | + "--chain-id", xion.Config().ChainID, |
| 698 | + ) |
| 699 | + require.NoError(t, err) |
| 700 | + t.Logf("tx hash: %s", registeredTxHash) |
| 701 | + |
| 702 | + txDetails, err := ExecQuery(t, ctx, xion.GetNode(), "tx", registeredTxHash) |
| 703 | + require.NoError(t, err) |
| 704 | + t.Logf("TxDetails: %s", txDetails) |
| 705 | + aaContractAddr := GetAAContractAddress(t, txDetails) |
| 706 | + |
| 707 | + contractBalance, err := xion.GetBalance(ctx, aaContractAddr, xion.Config().Denom) |
| 708 | + require.NoError(t, err) |
| 709 | + require.Equal(t, math.NewInt(100000), contractBalance) |
| 710 | + |
| 711 | + contractState, err := ExecQuery(t, ctx, xion.GetNode(), "wasm", "contract-state", "smart", aaContractAddr, `{"authenticator_by_i_d":{ "id": 0 }}`) |
| 712 | + require.NoError(t, err) |
| 713 | + |
| 714 | + pubkey64, ok := contractState["data"].(string) |
| 715 | + require.True(t, ok) |
| 716 | + pubkeyRawJSON, err := base64.StdEncoding.DecodeString(pubkey64) |
| 717 | + require.NoError(t, err) |
| 718 | + var pubKeyMap jsonAuthenticator |
| 719 | + json.Unmarshal(pubkeyRawJSON, &pubKeyMap) |
| 720 | + require.Equal(t, account["key"], pubKeyMap["Secp256K1"]["pubkey"]) |
| 721 | + |
| 722 | + // Generate Msg Send without signatures |
| 723 | + jsonMsg := RawJSONMsgSend(t, aaContractAddr, recipientKeyAddress, xion.Config().Denom) |
| 724 | + require.NoError(t, err) |
| 725 | + require.True(t, json.Valid(jsonMsg)) |
| 726 | + |
| 727 | + sendFile, err := os.CreateTemp("", "*-msg-bank-send.json") |
| 728 | + require.NoError(t, err) |
| 729 | + defer os.Remove(sendFile.Name()) |
| 730 | + |
| 731 | + _, err = sendFile.Write(jsonMsg) |
| 732 | + require.NoError(t, err) |
| 733 | + |
| 734 | + err = UploadFileToContainer(t, ctx, xion.GetNode(), sendFile) |
| 735 | + require.NoError(t, err) |
| 736 | + |
| 737 | + // Sign and broadcast a transaction |
| 738 | + sendFilePath := strings.Split(sendFile.Name(), "/") |
| 739 | + _, err = ExecTx(t, ctx, xion.GetNode(), |
| 740 | + xionUser.KeyName(), |
| 741 | + "xion", "sign", |
| 742 | + xionUser.KeyName(), |
| 743 | + aaContractAddr, |
| 744 | + path.Join(xion.GetNode().HomeDir(), sendFilePath[len(sendFilePath)-1]), |
| 745 | + "--chain-id", xion.Config().ChainID, |
| 746 | + ) |
| 747 | + require.NoError(t, err) |
| 748 | + |
| 749 | + err = testutil.WaitForBlocks(ctx, 1, xion) |
| 750 | + require.NoError(t, err) |
| 751 | + |
| 752 | + // Confirm the updated balance |
| 753 | + balance, err := xion.GetBalance(ctx, recipientKeyAddress, xion.Config().Denom) |
| 754 | + require.NoError(t, err) |
| 755 | + require.Equal(t, math.NewInt(100000).Uint64(), balance.Uint64()) |
| 756 | + |
| 757 | + // Generate Key Rotation Msg |
| 758 | + account, err = ExecBin(t, ctx, xion.GetNode(), |
| 759 | + "keys", "show", |
| 760 | + xionUser.KeyName(), |
| 761 | + "--keyring-backend", keyring.BackendTest, |
| 762 | + "-p", |
| 763 | + ) |
| 764 | + |
| 765 | + // add secondary authenticator to account. in this case, the same key but in a different position |
| 766 | + jsonExecMsgStr, err := GenerateTx(t, ctx, xion.GetNode(), |
| 767 | + xionUser.KeyName(), |
| 768 | + "xion", "add-authenticator", aaContractAddr, |
| 769 | + "--authenticator-id", "1", |
| 770 | + "--chain-id", xion.Config().ChainID, |
| 771 | + ) |
| 772 | + require.NoError(t, err) |
| 773 | + jsonExecMsg := []byte(jsonExecMsgStr) |
| 774 | + require.True(t, json.Valid(jsonExecMsg)) |
| 775 | + |
| 776 | + rotateFile, err := os.CreateTemp("", "*-msg-exec-rotate-key.json") |
| 777 | + require.NoError(t, err) |
| 778 | + defer os.Remove(rotateFile.Name()) |
| 779 | + |
| 780 | + _, err = rotateFile.Write(jsonExecMsg) |
| 781 | + require.NoError(t, err) |
| 782 | + |
| 783 | + err = UploadFileToContainer(t, ctx, xion.GetNode(), rotateFile) |
| 784 | + require.NoError(t, err) |
| 785 | + |
| 786 | + rotateFilePath := strings.Split(rotateFile.Name(), "/") |
| 787 | + |
| 788 | + _, err = ExecTx(t, ctx, xion.GetNode(), |
| 789 | + xionUser.KeyName(), |
| 790 | + "xion", "sign", |
| 791 | + xionUser.KeyName(), |
| 792 | + aaContractAddr, |
| 793 | + path.Join(xion.GetNode().HomeDir(), rotateFilePath[len(rotateFilePath)-1]), |
| 794 | + "--chain-id", xion.Config().ChainID, |
| 795 | + ) |
| 796 | + require.NoError(t, err) |
| 797 | + updatedContractState, err := ExecQuery(t, ctx, xion.GetNode(), "wasm", "contract-state", "smart", aaContractAddr, `{"authenticator_by_i_d":{ "id": 1 }}`) |
| 798 | + require.NoError(t, err) |
| 799 | + |
| 800 | + updatedPubKey, ok := updatedContractState["data"].(string) |
| 801 | + require.True(t, ok) |
| 802 | + |
| 803 | + updatedPubKeyRawJSON, err := base64.StdEncoding.DecodeString(updatedPubKey) |
| 804 | + require.NoError(t, err) |
| 805 | + var updatedPubKeyMap jsonAuthenticator |
| 806 | + |
| 807 | + err = json.Unmarshal(updatedPubKeyRawJSON, &updatedPubKeyMap) |
| 808 | + require.NoError(t, err) |
| 809 | + require.Equal(t, account["key"], updatedPubKeyMap["Secp256K1"]["pubkey"]) |
| 810 | + |
| 811 | + cometWsClient, err := getCometClient(xion.GetHostRPCAddress()) // Note: add a close function |
| 812 | + require.NoError(t, err) |
| 813 | + fmt.Printf("%+v\n", xion.GetNode()) |
| 814 | + |
| 815 | + err = cometWsClient.Start() |
| 816 | + require.NoError(t, err) |
| 817 | + |
| 818 | + //eventStream, err := subscribeToEvent(t, ctx, cometTypes.EventTx, cometWsClient) |
| 819 | + eventStream, err := subscribeToEvent(t, ctx, cometWsClient) |
| 820 | + |
| 821 | + require.NoError(t, err) |
| 822 | + |
| 823 | + doneChan := make(chan struct{}) |
| 824 | + go receiveEvents(t, doneChan, eventStream) |
| 825 | + |
| 826 | + jsonExecMsgStr, err = GenerateTx(t, ctx, xion.GetNode(), |
| 827 | + xionUser.KeyName(), |
| 828 | + "xion", "emit", "arbitrary_data", aaContractAddr, |
| 829 | + "--authenticator-id", "0", |
| 830 | + "--chain-id", xion.Config().ChainID, |
| 831 | + ) |
| 832 | + |
| 833 | + require.NoError(t, err) |
| 834 | + jsonExecMsg = []byte(jsonExecMsgStr) |
| 835 | + require.True(t, json.Valid(jsonExecMsg)) |
| 836 | + |
| 837 | + rotateFile, err = os.CreateTemp("", "*-msg-exec-emit.json") |
| 838 | + require.NoError(t, err) |
| 839 | + defer os.Remove(rotateFile.Name()) |
| 840 | + |
| 841 | + _, err = rotateFile.Write(jsonExecMsg) |
| 842 | + require.NoError(t, err) |
| 843 | + |
| 844 | + err = UploadFileToContainer(t, ctx, xion.GetNode(), rotateFile) |
| 845 | + require.NoError(t, err) |
| 846 | + |
| 847 | + rotateFilePath = strings.Split(rotateFile.Name(), "/") |
| 848 | + |
| 849 | + _, err = ExecTx(t, ctx, xion.GetNode(), |
| 850 | + xionUser.KeyName(), |
| 851 | + "xion", "sign", |
| 852 | + xionUser.KeyName(), |
| 853 | + aaContractAddr, |
| 854 | + path.Join(xion.GetNode().HomeDir(), rotateFilePath[len(rotateFilePath)-1]), |
| 855 | + "--chain-id", xion.Config().ChainID, |
| 856 | + ) |
| 857 | + require.NoError(t, err) // it's returning an error and it's not throwing |
| 858 | + fmt.Println("we have thrown a transaction") |
| 859 | + |
| 860 | + //wg.Wait() |
| 861 | + <-doneChan |
| 862 | + stopClient(ctx, cometWsClient) |
| 863 | +} |
| 864 | + |
| 865 | +func getCometClient(hostAddr string) (cometClient.Client, error) { |
| 866 | + return rpchttp.New(hostAddr, "/websocket") |
| 867 | +} |
| 868 | + |
| 869 | +func subscribeToEvent(t *testing.T, ctx context.Context, cli cometClient.Client) (<-chan cometRpcCoreTypes.ResultEvent, error) { |
| 870 | + return cli.Subscribe(ctx, "helpers", "message.module='wasm' AND message.action='/cosmwasm.wasm.v1.MsgExecuteContract'") |
| 871 | +} |
| 872 | + |
| 873 | +func receiveEvents(t *testing.T, done chan<- struct{}, eventStream <-chan cometRpcCoreTypes.ResultEvent) { |
| 874 | + for { |
| 875 | + select { |
| 876 | + case event := <-eventStream: |
| 877 | + fmt.Println("event intercepted") |
| 878 | + contractAddress, ok := event.Events["wasm-account_emit._contract_address"] |
| 879 | + if !ok { |
| 880 | + fmt.Println("not desired event") |
| 881 | + continue |
| 882 | + } |
| 883 | + arbData, ok := event.Events["wasm-account_emit.data"] |
| 884 | + if !ok { |
| 885 | + fmt.Println("not desired event") |
| 886 | + continue |
| 887 | + } |
| 888 | + fmt.Println(contractAddress) |
| 889 | + fmt.Println(arbData) |
| 890 | + done <- struct{}{} |
| 891 | + return |
| 892 | + default: |
| 893 | + continue |
| 894 | + } |
| 895 | + } |
| 896 | +} |
| 897 | + |
| 898 | +func stopClient(ctx context.Context, cli cometClient.Client) error { |
| 899 | + if err := cli.UnsubscribeAll(ctx, "helpers"); err != nil { |
| 900 | + return err |
| 901 | + } |
| 902 | + |
| 903 | + if err := cli.Stop(); err != nil { |
| 904 | + return err |
| 905 | + } |
| 906 | + return nil |
| 907 | +} |
0 commit comments