tests: updated tests

saskenuba · saskenuba · commit d7d56bf126a6 · 2024-03-01T01:50:02.000-03:00
Signed-off-by: Martin &lt;martin@hotmail.com.br&gt;
diff --git a/src/domain/account.rs b/src/domain/account.rs
@@ -15,7 +15,7 @@ impl Account {
     fn generate(balance: i32, limit: u32) -> Account {
         Account {
             id: -1,
-            transactions: vec![Transaction::generate(balance)],
+            transactions: vec![Transaction::generate(balance, None)],
             credit_limit: limit,
         }
     }
@@ -48,16 +48,17 @@ mod tests {
     fn failure_more_funds() {
         let account = Account::generate(1_000, 10_000);
         let with_new_transaction = account
-            .add_transaction(Transaction::generate(-1000))
+            .add_transaction(Transaction::generate(-1000, None))
             .unwrap();
         assert_eq!(with_new_transaction.balance(), 0);
 
         let with_new_transaction = with_new_transaction
-            .add_transaction(Transaction::generate(-10_000))
+            .add_transaction(Transaction::generate(-10_000, None))
             .unwrap();
         assert_eq!(with_new_transaction.balance(), -10_000);
 
-        let with_new_transaction = with_new_transaction.add_transaction(Transaction::generate(-1));
+        let with_new_transaction =
+            with_new_transaction.add_transaction(Transaction::generate(-1, None));
         match with_new_transaction {
             Err(AccountError::InsufficientCredit) => {}
             _ => unreachable!(),
diff --git a/src/domain/transaction.rs b/src/domain/transaction.rs
@@ -34,14 +34,17 @@ impl TransactionDescription {
 
 impl Transaction {
     #[cfg(test)]
-    pub fn generate(amount: i32) -> Self {
+    pub fn generate<T>(amount: i32, description: T) -> Self
+    where
+        T: Into<Option<&'static str>>,
+    {
         let kind = if amount.is_negative() {
             TransactionKind::Debit
         } else {
             TransactionKind::Credit
         };
 
-        let description = TransactionDescription::new("xxx");
+        let description = TransactionDescription::new(description.into().unwrap_or("xxx"));
         Self {
             valor: NonZeroI32::new(amount).expect("Safe to unwrap"),
             tipo: kind,
