Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 14 additions & 14 deletions tests/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,13 +11,13 @@
DSM_AUTH_OTP_AUTHENTICATE_FAILED,
DEVICE_TOKEN,
)
from .const_dsm_6 import (
DSM_6_LOGIN,
DSM_6_LOGIN_2SA,
DSM_6_LOGIN_2SA_OTP,
DSM_6_INFORMATION,
DSM_6_UTILIZATION,
DSM_6_STORAGE,
from .api_data.dsm_6 import (
DSM_6_AUTH_LOGIN,
DSM_6_AUTH_LOGIN_2SA,
DSM_6_AUTH_LOGIN_2SA_OTP,
DSM_6_DSM_INFORMATION,
DSM_6_CORE_UTILIZATION,
DSM_6_STORAGE_STORAGE,
)

VALID_DSM_HOST = "nas.mywebsite.me"
Expand Down Expand Up @@ -64,30 +64,30 @@ def _execute_get_url(self, request_url):
if self.LOGIN_URI in request_url:
if VALID_USER_2SA in request_url and VALID_PASSWORD in request_url:
if "otp_code" not in request_url and "device_id" not in request_url:
return DSM_6_LOGIN_2SA
return DSM_6_AUTH_LOGIN_2SA

if "device_id" in request_url and DEVICE_TOKEN in request_url:
return DSM_6_LOGIN
return DSM_6_AUTH_LOGIN

if "otp_code" in request_url:
if VALID_OTP in request_url:
return DSM_6_LOGIN_2SA_OTP
return DSM_6_AUTH_LOGIN_2SA_OTP
return DSM_AUTH_OTP_AUTHENTICATE_FAILED

if VALID_USER in request_url and VALID_PASSWORD in request_url:
return DSM_6_LOGIN
return DSM_6_AUTH_LOGIN

if self.API_URI in request_url:
if not self._session_id or not self._syno_token:
return DSM_INSUFFICIENT_USER_PRIVILEGE

if SynoDSMInformation.API_KEY in request_url:
return DSM_6_INFORMATION
return DSM_6_DSM_INFORMATION

if SynoCoreUtilization.API_KEY in request_url:
return DSM_6_UTILIZATION
return DSM_6_CORE_UTILIZATION

if SynoStorage.API_KEY in request_url:
return DSM_6_STORAGE
return DSM_6_STORAGE_STORAGE

return None
1 change: 1 addition & 0 deletions tests/api_data/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
"""APIs raw data constants."""
10 changes: 10 additions & 0 deletions tests/api_data/dsm_5/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
"""DSM 5 datas."""
from .const_5_api_info import DSM_5_API_INFO
from .const_5_api_auth import (
DSM_5_AUTH_LOGIN,
DSM_5_AUTH_LOGIN_2SA,
DSM_5_AUTH_LOGIN_2SA_OTP,
)
from .core.const_5_core_utilization import DSM_5_CORE_UTILIZATION
from .dsm.const_5_dsm_info import DSM_5_DSM_INFORMATION
from .storage.const_5_storage_storage import DSM_5_STORAGE_STORAGE
12 changes: 12 additions & 0 deletions tests/api_data/dsm_5/const_5_api_auth.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
"""DSM 5 SYNO.API.Auth data."""
from tests.const import ( # pylint: disable=unused-import
SESSION_ID,
DEVICE_TOKEN,
SYNO_TOKEN,
DSM_AUTH_OTP_NOT_SPECIFIED,
)


DSM_5_AUTH_LOGIN = {}
DSM_5_AUTH_LOGIN_2SA = {}
DSM_5_AUTH_LOGIN_2SA_OTP = {}
Loading