async def test_read_body_error(self, mock_warn, mock_aioresponse): # noqa: F811 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.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 ): 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