* Directly call async_write_ha_state * Address comments * Fix tests
* Add initial tests * Adjust test * Typo