mirror of
https://github.com/python-kasa/python-kasa.git
synced 2025-01-23 13:17:06 +00:00
82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
|
import errno
|
||
|
|
import json
|
||
|
|
import logging
|
||
|
|
import secrets
|
||
|
|
import struct
|
||
|
|
import sys
|
||
|
|
import time
|
||
|
|
from contextlib import nullcontext as does_not_raise
|
||
|
|
from itertools import chain
|
||
|
|
|
||
|
|
import httpx
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from ..aestransport import AesTransport
|
||
|
|
from ..credentials import Credentials
|
||
|
|
from ..exceptions import (
|
||
|
|
SMART_RETRYABLE_ERRORS,
|
||
|
|
SMART_TIMEOUT_ERRORS,
|
||
|
|
SmartDeviceException,
|
||
|
|
SmartErrorCode,
|
||
|
|
)
|
||
|
|
from ..iotprotocol import IotProtocol
|
||
|
|
from ..klaptransport import KlapEncryptionSession, KlapTransport, _sha256
|
||
|
|
from ..smartprotocol import SmartProtocol
|
||
|
|
|
||
|
|
DUMMY_QUERY = {"foobar": {"foo": "bar", "bar": "foo"}}
|
||
|
|
ERRORS = [e for e in SmartErrorCode if e != 0]
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("error_code", ERRORS, ids=lambda e: e.name)
|
||
|
|
async def test_smart_device_errors(mocker, error_code):
|
||
|
|
host = "127.0.0.1"
|
||
|
|
mock_response = {"result": {"great": "success"}, "error_code": error_code.value}
|
||
|
|
|
||
|
|
mocker.patch.object(AesTransport, "perform_handshake")
|
||
|
|
mocker.patch.object(AesTransport, "perform_login")
|
||
|
|
|
||
|
|
send_mock = mocker.patch.object(AesTransport, "send", return_value=mock_response)
|
||
|
|
|
||
|
|
protocol = SmartProtocol(host, transport=AesTransport(host))
|
||
|
|
with pytest.raises(SmartDeviceException):
|
||
|
|
await protocol.query(DUMMY_QUERY, retry_count=2)
|
||
|
|
|
||
|
|
if error_code in chain(SMART_TIMEOUT_ERRORS, SMART_RETRYABLE_ERRORS):
|
||
|
|
expected_calls = 3
|
||
|
|
else:
|
||
|
|
expected_calls = 1
|
||
|
|
assert send_mock.call_count == expected_calls
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("error_code", ERRORS, ids=lambda e: e.name)
|
||
|
|
async def test_smart_device_errors_in_multiple_request(mocker, error_code):
|
||
|
|
host = "127.0.0.1"
|
||
|
|
mock_response = {
|
||
|
|
"result": {
|
||
|
|
"responses": [
|
||
|
|
{"method": "foobar1", "result": {"great": "success"}, "error_code": 0},
|
||
|
|
{
|
||
|
|
"method": "foobar2",
|
||
|
|
"result": {"great": "success"},
|
||
|
|
"error_code": error_code.value,
|
||
|
|
},
|
||
|
|
{"method": "foobar3", "result": {"great": "success"}, "error_code": 0},
|
||
|
|
]
|
||
|
|
},
|
||
|
|
"error_code": 0,
|
||
|
|
}
|
||
|
|
|
||
|
|
mocker.patch.object(AesTransport, "perform_handshake")
|
||
|
|
mocker.patch.object(AesTransport, "perform_login")
|
||
|
|
|
||
|
|
send_mock = mocker.patch.object(AesTransport, "send", return_value=mock_response)
|
||
|
|
|
||
|
|
protocol = SmartProtocol(host, transport=AesTransport(host))
|
||
|
|
with pytest.raises(SmartDeviceException):
|
||
|
|
await protocol.query(DUMMY_QUERY, retry_count=2)
|
||
|
|
if error_code in chain(SMART_TIMEOUT_ERRORS, SMART_RETRYABLE_ERRORS):
|
||
|
|
expected_calls = 3
|
||
|
|
else:
|
||
|
|
expected_calls = 1
|
||
|
|
assert send_mock.call_count == expected_calls
|