vacuums: update tests for IDLE state addition.

tests/devices/test_kyvol_e30_vacuum.py

@@ -4,6 +4,7 @@ from homeassistant.components.vacuum import (
     STATE_CLEANING,
     STATE_DOCKED,
     STATE_ERROR,
+    STATE_IDLE,
     STATE_RETURNING,
     VacuumEntityFeature,
 )
@@ -161,7 +162,7 @@ class TestKyvolE30Vacuum(MultiButtonTests, MultiSensorTests, TuyaDeviceTestCase)
         self.dps[COMMAND_DPS] = "return_to_base"
         self.assertEqual(self.subject.state, STATE_RETURNING)
         self.dps[COMMAND_DPS] = "standby"
-        self.assertEqual(self.subject.state, STATE_DOCKED)
+        self.assertEqual(self.subject.state, STATE_IDLE)
         self.dps[COMMAND_DPS] = "random"
         self.assertEqual(self.subject.state, STATE_CLEANING)
         self.dps[POWER_DPS] = False

tests/devices/test_lefant_m213_vacuum.py

@@ -3,6 +3,7 @@ from homeassistant.components.vacuum import (
     STATE_CLEANING,
     STATE_DOCKED,
     STATE_ERROR,
+    STATE_IDLE,
     STATE_RETURNING,
     VacuumEntityFeature,
 )
@@ -113,7 +114,7 @@ class TestLefantM213Vacuum(MultiSensorTests, TuyaDeviceTestCase):
         self.dps[STATUS_DPS] = "4"
         self.assertEqual(self.subject.state, STATE_RETURNING)
         self.dps[STATUS_DPS] = "7"
-        self.assertEqual(self.subject.state, STATE_DOCKED)
+        self.assertEqual(self.subject.state, STATE_IDLE)
         self.dps[STATUS_DPS] = "6"
         self.assertEqual(self.subject.state, STATE_CLEANING)
         self.dps[POWER_DPS] = False