async def test_new_session_error(self, mock_error, mock_aioresponse): # noqa: F811
"""Test new_session with error."""
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
mock_aioresponse.post(
f"{self.base_url}/dyn/login.json", payload={"err": "dummy-error"}
)
with pytest.raises(SmaAuthenticationException):
await sma.new_session()
assert mock_error.call_count == 1
mock_aioresponse.post(f"{self.base_url}/dyn/login.json", payload={"err": 404})
with pytest.raises(SmaAuthenticationException):
await sma.new_session()
assert mock_error.call_count == 2
# Temporary update the URL to have https
sma._url = f"https://{self.host}"
mock_aioresponse.post(
f"https://{self.host}/dyn/login.json", payload={"err": 404}
)
with pytest.raises(SmaAuthenticationException):
await sma.new_session()
sma._url = f"http://{self.host}"
assert mock_error.call_count == 3
mock_aioresponse.post(f"{self.base_url}/dyn/login.json", payload={"err": 503})
with pytest.raises(SmaAuthenticationException):
await sma.new_session()
assert mock_error.call_count == 4
async def test_read_body_error(self, mock_warn, mock_aioresponse): # noqa: F811
"""Test read_body with SmaReadException."""
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD", payload={"err": 401}
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
sma._sid = "ABCD"
with pytest.raises(SmaReadException):
await sma._read_body("/dyn/getValues.json", payload={"dummy": "payload"})
assert mock_warn.call_count == 1
async def test_device_info_fallback(self, mock_aioresponse): # noqa: F811
"""Test device_info fallback."""
mock_aioresponse.post(
f"{self.base_url}/dyn/login.json", payload={"result": {"sid": "ABCD"}}
)
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": {
"6800_08822B00": {
"1": [{"validVals": [461], "val": [{"tag": 461}]}]
},
"6800_10821E00": {"1": [{"val": MOCK_DEVICE["name"]}]},
}
}
},
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
assert await sma.new_session()
result = await sma.device_info()
assert result
assert result["manufacturer"] == "SMA"
assert result["name"] == MOCK_DEVICE["name"]
assert result["type"] == ""
assert result["serial"] == "9999999999"
assert result["sw_version"] == ""
async def test_read_no_password(self, mock_warn, mock_aioresponse): # noqa: F811
"""Test read_body without password."""
mock_aioresponse.post(
f"{self.base_url}/dyn/getDashValues.json",
payload={
"result": {
"0199-xxxxx385": {
"6800_08822000": {
"1": [
{
"validVals": [9401, 9402, 9403, 9404, 9405],
"val": [{"tag": 9402}],
}
]
},
}
}
},
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host)
sensors = Sensors(device_type_sensor)
assert await sma.read(sensors)
assert sensors["6800_08822000"].value == "Sunny Boy 3.6"
assert mock_warn.call_count == 0
async def test_new_session_fail(self, mock_aioresponse): # noqa: F811
"""Test new_session with empty result."""
mock_aioresponse.post(f"{self.base_url}/dyn/login.json", payload={"result": {}})
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
with pytest.raises(SmaAuthenticationException):
await sma.new_session()
async def test_read_body_unexpected(
self,
mock_warn,
mock_aioresponse # noqa: F811
):
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={"result": {
"0199-xxxxx385": {},
"0199-yyyyy385": {},
}},
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
sma.sma_sid = "ABCD"
result_body = await sma._read_body("/dyn/getValues.json",
payload={"dummy": "payload"})
assert result_body == {}
assert mock_warn.call_count == 1
async def test_request_json_connect_error(self, mock_aioresponse): # noqa: F811
"""Test request_json with a SmaConnectionException."""
mock_aioresponse.get(
f"{self.base_url}/dummy-url",
exception=aiohttp.client_exceptions.ClientConnectionError("mocked error"),
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
with pytest.raises(SmaConnectionException):
await sma._get_json("/dummy-url")
async def test_timeout_error(self, mock_aioresponse): # noqa: F811
"""Test request_json with a SmaConnectionException from TimeoutError."""
mock_aioresponse.get(
f"{self.base_url}/dummy-url",
exception=asyncio.TimeoutError("mocked error"),
repeat=True,
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
with pytest.raises(SmaConnectionException):
await sma._get_json("/dummy-url")
async def test_request_json_server_disconnect_error(
self, mock_aioresponse # noqa: F811
):
"""Test request_json with a SmaConnectionException from ServerDisconnectedError."""
mock_aioresponse.get(
f"{self.base_url}/dummy-url",
exception=aiohttp.client_exceptions.ServerDisconnectedError("mocked error"),
repeat=True,
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
with pytest.raises(SmaConnectionException):
await sma._get_json("/dummy-url")
async def test_device_info_fail(self, mock_aioresponse): # noqa: F811
"""Test device_info with SmaReadException."""
mock_aioresponse.post(
f"{self.base_url}/dyn/login.json", payload={"result": {"sid": "ABCD"}}
)
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={},
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
assert await sma.new_session()
with pytest.raises(SmaReadException):
await sma.device_info()
async def test_new_session(self, mock_warn, mock_aioresponse): # noqa: F811
"""Test new_session."""
mock_aioresponse.post(
f"{self.base_url}/dyn/login.json", payload={"result": {"sid": "ABCD"}}
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "extralongpassword")
assert await sma.new_session()
assert mock_warn.call_count == 1
await sma.close_session()
await sma.close_session()
assert mock_warn.call_count == 1
async def test_new_session_error(self, mock_aioresponse): # noqa: F811
"""Test new_session with error."""
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
mock_aioresponse.post(
f"{self.base_url}/dyn/login.json", payload={"err": "dummy-error"}
)
with pytest.raises(SmaAuthenticationException):
await sma.new_session()
mock_aioresponse.post(f"{self.base_url}/dyn/login.json", payload={"err": 503})
with pytest.raises(SmaAuthenticationException):
await sma.new_session()
async def test_get_sensors_empty_result(self, mock_aioresponse): # noqa: F811
"""Test get_sensors with empty result."""
mock_aioresponse.post(
f"{self.base_url}/dyn/login.json", payload={"result": {"sid": "ABCD"}}
)
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={"result": {}},
repeat=True,
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
assert len(await sma.get_sensors()) == len(sensor_map[DEVCLASS_INVERTER])
async def test_request_json_invalid_json(
self, mock_warn, mock_aioresponse # noqa: F811
):
"""Test request_json with invalid json."""
mock_aioresponse.get(
f"{self.base_url}/dummy-url",
body="THIS IS NOT A VALID JSON",
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
json = await sma._get_json("/dummy-url")
assert isinstance(json, dict)
assert json == {}
assert mock_warn.call_count == 1
async def test_read_logger_error(self, mock_aioresponse): # noqa: F811
"""Test read_logger with SmaReadException."""
mock_aioresponse.post(
f"{self.base_url}/dyn/login.json", payload={"result": {"sid": "ABCD"}}
)
mock_aioresponse.post(
f"{self.base_url}/dyn/getLogger.json?sid=ABCD",
payload={"result": {"0199-xxxxx385": "NOT A LIST"}},
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
with pytest.raises(SmaReadException):
await sma.read_logger(28704, 1622592000, 1622491200)
async def test_get_sensors_no_result_body(self, mock_aioresponse): # noqa: F811
"""Test get_sensors with no result body."""
mock_aioresponse.post(
f"{self.base_url}/dyn/login.json", payload={"result": {"sid": "ABCD"}}
)
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={},
repeat=True,
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
with pytest.raises(SmaReadException):
await sma.get_sensors()
async def test_get_sensors(self, mock_aioresponse): # noqa: F811
"""Test get_sensors."""
mock_aioresponse.post(f"{self.base_url}/dyn/login.json",
payload={"result": {
"sid": "ABCD"
}})
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": {
"6180_08214800": {
"1": [{
"val": [{
"tag": "123"
}]
}],
},
"6800_008AA300": {
"1": [{
"val": "0123456"
}],
},
"6800_10852600": {
"1": [
{
"val": "1"
},
{
"val": "2"
},
{
"val": None
},
]
},
}
}
},
repeat=True,
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
assert len(await sma.get_sensors()) == (
len(sensor_map[DEVCLASS_INVERTER]) +
len(sensor_map[ENERGY_METER_VIA_INVERTER]) +
(len(sensor_map[OPTIMIZERS_VIA_INVERTER]) * 2))
async def test_device_info(self, mock_aioresponse): # noqa: F811
"""Test device_info."""
mock_aioresponse.post(f"{self.base_url}/dyn/login.json",
payload={"result": {
"sid": "ABCD"
}})
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": {
"6800_08822B00": {
"1": [{
"validVals": [461],
"val": [{
"tag": 461
}]
}]
},
"6800_00A21E00": {
"1": [{
"low": 0,
"high": None,
"val": MOCK_DEVICE["serial"]
}]
},
"6800_08822000": {
"1": [{
"validVals": [9401, 9402, 9403, 9404, 9405],
"val": [{
"tag": 9402
}],
}]
},
"6800_10821E00": {
"1": [{
"val": MOCK_DEVICE["name"]
}]
},
}
}
},
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
result = await sma.device_info()
assert result
assert result == MOCK_DEVICE
async def test_read_dash_logger(self, mock_aioresponse): # noqa: F811
"""Test read_dash_logger."""
mock_aioresponse.post(
f"{self.base_url}/dyn/getDashLogger.json",
payload={
"result": {
"0199-xxxxx385": {
"7000": {
"1": [
{"t": 1622569500, "v": 4565239},
{"t": 1622569800, "v": 4565249},
]
},
"7020": {
"1": [
{"t": 1622498400, "v": 4542749},
{"t": 1622584800, "v": 4565355},
]
},
}
}
},
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host)
read_dash_logger = await sma.read_dash_logger()
assert read_dash_logger == {
"7000": {
"1": [
{"t": 1622569500, "v": 4565239},
{"t": 1622569800, "v": 4565249},
]
},
"7020": {
"1": [
{"t": 1622498400, "v": 4542749},
{"t": 1622584800, "v": 4565355},
]
},
}
async def test_post_json(self):
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
with patch("pysma.SMA._request_json") as mock_request_json:
await sma._post_json("dummy_url")
mock_request_json.assert_called_once_with(
"POST",
"dummy_url",
data="{}",
headers={"content-type": "application/json"},
)
with patch("pysma.SMA._request_json") as mock_request_json:
await sma._post_json("dummy_url", {"data": "dummy"})
mock_request_json.assert_called_once_with(
"POST",
"dummy_url",
data='{"data": "dummy"}',
headers={"content-type": "application/json"},
)
async def test_read_logger(self, mock_aioresponse): # noqa: F811
"""Test read_logger."""
mock_aioresponse.post(f"{self.base_url}/dyn/login.json",
payload={"result": {
"sid": "ABCD"
}})
mock_aioresponse.post(
f"{self.base_url}/dyn/getLogger.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": [
{
"t": 1622498400,
"v": 4542749
},
{
"t": 1622584800,
"v": 4565355
},
]
}
},
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
read_logger = await sma.read_logger(28704, 1622592000, 1622491200)
assert read_logger == [
{
"t": 1622498400,
"v": 4542749
},
{
"t": 1622584800,
"v": 4565355
},
]
async def test_get_devclass(self, mock_aioresponse): # noqa: F811
"""Test get_devclass."""
mock_aioresponse.post(
f"{self.base_url}/dyn/login.json",
payload={"result": {
"sid": "ABCD"
}},
)
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={"result": {
"0199-xxxxx385": {}
}},
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
assert await sma.get_devclass() is None
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": {
"6180_08214800": {
"1": [{
"val": [{
"tag": "123"
}]
}],
},
}
}
},
)
assert await sma.get_devclass() == DEVCLASS_INVERTER
sma.devclass = None
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={"result": {
"0199-xxxxx385": {
"6180_08214800": {},
}
}},
)
assert await sma.get_devclass() is None
sma.devclass = None
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": {
"6180_08214800": {
"val": "402",
},
}
}
},
)
assert await sma.get_devclass() is None
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": {
"6800_12345678": {
"1": [{
"val": "value1"
}],
"7": [{
"val": "value2"
}],
},
}
}
},
)
with pytest.raises(KeyError):
await sma.get_devclass()
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": {
"6180_08214800": {
"1": [{
"val": [{
"tag": "123"
}]
}],
},
}
}
},
)
sma.devclass = None
assert await sma.get_devclass("BOGUS_BODY") == DEVCLASS_INVERTER
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": {
"6180_08214800": {
"1": [{
"val": [{
"tag": "123"
}]
}],
},
}
}
},
)
assert await sma.get_devclass({}) == DEVCLASS_INVERTER
sma.devclass = "test"
assert await sma.get_devclass() == "test"
async def test_new_session_invalid_group(self,
mock_aioresponse): # noqa: F811
"""Test new_session with invalid group."""
session = aiohttp.ClientSession()
with pytest.raises(KeyError):
SMA(session, self.host, "pass", "invalid-group")
async def test_get_devclass(self, mock_aioresponse): # noqa: F811
"""Test get_devclass."""
mock_aioresponse.post(
f"{self.base_url}/dyn/login.json",
payload={"result": {"sid": "ABCD"}},
)
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={"result": {"0199-xxxxx385": {}}},
)
session = aiohttp.ClientSession()
sma = SMA(session, self.host, "pass")
assert await sma.get_devclass() is None
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": {
"6180_08214800": {
"7": [{"val": [{"tag": "123"}]}],
},
}
}
},
)
assert await sma.get_devclass() == DEVCLASS_BATTERY
sma._devclass = None
result_body = {
"6180_08214800": {},
}
assert await sma.get_devclass(result_body) is None
sma._devclass = None
result_body = {
"6180_08214800": {
"val": [{"tag": "123"}],
},
}
assert await sma.get_devclass(result_body) == DEVCLASS_INVERTER
sma._devclass = None
result_body = {
"6800_00823400": {"val": 33751296, "min": 0, "max": 4294967294},
"6800_10821E00": {"val": "SN: 1930023194"},
"6800_00A21E00": {"val": 1930023194, "min": 0, "max": 4294967294},
"6800_08822000": {
"val": [{"tag": 9301}],
"validVals": [9301, 9303, 9302],
},
}
assert await sma.get_devclass(result_body) == DEVCLASS_INVERTER
sma._devclass = None
result_body = {
"6800_12345678": {
"1": [{"val": "value1"}],
"7": [{"val": "value2"}],
},
}
with pytest.raises(KeyError):
await sma.get_devclass(result_body)
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": {
"6180_08214800": {
"1": [{"val": [{"tag": "123"}]}],
},
}
}
},
)
sma._devclass = None
assert await sma.get_devclass("BOGUS_BODY") == DEVCLASS_INVERTER
mock_aioresponse.post(
f"{self.base_url}/dyn/getValues.json?sid=ABCD",
payload={
"result": {
"0199-xxxxx385": {
"6180_08214800": {
"1": [{"val": [{"tag": "123"}]}],
},
}
}
},
)
assert await sma.get_devclass({}) == DEVCLASS_INVERTER
sma._devclass = "test"
assert await sma.get_devclass() == "test"
