remove statuses from tests

cashubtc · Nov 16, 2023 · cdbe3a4 · cdbe3a4
1 parent 29ff9b1
commit cdbe3a4
Show file tree

Hide file tree

Showing 9 changed files with 8 additions and 22 deletions.
diff --git a/cashu/mint/startup.py b/cashu/mint/startup.py
@@ -54,10 +54,8 @@ async def start_mint_init():
     status = await ledger.lightning.status()
     if status.error_message:
         logger.warning(
-            (
-                f"The backend for {ledger.lightning.__class__.__name__} isn't"
-                f" working properly: '{status.error_message}'"
-            ),
+            f"The backend for {ledger.lightning.__class__.__name__} isn't"
+            f" working properly: '{status.error_message}'",
             RuntimeWarning,
         )
     logger.info(f"Lightning balance: {status.balance_msat} msat")

diff --git a/cashu/wallet/api/router.py b/cashu/wallet/api/router.py
@@ -304,11 +304,9 @@ async def burn(
         wallet = await mint_wallet(mint)
     if not (all or token or force or delete) or (token and all):
         raise Exception(
-            (
-                "enter a token or use --all to burn all pending tokens, --force to"
-                " check all tokens or --delete with send ID to force-delete pending"
-                " token from list if mint is unavailable."
-            ),
+            "enter a token or use --all to burn all pending tokens, --force to"
+            " check all tokens or --delete with send ID to force-delete pending"
+            " token from list if mint is unavailable.",
         )
     if all:
         # check only those who are flagged as reserved
@@ -440,7 +438,6 @@ async def restore(
     await wallet.load_mint()
     await wallet.restore_promises_from_to(0, to)
     await wallet.invalidate(wallet.proofs)
-    wallet.status()
     return RestoreResponse(balance=wallet.available_balance)
 
 

diff --git a/cashu/wallet/cli/cli.py b/cashu/wallet/cli/cli.py
@@ -443,6 +443,7 @@ async def send_command(
         await send_nostr(
             wallet, amount=amount, pubkey=nostr or nopt, verbose=verbose, yes=yes
         )
+    print_balance(ctx)
 
 
 @cli.command("receive", help="Receive tokens.")
@@ -497,6 +498,8 @@ async def receive_cli(
                 await receive(wallet, tokenObj)
     else:
         print("Error: enter token or use either flag --nostr or --all.")
+        return
+    print_balance(ctx)
 
 
 @cli.command("burn", help="Burn spent tokens.")

diff --git a/tests/test_mint_operations.py b/tests/test_mint_operations.py
@@ -17,7 +17,6 @@ async def wallet1(mint):
         name="wallet1",
     )
     await wallet1.load_mint()
-    wallet1.status()
     yield wallet1
 
 

diff --git a/tests/test_wallet.py b/tests/test_wallet.py
@@ -55,7 +55,6 @@ async def wallet1(mint):
         name="wallet1",
     )
     await wallet1.load_mint()
-    wallet1.status()
     yield wallet1
 
 
@@ -67,7 +66,6 @@ async def wallet2(mint):
         name="wallet2",
     )
     await wallet2.load_mint()
-    wallet2.status()
     yield wallet2
 
 
@@ -85,7 +83,6 @@ async def wallet3(mint):
     await wallet3.db.execute("DELETE FROM proofs")
     await wallet3.db.execute("DELETE FROM proofs_used")
     await wallet3.load_mint()
-    wallet3.status()
     yield wallet3
 
 

diff --git a/tests/test_wallet_api.py b/tests/test_wallet_api.py
@@ -19,7 +19,6 @@ async def wallet(mint):
         name="wallet",
     )
     await wallet.load_mint()
-    wallet.status()
     yield wallet
 
 

diff --git a/tests/test_wallet_htlc.py b/tests/test_wallet_htlc.py
@@ -41,7 +41,6 @@ async def wallet1(mint):
     )
     await migrate_databases(wallet1.db, migrations)
     await wallet1.load_mint()
-    wallet1.status()
     yield wallet1
 
 
@@ -53,7 +52,6 @@ async def wallet2(mint):
     await migrate_databases(wallet2.db, migrations)
     wallet2.private_key = PrivateKey(secrets.token_bytes(32), raw=True)
     await wallet2.load_mint()
-    wallet2.status()
     yield wallet2
 
 

diff --git a/tests/test_wallet_p2pk.py b/tests/test_wallet_p2pk.py
@@ -42,7 +42,6 @@ async def wallet1(mint):
     )
     await migrate_databases(wallet1.db, migrations)
     await wallet1.load_mint()
-    wallet1.status()
     yield wallet1
 
 
@@ -54,7 +53,6 @@ async def wallet2(mint):
     await migrate_databases(wallet2.db, migrations)
     wallet2.private_key = PrivateKey(secrets.token_bytes(32), raw=True)
     await wallet2.load_mint()
-    wallet2.status()
     yield wallet2
 
 

diff --git a/tests/test_wallet_restore.py b/tests/test_wallet_restore.py
@@ -53,7 +53,6 @@ async def wallet1(mint):
         name="wallet1",
     )
     await wallet1.load_mint()
-    wallet1.status()
     yield wallet1
 
 
@@ -65,7 +64,6 @@ async def wallet2(mint):
         name="wallet2",
     )
     await wallet2.load_mint()
-    wallet2.status()
     yield wallet2
 
 
@@ -83,7 +81,6 @@ async def wallet3(mint):
     await wallet3.db.execute("DELETE FROM proofs")
     await wallet3.db.execute("DELETE FROM proofs_used")
     await wallet3.load_mint()
-    wallet3.status()
     yield wallet3