Pytest-homeassistant-custom-component

Out of frustration in figuring out how to write tests for custom components, I have created a, hopefully, easy to use library. It aims to enable all the fixtures and helpers that exist in homeassistant/core. It is semi-automatically updated from the latest dev branch in homeassistant/core, with the goal of being fully automatic in the future.

I hope others may find it useful, and that it leads to less frustrating dev experience for custom component developers.

For all those HACS components, you will need to add an __init__.py file in your custom_components folder.

10 Likes

Very nice, I had created something similar (https://github.com/boralyl/pytest-homeassistant) by pulling out fixtures, but itā€™s not automatically updated at all and has broken before w/ new releases. Iā€™ll have to dig into how you went about this Iā€™m not terribly happy with my solution.

Hereā€™s a post I made about unit testing custom components: https://aarongodfrey.dev/home%20automation/building_a_home_assistant_custom_component_part_2/

1 Like

To be very clear, I did see your package, and considered contributing there, but I couldnā€™t tell how you were deciding to choose which pieces to incorporate. Iā€™d be very willing to accept contributions if you see any improvements that could be made.

Everything ā€˜happensā€™ in the generate_package.py file and the GitHub actions.

I also added an acknowledgement to your package in the README.

Iā€™d love to use this, but Iā€™m not sure how to leverage it and would love your thoughts.

Trying to create tests for https://github.com/moralmunky/Home-Assistant-Mail-And-Packages

fahrā€™s blog post gives a decent overview. You pip install pytest-homeassistant-custom-component or pip install pytest-homeassistant, if you want to use fahrā€™s version. I would suggest doing this in a virtual environment.

Then you write your tests just like in the home-assistant/core repository in a tests. Then run the tests with pytest. There is an example custom component and tests in my repository as well.

1 Like

I took a quick look through the repository you linked. I would probably start by mocking the update_method of the EmailData class and using MockConfigEntry to set up your integration to bypass the config flow logic. Then you would have a test for only the workings of the sensor entity. Then you could mock out the actual IO and test the logic of the EmailData class only.

You could also start from the other end by mocking out any IO in the config flow and test that your config flow works the way it should form by form.

2 Likes

Honestly, I just pulled out the fixtures that I ended up using in my custom components, and ignored the rest. I really like your implementation since itā€™s automated and probably less likely to run into issues with upgrades.

Ya not sure how to actually do that :frowning:

This I have done.

and lost again :frowning:

https://docs.python.org/3/library/unittest.mock.html

I suggest using the patch function in your test. Youā€™ll need to put an __init__.py in the custom_components folder too. This a rabbit hole to fall in, but you can then make changes to the custom component more confidently.

Something like this:

async def test_some_stuff(hass):

with patch("custom_components.myintegration.sensor.EmailData.update") as mock_update:
    # do some stuff with a mocked method

1 Like

ah ok that makes sense, Iā€™m actually migrating all the functions and such out of the sensor.py into __init__.py and keeping only the sensor bits in there.

Thanks for the pointer!

Sorry to be a bother, I have this most of the way done, but Iā€™m getting stuck and would like to pick your brain if youā€™re game.

I am getting the following error (during pytest):

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.7.9/x64/lib/python3.7/site-packages/homeassistant/helpers/entity_platform.py", line 193, in _async_setup_platform
    await asyncio.shield(task)
  File "/home/runner/work/Home-Assistant-Mail-And-Packages/Home-Assistant-Mail-And-Packages/custom_components/mail_and_packages/sensor.py", line 70, in async_setup_entry
    data = EmailData(hass, config)
  File "/home/runner/work/Home-Assistant-Mail-And-Packages/Home-Assistant-Mail-And-Packages/custom_components/mail_and_packages/sensor.py", line 101, in __init__
    self.update = Throttle(self._scan_interval)(self.update)
  File "/opt/hostedtoolcache/Python/3.7.9/x64/lib/python3.7/site-packages/homeassistant/util/__init__.py", line 206, in __call__
    and "." not in method.__qualname__.split(".<locals>.")[-1]
  File "/opt/hostedtoolcache/Python/3.7.9/x64/lib/python3.7/unittest/mock.py", line 601, in __getattr__
    raise AttributeError(name)
AttributeError: __qualname__

Seems it doesnā€™t like the self.update's Throttle?
(I am switching this out to use the update coordinator in a yet to be released version.)

It looks like you are trying to dynamically update the fetch interval?

If you are using DataUpdateCoordinator, there is no need to use Throttle explicitly. The coordinator class has a built in debouncer. If the default debouncer time doesnā€™t work for you, you can adjust it. The coordinator also makes it easy to dynamically update the fetch interval.

Iā€™m not sure to your actual question. Your method doesnā€™t have __qualname__ for sure, but Iā€™m not sure what this means.

Ya I think Iā€™ll give it a try on my updated version thatā€™s using the coordinator see if that pans out.

Thanks again!

A note to any developers using this package. I fixed a bug in my GitHub actions that prevented the package to be uploaded to PyPI. The package was 1-2 versions behind homeassisant on PyPI because of this, but it is now fully up to date.

2 Likes

I have updated this library to handle the transition to fully drop Python 3.7. This has occurred in version 0.1.0. Additionally, the dependencies that were not directly related to pytest testing (the obvious ones anyway) were moved to a separate requirements_dev.txt. This enables developers to pin their own versions of those libraries, such as pylint.

1 Like

With recent changes in homeassistant causing some issues with the use of this package, Iā€™d like to solicit feedback on path(s) forward. Right now, one of the main pain points is linking a version of homeassistant in testing to the version of pytest-homeassistant-custom-component (phcc for short). phcc currently does not pin a homeassistant version in its dependencies, which will cause users of version 0.2.0 of phcc to get errors unless they pin homeassistant to versions prior to 2021.3.0.

  1. Pin the homeassistant requirement to currently dev or stable version of homeassistant at the time I generate the package. The downside to this is that it will break sometimes like recently when major changes were made to the testing that required the coexisting infrastructure in homeassistant package.

  2. Generate phcc package only for stable or dev releases. The upside is that components could test against specific home assistant versions. The downside is increased maintainability requirements to make sure they are in sync (this is a big downside to me), and not being able to proactively test against upcoming changes in dev.

  3. Pin the homeassistant requirement to a GitHub commit ref. This one is basically drop in to the current workflow and would install a version of homeassistant according to the exact commit used to extract phcc. This seems the best path forward, but Iā€™m unaware of any potential pitfalls of such an approach. Hence my solicitation of feedback before implementing.

I went with option 2, so that it is now generated based on the latest release by homeassistant/core, including beta releases. The latest version was generated based off of 2021.4.0b4.

The downside is that we wonā€™t be able to test based on bleeding edge changes, if needed (this didnā€™t really work in the previous state either). My plan is to introduce a cli to allow developers to reference a local homeassistant/core repo and generate off an arbitrary commit ref. This allows for the ability to get cutting edge testing if needed.

2 Likes

Hi,
Iā€™m just starting to write tests for my custom component and stumbled upon your addon and this post.

I just canā€™t figure out where to put the files and how to run the tests.

So far I use a ha-core vscode devcontainer in which I clone my repo and ln -s it to the custom-components folder.
Now I created a ā€œtestsā€ folder in the folder the repo is cloned to. But there I donā€™t have access to the imports-solving of the venv used by HA.

So first problem is, where to link the tests, so I can take advantage of the power of vscode.
Second is how to run the tests then.

Would be very happy to get some advice.