From 80c8d7579d271bba36c40b5f44201ea767e53878 Mon Sep 17 00:00:00 2001 From: Amine Khaldi Date: Wed, 10 Jan 2024 21:37:55 +0100 Subject: [PATCH] Annotate test_wallet_user_store.py (#17260) Annotate test_wallet_user_store.py. --- mypy-exclusions.txt | 1 - tests/wallet/test_wallet_user_store.py | 3 ++- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mypy-exclusions.txt b/mypy-exclusions.txt index a1b7761f49..9a9c9aee3b 100644 --- a/mypy-exclusions.txt +++ b/mypy-exclusions.txt @@ -117,6 +117,5 @@ tests.wallet.test_taproot tests.wallet.test_wallet_blockchain tests.wallet.test_wallet_interested_store tests.wallet.test_wallet_key_val_store -tests.wallet.test_wallet_user_store tools.analyze-chain tools.run_block diff --git a/tests/wallet/test_wallet_user_store.py b/tests/wallet/test_wallet_user_store.py index 6bb7dd03ab..abd8e11f5d 100644 --- a/tests/wallet/test_wallet_user_store.py +++ b/tests/wallet/test_wallet_user_store.py @@ -8,7 +8,7 @@ from tests.util.db_connection import DBConnection @pytest.mark.anyio -async def test_store(): +async def test_store() -> None: async with DBConnection(1) as db_wrapper: store = await WalletUserStore.create(db_wrapper) await store.init_wallet() @@ -17,6 +17,7 @@ async def test_store(): assert (await store.get_last_wallet()).id == i wallet = await store.create_wallet("CAT_WALLET", WalletType.CAT, "abc") assert wallet.id == i + 1 + assert wallet is not None assert wallet.id == 5 for i in range(2, 6):