From f210b74790d11a8d42b027689a4c8a9fdcbae0b1 Mon Sep 17 00:00:00 2001
From: jb101010-2 <168106462+jb101010-2@users.noreply.github.com>
Date: Mon, 9 Dec 2024 23:50:04 +0100
Subject: [PATCH] Suez_water: close session after config flow (#132714)

---
 .../components/suez_water/config_flow.py      | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/homeassistant/components/suez_water/config_flow.py b/homeassistant/components/suez_water/config_flow.py
index ac09cf4a1d3..2a1edea35f1 100644
--- a/homeassistant/components/suez_water/config_flow.py
+++ b/homeassistant/components/suez_water/config_flow.py
@@ -37,16 +37,19 @@ async def validate_input(data: dict[str, Any]) -> None:
             data[CONF_PASSWORD],
             counter_id,
         )
-        if not await client.check_credentials():
-            raise InvalidAuth
-    except PySuezError as ex:
-        raise CannotConnect from ex
-
-    if counter_id is None:
         try:
-            data[CONF_COUNTER_ID] = await client.find_counter()
+            if not await client.check_credentials():
+                raise InvalidAuth
         except PySuezError as ex:
-            raise CounterNotFound from ex
+            raise CannotConnect from ex
+
+        if counter_id is None:
+            try:
+                data[CONF_COUNTER_ID] = await client.find_counter()
+            except PySuezError as ex:
+                raise CounterNotFound from ex
+    finally:
+        await client.close_session()
 
 
 class SuezWaterConfigFlow(ConfigFlow, domain=DOMAIN):