feat: working on app tests

EKOKE-DAO · Nov 10, 2023 · 43e1da1 · 43e1da1
1 parent f44bdaf
commit 43e1da1
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/dilazionato/src/app.rs b/src/dilazionato/src/app.rs
@@ -563,8 +563,17 @@ mod test {
 
     use pretty_assertions::assert_eq;
 
+    #[test]
+    fn test_should_get_tx() {
+        assert!(SellContract::transaction(Nat::from(1)).is_err());
+        let id = TxHistory::register_token_mint(&mock_token(1, 1));
+        assert!(SellContract::transaction(Nat::from(id)).is_ok());
+    }
+
     #[test]
     fn test_should_get_total_transactions() {
+        assert_eq!(SellContract::total_transactions(), Nat::from(0));
         let _ = TxHistory::register_token_mint(&mock_token(1, 1));
+        assert_eq!(SellContract::total_transactions(), Nat::from(1));
     }
 }
diff --git a/src/dilazionato/src/app/storage/tx_history.rs b/src/dilazionato/src/app/storage/tx_history.rs
@@ -18,7 +18,7 @@ impl TxHistory {
     }
 
     /// Register a token mint
-    pub fn register_token_mint(token: &Token) {
+    pub fn register_token_mint(token: &Token) -> Nat {
         let event = TxEvent {
             caller: crate::utils::caller(),
             details: vec![
@@ -44,8 +44,10 @@ impl TxHistory {
         };
         let id = Self::next_id();
         with_tx_history_mut(|tx_history| {
-            tx_history.insert(id.into(), event.into());
+            tx_history.insert(id.clone().into(), event.into());
         });
+
+        id
     }
 
     pub fn register_token_burn(token: &Token) -> Nat {