Skip to content

Commit

Permalink
cases
Browse files Browse the repository at this point in the history
  • Loading branch information
dni committed Apr 23, 2024
1 parent ed5b9c9 commit b5963c1
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 7 deletions.
4 changes: 2 additions & 2 deletions lnurl/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from .exceptions import InvalidLnurl, InvalidUrl, LnurlResponseException
from .helpers import _url_encode
from .models import LnurlAuthResponse, LnurlResponse, LnurlResponseModel
from .types import ClearnetUrl, DebugUrl, Lnaddress, Lnurl, OnionUrl
from .types import ClearnetUrl, DebugUrl, LnAddress, Lnurl, OnionUrl


def decode(bech32_lnurl: str) -> Union[OnionUrl, ClearnetUrl, DebugUrl]:
Expand Down Expand Up @@ -42,7 +42,7 @@ def handle(
) -> LnurlResponseModel:
try:
if "@" in bech32_lnurl:
lnaddress = Lnaddress(bech32_lnurl)
lnaddress = LnAddress(bech32_lnurl)
return get(lnaddress.url, response_class=response_class, verify=verify)
lnurl = Lnurl(bech32_lnurl)
except (ValidationError, ValueError):
Expand Down
4 changes: 2 additions & 2 deletions lnurl/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -204,10 +204,10 @@ def is_login(self) -> bool:
return "tag" in self.url.query_params and self.url.query_params["tag"] == "login"


class Lnaddress(ReprMixin, str):
class LnAddress(ReprMixin, str):
"""Lightning address of form `user@host`"""

def __new__(cls, address: str, **_) -> "Lnaddress":
def __new__(cls, address: str, **_) -> "LnAddress":
return str.__new__(cls, address)

def __init__(self, address: str):
Expand Down
6 changes: 3 additions & 3 deletions tests/test_types.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
DebugUrl,
LightningInvoice,
LightningNodeUri,
Lnaddress,
LnAddress,
Lnurl,
LnurlPayMetadata,
OnionUrl,
Expand Down Expand Up @@ -194,7 +194,7 @@ def test_invalid_data(self, metadata):
],
)
def test_valid_lnaddress(self, lnaddress):
lnaddress = Lnaddress(lnaddress)
lnaddress = LnAddress(lnaddress)
assert isinstance(lnaddress.url, (OnionUrl, DebugUrl, ClearnetUrl))

@pytest.mark.parametrize(
Expand All @@ -205,4 +205,4 @@ def test_valid_lnaddress(self, lnaddress):
)
def test_invalid_lnaddress(self, lnaddress):
with pytest.raises(ValueError):
lnaddress = Lnaddress(lnaddress)
lnaddress = LnAddress(lnaddress)

0 comments on commit b5963c1

Please sign in to comment.