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_read_body_unexpected( self, mock_warn, mock_aioresponse # noqa: F811 ): """Test read_body with unexpected body.""" 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._sid = "ABCD" result_body = await sma._read_body( "/dyn/getValues.json", payload={"dummy": "payload"} ) assert result_body == {} assert mock_warn.call_count == 1