Problem: encryption-key cmd is not supported

CHANGELOG.md

@@ -19,6 +19,10 @@
 
 * (rpc) [#1397](https://github.com/crypto-org-chain/cronos/pull/1397) Avoid panic on invalid elasticity_multiplier.
 
+### Features
+
+- [#1406](https://github.com/crypto-org-chain/cronos/pull/1406) Add set-encryption-key for encryption module.
+
 *April 8, 2024*
 
 ## v1.2.0-rc1

integration_tests/cosmoscli.py

@@ -1840,3 +1840,34 @@ def query_bank_send(self, *denoms):
                 output="json",
             )
         ).get("send_enabled", [])
+
+    def query_e2ee_key(self, address):
+        return json.loads(
+            self.raw(
+                "q",
+                "e2ee",
+                "key",
+                address,
+                home=self.data_dir,
+                output="json",
+            )
+        )
+
+    def set_e2ee_key(self, key, **kwargs):
+        kwargs.setdefault("gas_prices", DEFAULT_GAS_PRICE)
+        kwargs.setdefault("gas", DEFAULT_GAS)
+        rsp = json.loads(
+            self.raw(
+                "tx",
+                "e2ee",
+                "set-encryption-key",
+                key,
+                "-y",
+                home=self.data_dir,
+                stderr=subprocess.DEVNULL,
+                **kwargs,
+            )
+        )
+        if rsp["code"] == 0:
+            rsp = self.event_query_tx_for(rsp["txhash"])
+        return rsp

integration_tests/test_e2ee.py

@@ -0,0 +1,10 @@
+import base64
+
+
+def test_set_key(cronos):
+    cli = cronos.cosmos_cli()
+    key = base64.b64encode(b"new_key").decode("utf-8")
+    cli.set_e2ee_key(key, _from="community")
+    adr = cli.address("community")
+    p = cli.query_e2ee_key(adr)
+    assert p["key"] == key

x/e2ee/autocli.go

@@ -18,5 +18,18 @@ func (am AppModule) AutoCLIOptions() *autocliv1.ModuleOptions {
 				},
 			},
 		},
+		Tx: &autocliv1.ServiceCommandDescriptor{
+			Service: "e2ee.Msg",
+			RpcCommandOptions: []*autocliv1.RpcCommandOptions{
+				{
+					RpcMethod: "RegisterEncryptionKey",
+					Use:       "set-encryption-key [key]",
+					Short:     "Set encryption key is stored associated with the user address.",
+					PositionalArgs: []*autocliv1.PositionalArgDescriptor{
+						{ProtoField: "key"},
+					},
+				},
+			},
+		},
 	}
 }