From a3332410dccc023a321ab4914ede3bf985a9171d Mon Sep 17 00:00:00 2001
From: Kevin Hellemun <17928966+OGKevin@users.noreply.github.com>
Date: Mon, 1 Nov 2021 00:29:57 +0100
Subject: [PATCH] Add ROCKROBO_E2 to supported vacuums for xiaomi_miio (#58817)

https://github.com/rytilahti/python-miio/blob/e1adea55f3be237f6e6904210b6f7b52162bf154/miio/vacuum.py#L129
---
 homeassistant/components/xiaomi_miio/const.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/homeassistant/components/xiaomi_miio/const.py b/homeassistant/components/xiaomi_miio/const.py
index 6630def38ef..578a6d3ffab 100644
--- a/homeassistant/components/xiaomi_miio/const.py
+++ b/homeassistant/components/xiaomi_miio/const.py
@@ -199,8 +199,10 @@ MODELS_LIGHT = (
 # TODO: use const from pythonmiio once new release with the constant has been published. # pylint: disable=fixme
 ROCKROBO_S4 = "roborock.vacuum.s4"
 ROCKROBO_S5_MAX = "roborock.vacuum.s5e"
+ROCKROBO_E2 = "roborock.vacuum.e2"
 MODELS_VACUUM = [
     ROCKROBO_V1,
+    ROCKROBO_E2,
     ROCKROBO_S4,
     ROCKROBO_S5,
     ROCKROBO_S5_MAX,
@@ -209,6 +211,7 @@ MODELS_VACUUM = [
     ROCKROBO_S7,
 ]
 MODELS_VACUUM_WITH_MOP = [
+    ROCKROBO_E2,
     ROCKROBO_S5,
     ROCKROBO_S5_MAX,
     ROCKROBO_S6,