[Tests] Recovery new testcase (automation for #1011) (#1017)

* added few getters for jormungandr process and starter
* added LedgerSnapthot struct to hold comparable data. introduced some helper method to avoid duplication. added new test case

Author: dkijania

jormungandr-integration-tests/src/common/jormungandr/process.rs

@@ -54,6 +54,10 @@ impl JormungandrProcess {
     pub fn rest_address(&self) -> String {
         self.config.get_node_address()
     }
+
+    pub fn config(&self) -> JormungandrConfig {
+        self.config.clone()
+    }
 }
 
 impl Drop for JormungandrProcess {

jormungandr-integration-tests/src/common/jormungandr/starter.rs

@@ -139,6 +139,11 @@ impl Starter {
         self
     }
 
+    pub fn role(&mut self, role: Role) -> &mut Self {
+        self.role = role;
+        self
+    }
+
     pub fn verify_by(&mut self, verification_mode: StartupVerificationMode) -> &mut Self {
         self.verification_mode = verification_mode;
         self
@@ -258,12 +263,13 @@ impl Starter {
     }
 }
 
-pub fn restart_jormungandr_node_as_leader(process: JormungandrProcess) -> JormungandrProcess {
+pub fn restart_jormungandr_node(process: JormungandrProcess, role: Role) -> JormungandrProcess {
     let config = process.config.clone();
     std::mem::drop(process);
 
     Starter::new()
         .config(config)
+        .role(role)
         .start()
         .expect("Jormungandr restart failed")
 }

jormungandr-integration-tests/src/jormungandr/recovery.rs

@@ -1,11 +1,73 @@
 use crate::common::{
     configuration::genesis_model::Fund,
+    data::address::{Account, AddressDataProvider, Utxo},
     jcli_wrapper,
     jcli_wrapper::jcli_transaction_wrapper::JCLITransactionWrapper,
-    jormungandr::{starter::restart_jormungandr_node_as_leader, ConfigurationBuilder, Starter},
+    jormungandr::{
+        starter::restart_jormungandr_node, ConfigurationBuilder, JormungandrProcess, Role, Starter,
+    },
     startup,
 };
 
+use jormungandr_lib::interfaces::{AccountState, SettingsDto, UTxOInfo};
+
+#[derive(Clone, Debug)]
+struct LedgerSnapshot {
+    settings: SettingsDto,
+    utxos: Vec<UTxOInfo>,
+    account_state: AccountState,
+}
+
+impl PartialEq for LedgerSnapshot {
+    fn eq(&self, other: &Self) -> bool {
+        self.settings == other.settings
+            && self.utxos == other.utxos
+            && self.account_state == other.account_state
+    }
+}
+impl Eq for LedgerSnapshot {}
+
+impl LedgerSnapshot {
+    pub fn new(settings: SettingsDto, utxos: Vec<UTxOInfo>, account_state: AccountState) -> Self {
+        LedgerSnapshot {
+            settings,
+            utxos,
+            account_state,
+        }
+    }
+}
+
+fn take_snapshot(account_receiver: &Account, jormungandr: &JormungandrProcess) -> LedgerSnapshot {
+    let settings = jcli_wrapper::assert_get_rest_settings(&jormungandr.rest_address());
+    let utxos = jcli_wrapper::assert_rest_utxo_get(&jormungandr.rest_address());
+    let account = jcli_wrapper::assert_rest_account_get_stats(
+        &account_receiver.address,
+        &jormungandr.rest_address(),
+    );
+
+    LedgerSnapshot::new(settings, utxos, account)
+}
+
+pub fn do_simple_transaction<T: AddressDataProvider>(
+    sender: &T,
+    account_receiver: &Account,
+    utxo_receiver: &Utxo,
+    jormungandr: &JormungandrProcess,
+) {
+    let config = jormungandr.config();
+    let utxo = startup::get_utxo_for_address(sender, &jormungandr.rest_address());
+
+    let transaction_message = JCLITransactionWrapper::new_transaction(&config.genesis_block_hash)
+        .assert_add_input_from_utxo(&utxo)
+        .assert_add_output(&account_receiver.address, &50.into())
+        .assert_add_output(&utxo_receiver.address, &50.into())
+        .assert_finalize()
+        .seal_with_witness_for_address(sender)
+        .assert_to_message();
+
+    jcli_wrapper::assert_transaction_in_block(&transaction_message, &jormungandr.rest_address());
+}
+
 #[test]
 pub fn test_node_recovers_from_node_restart() {
     let sender = startup::create_new_utxo_address();
@@ -21,47 +83,41 @@ pub fn test_node_recovers_from_node_restart() {
 
     let jormungandr = Starter::new().config(config.clone()).start().unwrap();
 
-    let utxo = startup::get_utxo_for_address(&sender, &jormungandr.rest_address());
+    do_simple_transaction(&sender, &account_receiver, &utxo_receiver, &jormungandr);
+    let snapshot_before = take_snapshot(&account_receiver, &jormungandr);
+    let jormungandr = restart_jormungandr_node(jormungandr, Role::Leader);
+    let snapshot_after = take_snapshot(&account_receiver, &jormungandr);
 
-    let transaction_message = JCLITransactionWrapper::new_transaction(&config.genesis_block_hash)
-        .assert_add_input_from_utxo(&utxo)
-        .assert_add_output(&account_receiver.address, &50.into())
-        .assert_add_output(&utxo_receiver.address, &50.into())
-        .assert_finalize()
-        .seal_with_witness_for_address(&sender)
-        .assert_to_message();
+    assert_eq!(
+        snapshot_before, snapshot_after,
+        "Different snaphot after restart {:?} vs {:?}",
+        snapshot_before, snapshot_after
+    );
+}
 
-    jcli_wrapper::assert_transaction_in_block(&transaction_message, &jormungandr.rest_address());
+#[test]
+pub fn test_node_recovers_kill_signal() {
+    let sender = startup::create_new_utxo_address();
+    let account_receiver = startup::create_new_account_address();
+    let utxo_receiver = startup::create_new_utxo_address();
 
-    let expected_settings = jcli_wrapper::assert_get_rest_settings(&jormungandr.rest_address());
-    let expected_utxos = jcli_wrapper::assert_rest_utxo_get(&jormungandr.rest_address());
-    let expected_account_state = jcli_wrapper::assert_rest_account_get_stats(
-        &account_receiver.address,
-        &jormungandr.rest_address(),
-    );
+    let config = ConfigurationBuilder::new()
+        .with_funds(vec![Fund {
+            address: sender.address.clone(),
+            value: 100.into(),
+        }])
+        .build();
 
-    let jormungandr = restart_jormungandr_node_as_leader(jormungandr);
+    let jormungandr = Starter::new().config(config.clone()).start().unwrap();
 
-    let actual_settings = jcli_wrapper::assert_get_rest_settings(&jormungandr.rest_address());
-    let actual_utxos = jcli_wrapper::assert_rest_utxo_get(&jormungandr.rest_address());
-    let actual_account_state = jcli_wrapper::assert_rest_account_get_stats(
-        &account_receiver.address,
-        &jormungandr.rest_address(),
-    );
+    do_simple_transaction(&sender, &account_receiver, &utxo_receiver, &jormungandr);
+    let snapshot_before = take_snapshot(&account_receiver, &jormungandr);
+    let jormungandr = restart_jormungandr_node(jormungandr, Role::Passive);
+    let snapshot_after = take_snapshot(&account_receiver, &jormungandr);
 
     assert_eq!(
-        actual_settings, expected_settings,
-        "Different setting after restart {:?} vs {:?}",
-        actual_settings, expected_settings
-    );
-    assert_eq!(
-        actual_utxos, expected_utxos,
-        "Different utxos after restart {:?} vs {:?}",
-        actual_utxos, expected_utxos
-    );
-    assert_eq!(
-        actual_account_state, expected_account_state,
-        "Different account state after restart {:?} vs {:?}",
-        actual_account_state, expected_account_state
+        snapshot_before, snapshot_after,
+        "Different snaphot after restart {:?} vs {:?}",
+        snapshot_before, snapshot_after
     );
 }