Rename MockToggleDevice to MockToggleEntity

tests/common.py

@@ -602,40 +602,40 @@ def __init__(
         )
 
 
-class MockToggleDevice(entity.ToggleEntity):
+class MockToggleEntity(entity.ToggleEntity):
     """Provide a mock toggle device."""
 
-    def __init__(self, name, state):
-        """Initialize the mock device."""
+    def __init__(self, name, state, unique_id=None):
+        """Initialize the mock entity."""
         self._name = name or DEVICE_DEFAULT_NAME
         self._state = state
         self.calls = []
 
     @property
     def name(self):
-        """Return the name of the device if any."""
+        """Return the name of the entity if any."""
         self.calls.append(("name", {}))
         return self._name
 
     @property
     def state(self):
-        """Return the name of the device if any."""
+        """Return the state of the entity if any."""
         self.calls.append(("state", {}))
         return self._state
 
     @property
     def is_on(self):
-        """Return true if device is on."""
+        """Return true if entity is on."""
         self.calls.append(("is_on", {}))
         return self._state == STATE_ON
 
     def turn_on(self, **kwargs):
-        """Turn the device on."""
+        """Turn the entity on."""
         self.calls.append(("turn_on", kwargs))
         self._state = STATE_ON
 
     def turn_off(self, **kwargs):
-        """Turn the device off."""
+        """Turn the entity off."""
         self.calls.append(("turn_off", kwargs))
         self._state = STATE_OFF