handle relayer peg client initialization state

relayer/src/relayer.rs

@@ -1,4 +1,5 @@
 use bitcoincore_rpc::{Auth, Client, Error, RpcApi};
+use nomic_client::{Client as PegClient, ClientError as PegClientError};
 use nomic_primitives::transaction::Transaction;
 use std::env;
 
@@ -61,13 +62,15 @@ impl RelayerState {
 pub struct RelayerStateMachine {
     pub state: RelayerState,
     rpc: Option<Client>,
+    peg_client: Option<PegClient>,
 }
 
 impl RelayerStateMachine {
     pub fn new() -> Self {
         RelayerStateMachine {
             state: RelayerState::InitializeBitcoinRpc,
             rpc: None,
+            peg_client: None,
         }
     }
 
@@ -85,6 +88,16 @@ impl RelayerStateMachine {
                     Err(_) => InitializeBitcoinRpcFailure,
                 }
             }
+            InitializePegClient => {
+                let peg_client = PegClient::new();
+                match peg_client {
+                    Ok(peg_client) => {
+                        self.peg_client = Some(peg_client);
+                        InitializePegClientSuccess
+                    }
+                    Err(_) => InitializePegClientFailure,
+                }
+            }
             _ => panic!("Relayer is in an unhandled state"),
         }
     }
@@ -109,8 +122,10 @@ mod tests {
     #[test]
     fn run_relayer_state_machine() {
         let mut sm = RelayerStateMachine::new();
-        let event = sm.run();
-        sm.state = sm.state.next(event);
-        println!("sm state: {:?}", sm.state);
+        for _ in 0..2 {
+            let event = sm.run();
+            sm.state = sm.state.next(event);
+            println!("sm state: {:?}", sm.state);
+        }
     }
 }