From e856ba11d038dda82cf51b679cacd230194ae987 Mon Sep 17 00:00:00 2001
From: Josef Zweck <24647999+zweckj@users.noreply.github.com>
Date: Sat, 23 Nov 2024 14:34:03 +0100
Subject: [PATCH] Bump pylamarzocco to 1.2.11 (#131331)

Co-authored-by: Marc Mueller <30130371+cdce8p@users.noreply.github.com>
---
 homeassistant/components/lamarzocco/coordinator.py | 3 ++-
 homeassistant/components/lamarzocco/manifest.json  | 2 +-
 requirements_all.txt                               | 2 +-
 requirements_test_all.txt                          | 2 +-
 tests/components/lamarzocco/test_init.py           | 3 ++-
 5 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/homeassistant/components/lamarzocco/coordinator.py b/homeassistant/components/lamarzocco/coordinator.py
index 646aad0e8dd..d7539fd9ca4 100644
--- a/homeassistant/components/lamarzocco/coordinator.py
+++ b/homeassistant/components/lamarzocco/coordinator.py
@@ -13,6 +13,7 @@ from pylamarzocco.client_cloud import LaMarzoccoCloudClient
 from pylamarzocco.client_local import LaMarzoccoLocalClient
 from pylamarzocco.exceptions import AuthFail, RequestNotSuccessful
 from pylamarzocco.lm_machine import LaMarzoccoMachine
+from websockets.protocol import State
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_MODEL, CONF_NAME, EVENT_HOMEASSISTANT_STOP
@@ -85,7 +86,7 @@ class LaMarzoccoUpdateCoordinator(DataUpdateCoordinator[None]):
                 if (
                     self._local_client is not None
                     and self._local_client.websocket is not None
-                    and self._local_client.websocket.open
+                    and self._local_client.websocket.state is State.OPEN
                 ):
                     self._local_client.terminating = True
                     await self._local_client.websocket.close()
diff --git a/homeassistant/components/lamarzocco/manifest.json b/homeassistant/components/lamarzocco/manifest.json
index 8b78a92feca..4aef30a5c26 100644
--- a/homeassistant/components/lamarzocco/manifest.json
+++ b/homeassistant/components/lamarzocco/manifest.json
@@ -33,5 +33,5 @@
   "integration_type": "device",
   "iot_class": "cloud_polling",
   "loggers": ["pylamarzocco"],
-  "requirements": ["pylamarzocco==1.2.7"]
+  "requirements": ["pylamarzocco==1.2.11"]
 }
diff --git a/requirements_all.txt b/requirements_all.txt
index 2d92dca771e..c9c17f5b6b8 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -2027,7 +2027,7 @@ pykwb==0.0.8
 pylacrosse==0.4
 
 # homeassistant.components.lamarzocco
-pylamarzocco==1.2.7
+pylamarzocco==1.2.11
 
 # homeassistant.components.lastfm
 pylast==5.1.0
diff --git a/requirements_test_all.txt b/requirements_test_all.txt
index 4e42a1e8761..943c919429a 100644
--- a/requirements_test_all.txt
+++ b/requirements_test_all.txt
@@ -1632,7 +1632,7 @@ pykrakenapi==0.1.8
 pykulersky==0.5.2
 
 # homeassistant.components.lamarzocco
-pylamarzocco==1.2.7
+pylamarzocco==1.2.11
 
 # homeassistant.components.lastfm
 pylast==5.1.0
diff --git a/tests/components/lamarzocco/test_init.py b/tests/components/lamarzocco/test_init.py
index b99077a9059..5ef0eca13af 100644
--- a/tests/components/lamarzocco/test_init.py
+++ b/tests/components/lamarzocco/test_init.py
@@ -5,6 +5,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 from pylamarzocco.const import FirmwareType
 from pylamarzocco.exceptions import AuthFail, RequestNotSuccessful
 import pytest
+from websockets.protocol import State
 
 from homeassistant.components.lamarzocco.config_flow import CONF_MACHINE
 from homeassistant.components.lamarzocco.const import DOMAIN
@@ -182,7 +183,7 @@ async def test_websocket_closed_on_unload(
     ) as local_client:
         client = local_client.return_value
         client.websocket = AsyncMock()
-        client.websocket.connected = True
+        client.websocket.state = State.OPEN
         await async_init_integration(hass, mock_config_entry)
         hass.bus.async_fire(EVENT_HOMEASSISTANT_STOP)
         await hass.async_block_till_done()