From 350fd09dcefcc7fc27f2a6c69392e48a1f90e5ff Mon Sep 17 00:00:00 2001
From: matograine <tom.ngr@zaclys.net>
Date: Tue, 15 Dec 2020 18:20:31 +0100
Subject: [PATCH] [test] #363 : make patched_get_source compute and return
 balance

---
 tests/patched/money.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/patched/money.py b/tests/patched/money.py
index 9592fe66..8e52e77a 100644
--- a/tests/patched/money.py
+++ b/tests/patched/money.py
@@ -63,6 +63,7 @@ async def patched_get_sources(pubkey):
             )
             balance += amount_in_current_base(listinput[-1])
             a += 1
+        return balance
 
     def listinput_TX(listinput, balance, max_tx):
         a = 0
@@ -78,6 +79,7 @@ async def patched_get_sources(pubkey):
             )
             balance += amount_in_current_base(listinput[-1])
             a += 1
+        return balance
 
     listinput, n = list(), 0
     balance = 0
@@ -123,8 +125,8 @@ async def patched_get_sources(pubkey):
         max_ud = 0
         max_tx = 0
 
-    listinput_UD(listinput, balance, pubkey, max_ud)
-    listinput_TX(listinput, balance, max_tx)
+    balance = listinput_UD(listinput, balance, pubkey, max_ud)
+    balance = listinput_TX(listinput, balance, max_tx)
 
     patched_get_sources.counter += 1
     return listinput, balance
-- 
GitLab