Improve smarla typing in tests (#162163)

tests/components/smarla/test_number.py

@@ -93,11 +93,15 @@ async def test_number_state_update(
 
     entity_id = entity_info["entity_id"]
 
-    assert hass.states.get(entity_id).state == "1.0"
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == "1.0"
 
     mock_number_property.get.return_value = 100
 
     await update_property_listeners(mock_number_property)
     await hass.async_block_till_done()
 
-    assert hass.states.get(entity_id).state == "100.0"
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == "100.0"

tests/components/smarla/test_sensor.py

@@ -75,11 +75,15 @@ async def test_sensor_state_update(
 
     entity_id = entity_info["entity_id"]
 
-    assert hass.states.get(entity_id).state == "0"
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == "0"
 
     mock_sensor_property.get.return_value = entity_info["test_value"]
 
     await update_property_listeners(mock_sensor_property)
     await hass.async_block_till_done()
 
-    assert hass.states.get(entity_id).state == "1"
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == "1"

tests/components/smarla/test_switch.py

@@ -104,11 +104,15 @@ async def test_switch_state_update(
 
     entity_id = entity_info["entity_id"]
 
-    assert hass.states.get(entity_id).state == STATE_OFF
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == STATE_OFF
 
     mock_switch_property.get.return_value = True
 
     await update_property_listeners(mock_switch_property)
     await hass.async_block_till_done()
 
-    assert hass.states.get(entity_id).state == STATE_ON
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == STATE_ON