feat: allow rpc mock to insert multiple same requests by migrating from Hashmap to Vec

Author: Nagaprasadvr

rpc-client/src/mock_sender.rs

@@ -41,7 +41,40 @@ use {
 
 pub const PUBKEY: &str = "7RoSF9fUmdphVCpabEoefH81WwrW7orsWonXWqTXkKV8";
 
-pub type Mocks = HashMap<RpcRequest, Value>;
+#[derive(Default)]
+pub struct Mocks(Vec<(RpcRequest, Value)>);
+
+impl Mocks {
+    pub fn new() -> Self {
+        Self(Vec::new())
+    }
+
+    pub fn from<const N: usize>(mocks: [(RpcRequest, Value); N]) -> Self {
+        Self(mocks.to_vec())
+    }
+
+    pub fn insert(&mut self, request: RpcRequest, response: Value) {
+        self.0.push((request, response));
+    }
+
+    pub fn remove(&mut self, request: &RpcRequest) -> Option<Value> {
+        self.0
+            .iter()
+            .position(|(r, _)| r == request)
+            .map(|index| self.0.remove(index).1)
+    }
+}
+
+impl FromIterator<(RpcRequest, Value)> for Mocks {
+    fn from_iter<T: IntoIterator<Item = (RpcRequest, Value)>>(iter: T) -> Self {
+        let mut mocks = Mocks::new();
+        for (request, response) in iter {
+            mocks.insert(request, response);
+        }
+        mocks
+    }
+}
+
 pub struct MockSender {
     mocks: RwLock<Mocks>,
     url: String,

rpc-client/src/nonblocking/rpc_client.rs

@@ -4707,7 +4707,7 @@ pub(crate) fn parse_keyed_accounts(
 
 #[doc(hidden)]
 pub fn create_rpc_client_mocks() -> crate::mock_sender::Mocks {
-    let mut mocks = std::collections::HashMap::new();
+    let mut mocks = crate::mock_sender::Mocks::default();
 
     let get_account_request = RpcRequest::GetAccountInfo;
     let get_account_response = serde_json::to_value(Response {

rpc-client/src/rpc_client.rs

@@ -3677,7 +3677,7 @@ impl RpcClient {
 /// Mocks for documentation examples
 #[doc(hidden)]
 pub fn create_rpc_client_mocks() -> crate::mock_sender::Mocks {
-    let mut mocks = std::collections::HashMap::new();
+    let mut mocks = crate::mock_sender::Mocks::default();
 
     let get_account_request = RpcRequest::GetAccountInfo;
     let get_account_response = serde_json::to_value(Response {
@@ -3980,7 +3980,7 @@ mod tests {
 
         // Test: with context
         {
-            let mocks: Mocks = [(
+            let mocks: Mocks = Mocks::from([(
                 RpcRequest::GetProgramAccounts,
                 serde_json::to_value(OptionalContext::Context(Response {
                     context: RpcResponseContext {
@@ -3990,9 +3990,7 @@ mod tests {
                     value: vec![keyed_account],
                 }))
                 .unwrap(),
-            )]
-            .into_iter()
-            .collect();
+            )]);
             let rpc_client = RpcClient::new_mock_with_mocks("mock_client".to_string(), mocks);
             let result = rpc_client
                 .get_program_accounts_with_config(