fix: clean up amazon images (#986)

* fix: clean up amazon images * fix test
moralmunky · Nov 1, 2024 · 843b410 · 843b410
1 parent 2964177
commit 843b410
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/custom_components/mail_and_packages/helpers.py b/custom_components/mail_and_packages/helpers.py
@@ -944,6 +944,8 @@ def get_count(
 
     # Return Amazon delivered info
     if sensor_type == AMAZON_DELIVERED:
+        _LOGGER.debug("Cleaning up amazon images...")
+        cleanup_images(f"{image_path}amazon/")
         result[ATTR_COUNT] = amazon_search(
             account, image_path, hass, amazon_image_name, amazon_domain
         )

diff --git a/tests/test_helpers.py b/tests/test_helpers.py
@@ -213,7 +213,6 @@ async def test_process_emails_bad(hass, mock_imap_no_email, mock_update):
         data=FAKE_CONFIG_DATA_BAD,
     )
 
-    entry.version = 2
     entry.add_to_hass(hass)
     assert await hass.config_entries.async_setup(entry.entry_id)
     await hass.async_block_till_done()