Merge pull request #63 from iprak/handle-annotations

Handle compatibility with python 3.8

custom_components/yahoofinance/__init__.py

@@ -4,6 +4,8 @@
 https://github.com/iprak/yahoofinance
 """
 
+from __future__ import annotations
+
 from datetime import timedelta
 import logging
 from typing import Final, Union

custom_components/yahoofinance/const.py

@@ -1,5 +1,7 @@
 """Constants for Yahoo Finance sensor."""
 
+from __future__ import annotations
+
 from typing import Final
 
 # Additional attributes exposed by the sensor

custom_components/yahoofinance/coordinator.py

@@ -3,6 +3,7 @@
 
 https://github.com/iprak/yahoofinance
 """
+from __future__ import annotations
 
 from datetime import timedelta
 import logging

custom_components/yahoofinance/sensor.py

@@ -4,6 +4,8 @@
 https://github.com/iprak/yahoofinance
 """
 
+from __future__ import annotations
+
 import logging
 from timeit import default_timer as timer
 from typing import Union

tests/test_init.py

@@ -168,12 +168,14 @@ async def test_setup_optionally_requests_coordinator_refresh(
 async def test_refresh_symbols_service(hass, enable_custom_integrations):
     """Test refresh_symbols service callback."""
 
+    # Mock the refresh callback `_async_update` for testing
     with patch(
-        f"{YSUC}.async_request_refresh", AsyncMock(return_value=None)
+        f"{YSUC}._async_update", AsyncMock(return_value=None)
     ) as mock_async_request_refresh:
 
         assert await async_setup_component(hass, DOMAIN, SAMPLE_CONFIG) is True
         await hass.async_block_till_done()
+        assert mock_async_request_refresh.call_count == 1
 
         await hass.services.async_call(
             DOMAIN,
@@ -183,7 +185,7 @@ async def test_refresh_symbols_service(hass, enable_custom_integrations):
         )
         await hass.async_block_till_done()
 
-        assert mock_async_request_refresh.call_count == 1
+        assert mock_async_request_refresh.call_count == 2
 
 
 def test_SymbolDefinition_comparison():