From 2c708d8b1994da2c44066461416f97e3dac7f343 Mon Sep 17 00:00:00 2001 From: Cesar E Garza Date: Wed, 23 Oct 2024 00:31:54 -0500 Subject: [PATCH] fixed test and formatting --- src/splatnet3_scraper/auth/nso.py | 4 ++-- tests/auth/test_nso.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/splatnet3_scraper/auth/nso.py b/src/splatnet3_scraper/auth/nso.py index 066746f..fd1f9cf 100644 --- a/src/splatnet3_scraper/auth/nso.py +++ b/src/splatnet3_scraper/auth/nso.py @@ -334,7 +334,7 @@ def parse_npf_uri(self, uri: str) -> str: str: The session token code. This is *NOT* the session token, but is used to obtain the session token. """ - return uri.split("&")[0][uri.find("=")+1:] + return uri.split("&")[0][uri.find("=") + 1 :] def get_session_token(self, session_token_code: str) -> str: """Obtains the session token from the session token code. @@ -365,7 +365,7 @@ def get_session_token(self, session_token_code: str) -> str: params = { "client_id": "71b963c1b7b6d119", "session_token_code": session_token_code, - "session_token_code_verifier": self.verifier.replace(b'=', b''), + "session_token_code_verifier": self.verifier.replace(b"=", b""), } uri = "https://accounts.nintendo.com/connect/1.0.0/api/session_token" response = self.session.post(uri, headers=header, data=params) diff --git a/tests/auth/test_nso.py b/tests/auth/test_nso.py index 6f0fed9..89a6295 100644 --- a/tests/auth/test_nso.py +++ b/tests/auth/test_nso.py @@ -227,7 +227,7 @@ def mock_get(*args, **kwargs): return MockResponse(200, json={"session_token": "test"}) monkeypatch.setattr(requests.Session, "post", mock_get) - nso = self.get_new_nso(version="5.0.0", verifier="test_verifier") + nso = self.get_new_nso(version="5.0.0", verifier=b"test_verifier") assert nso._session_token is None session_token = nso.get_session_token("session_code") assert session_token == "test"