From c53b13cf0db37e5e50393dda837ce5cf4f50aea7 Mon Sep 17 00:00:00 2001
From: gluax <16431709+gluax@users.noreply.github.com>
Date: Thu, 15 Aug 2024 08:18:19 -0700
Subject: [PATCH] refactor: is eligible query to be more future proof

---
 src/msgs/staking/query.rs       | 2 +-
 src/msgs/staking/query_tests.rs | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/msgs/staking/query.rs b/src/msgs/staking/query.rs
index ed8f45a..997dad0 100644
--- a/src/msgs/staking/query.rs
+++ b/src/msgs/staking/query.rs
@@ -12,7 +12,7 @@ pub enum QueryMsg {
     #[cfg_attr(feature = "cosmwasm", returns(StakerAndSeq))]
     GetStakerAndSeq { public_key: String },
     #[cfg_attr(feature = "cosmwasm", returns(bool))]
-    IsExecutorEligible { public_key: String },
+    IsExecutorEligible { proof: String, dr_id: String },
     #[cfg_attr(feature = "cosmwasm", returns(StakingConfig))]
     GetStakingConfig {},
 }
diff --git a/src/msgs/staking/query_tests.rs b/src/msgs/staking/query_tests.rs
index 8b28bca..b94a1a8 100644
--- a/src/msgs/staking/query_tests.rs
+++ b/src/msgs/staking/query_tests.rs
@@ -49,11 +49,13 @@ fn json_get_staker_and_seq() {
 fn json_is_executor_eligible() {
     let expected_json = json!({
     "is_executor_eligible": {
-      "public_key": "public_key"
+      "proof": "public_key",
+      "dr_id": "dr_id"
     }
     });
     let msg: QueryMsg = StakingQueryMsg::IsExecutorEligible {
-        public_key: "public_key".to_string(),
+        proof: "public_key".to_string(),
+        dr_id: "dr_id".to_string(),
     }
     .into();
     assert_json_ok(msg, expected_json);