From 1ce8bfdaa438949da707d94ff7b12ff7b20ce0cc Mon Sep 17 00:00:00 2001
From: Josef Zweck <24647999+zweckj@users.noreply.github.com>
Date: Thu, 14 Nov 2024 16:34:17 +0100
Subject: [PATCH] Use test helpers for acaia buttons (#130626)

---
 .../acaia/snapshots/test_button.ambr          | 60 +++++++++----------
 tests/components/acaia/test_button.py         | 33 ++++++----
 2 files changed, 50 insertions(+), 43 deletions(-)

diff --git a/tests/components/acaia/snapshots/test_button.ambr b/tests/components/acaia/snapshots/test_button.ambr
index 7e2624923af..cd91ca1a17a 100644
--- a/tests/components/acaia/snapshots/test_button.ambr
+++ b/tests/components/acaia/snapshots/test_button.ambr
@@ -1,5 +1,5 @@
 # serializer version: 1
-# name: test_buttons[entry_button_reset_timer]
+# name: test_buttons[button.lunar_ddeeff_reset_timer-entry]
   EntityRegistryEntrySnapshot({
     'aliases': set({
     }),
@@ -32,7 +32,20 @@
     'unit_of_measurement': None,
   })
 # ---
-# name: test_buttons[entry_button_start_stop_timer]
+# name: test_buttons[button.lunar_ddeeff_reset_timer-state]
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'friendly_name': 'LUNAR-DDEEFF Reset timer',
+    }),
+    'context': <ANY>,
+    'entity_id': 'button.lunar_ddeeff_reset_timer',
+    'last_changed': <ANY>,
+    'last_reported': <ANY>,
+    'last_updated': <ANY>,
+    'state': 'unknown',
+  })
+# ---
+# name: test_buttons[button.lunar_ddeeff_start_stop_timer-entry]
   EntityRegistryEntrySnapshot({
     'aliases': set({
     }),
@@ -65,7 +78,20 @@
     'unit_of_measurement': None,
   })
 # ---
-# name: test_buttons[entry_button_tare]
+# name: test_buttons[button.lunar_ddeeff_start_stop_timer-state]
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'friendly_name': 'LUNAR-DDEEFF Start/stop timer',
+    }),
+    'context': <ANY>,
+    'entity_id': 'button.lunar_ddeeff_start_stop_timer',
+    'last_changed': <ANY>,
+    'last_reported': <ANY>,
+    'last_updated': <ANY>,
+    'state': 'unknown',
+  })
+# ---
+# name: test_buttons[button.lunar_ddeeff_tare-entry]
   EntityRegistryEntrySnapshot({
     'aliases': set({
     }),
@@ -98,33 +124,7 @@
     'unit_of_measurement': None,
   })
 # ---
-# name: test_buttons[state_button_reset_timer]
-  StateSnapshot({
-    'attributes': ReadOnlyDict({
-      'friendly_name': 'LUNAR-DDEEFF Reset timer',
-    }),
-    'context': <ANY>,
-    'entity_id': 'button.lunar_ddeeff_reset_timer',
-    'last_changed': <ANY>,
-    'last_reported': <ANY>,
-    'last_updated': <ANY>,
-    'state': 'unknown',
-  })
-# ---
-# name: test_buttons[state_button_start_stop_timer]
-  StateSnapshot({
-    'attributes': ReadOnlyDict({
-      'friendly_name': 'LUNAR-DDEEFF Start/stop timer',
-    }),
-    'context': <ANY>,
-    'entity_id': 'button.lunar_ddeeff_start_stop_timer',
-    'last_changed': <ANY>,
-    'last_reported': <ANY>,
-    'last_updated': <ANY>,
-    'state': 'unknown',
-  })
-# ---
-# name: test_buttons[state_button_tare]
+# name: test_buttons[button.lunar_ddeeff_tare-state]
   StateSnapshot({
     'attributes': ReadOnlyDict({
       'friendly_name': 'LUNAR-DDEEFF Tare',
diff --git a/tests/components/acaia/test_button.py b/tests/components/acaia/test_button.py
index 62eb8b61b8a..f68f85e253d 100644
--- a/tests/components/acaia/test_button.py
+++ b/tests/components/acaia/test_button.py
@@ -1,21 +1,24 @@
 """Tests for the acaia buttons."""
 
 from datetime import timedelta
-from unittest.mock import MagicMock
+from unittest.mock import MagicMock, patch
 
 from freezegun.api import FrozenDateTimeFactory
-import pytest
 from syrupy import SnapshotAssertion
 
 from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN, SERVICE_PRESS
-from homeassistant.const import ATTR_ENTITY_ID, STATE_UNAVAILABLE, STATE_UNKNOWN
+from homeassistant.const import (
+    ATTR_ENTITY_ID,
+    STATE_UNAVAILABLE,
+    STATE_UNKNOWN,
+    Platform,
+)
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import entity_registry as er
 
-from tests.common import async_fire_time_changed
-
-pytestmark = pytest.mark.usefixtures("init_integration")
+from . import setup_integration
 
+from tests.common import MockConfigEntry, async_fire_time_changed, snapshot_platform
 
 BUTTONS = (
     "tare",
@@ -28,24 +31,25 @@ async def test_buttons(
     hass: HomeAssistant,
     entity_registry: er.EntityRegistry,
     snapshot: SnapshotAssertion,
+    mock_scale: MagicMock,
+    mock_config_entry: MockConfigEntry,
 ) -> None:
     """Test the acaia buttons."""
-    for button in BUTTONS:
-        state = hass.states.get(f"button.lunar_ddeeff_{button}")
-        assert state
-        assert state == snapshot(name=f"state_button_{button}")
 
-        entry = entity_registry.async_get(state.entity_id)
-        assert entry
-        assert entry == snapshot(name=f"entry_button_{button}")
+    with patch("homeassistant.components.acaia.PLATFORMS", [Platform.BUTTON]):
+        await setup_integration(hass, mock_config_entry)
+    await snapshot_platform(hass, entity_registry, snapshot, mock_config_entry.entry_id)
 
 
 async def test_button_presses(
     hass: HomeAssistant,
     mock_scale: MagicMock,
+    mock_config_entry: MockConfigEntry,
 ) -> None:
     """Test the acaia button presses."""
 
+    await setup_integration(hass, mock_config_entry)
+
     for button in BUTTONS:
         await hass.services.async_call(
             BUTTON_DOMAIN,
@@ -63,10 +67,13 @@ async def test_button_presses(
 async def test_buttons_unavailable_on_disconnected_scale(
     hass: HomeAssistant,
     mock_scale: MagicMock,
+    mock_config_entry: MockConfigEntry,
     freezer: FrozenDateTimeFactory,
 ) -> None:
     """Test the acaia buttons are unavailable when the scale is disconnected."""
 
+    await setup_integration(hass, mock_config_entry)
+
     for button in BUTTONS:
         state = hass.states.get(f"button.lunar_ddeeff_{button}")
         assert state