heierbtc/lnbits
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix up decorators for tests

Browse Source
This commit is contained in:
dni 2022-07-20 12:21:39 +02:00
parent 463fbb39bd
commit 0e393111de
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lnbits/decorators.py
View File

@ -1,6 +1,6 @@
from http import HTTPStatus from http import HTTPStatus
from typing import Optional from typing import Union
from cerberus import Validator # type: ignore from cerberus import Validator # type: ignore
from fastapi import status from fastapi import status
@ -102,9 +102,9 @@ class WalletAdminKeyChecker(KeyChecker):
class WalletTypeInfo: class WalletTypeInfo:
wallet_type: int wallet_type: int
wallet: Wallet wallet: Union[Wallet, None]
def __init__(self, wallet_type: int, wallet: Wallet) -> None: def __init__(self, wallet_type: int, wallet: Union[Wallet, None]) -> None:
self.wallet_type = wallet_type self.wallet_type = wallet_type
self.wallet = wallet self.wallet = wallet
@ -171,7 +171,7 @@ async def get_key_type(
if e.status_code == HTTPStatus.BAD_REQUEST: if e.status_code == HTTPStatus.BAD_REQUEST:
raise raise
if e.status_code == HTTPStatus.UNAUTHORIZED: if e.status_code == HTTPStatus.UNAUTHORIZED:
return WalletTypeInfo(2, Wallet()) return WalletTypeInfo(2, None)
except: except:
raise raise
return wallet return wallet