An attempt remove the async set...tests are broken right now

Author: bookman-dius

custom_components/powersensor/AsyncSet.py

@@ -18,7 +18,6 @@
     async_dispatcher_send,
 )
 
-from .AsyncSet import AsyncSet
 from .const import (
     # Used config entry fields
     CFG_ROLES,
@@ -98,13 +97,13 @@ def __init__(
 
         self._monitor_add_plug_queue = None
         self._stop_task = False
-        self._plug_added_queue: AsyncSet = AsyncSet()
+        self._plug_added_queue: set = set()
         self._safe_to_process_plug_queue = False
 
     async def enqueue_plug_for_adding(self, network_info: dict):
         """On receiving zeroconf data this info is added to processing buffer to await creation of entity and api."""
         _LOGGER.debug("Adding to plug processing queue: %s", network_info)
-        await self._plug_added_queue.add(
+        self._plug_added_queue.add(
             (
                 network_info["mac"],
                 network_info["host"],
@@ -162,7 +161,7 @@ async def _monitor_plug_queue(self):
                             " Skipping and flushing from queue. ",
                             mac_address,
                         )
-                        await self._plug_added_queue.remove(
+                        self._plug_added_queue.remove(
                             (mac_address, host, port, name)
                         )
 
@@ -327,7 +326,7 @@ async def _acknowledge_plug_added_to_homeassistant(
         self, mac_address, host, port, name
     ):
         self._create_api(mac_address, host, port, name)
-        await self._plug_added_queue.remove((mac_address, host, port, name))
+        self._plug_added_queue.remove((mac_address, host, port, name))
 
     async def _plug_added(self, info):
         _LOGGER.debug(" Request to add plug received: %s", info)

custom_components/powersensor/PowersensorMessageDispatcher.py

@@ -1,6 +1,8 @@
 """Enum determining what measurements a Powersensor plug can report."""
+
 from enum import Enum
 
+
 class PlugMeasurements(Enum):
     """Enum to keep track of what measurements plugs can report."""
 

custom_components/powersensor/sensor/PlugMeasurements.py

@@ -42,7 +42,7 @@ class PowersensorVirtualHouseholdSensorEntityDescription(SensorEntityDescription
 FMT_WS_TO_KWH = lambda f: float(f)/3600000
 
 class PowersensorHouseholdEntity(SensorEntity):
-    """Powersensor Virtual Household entity"""
+    """Powersensor Virtual Household entity."""
 
     should_poll = False
     _attr_has_entity_name = True

custom_components/powersensor/sensor/PowersensorHouseholdEntity.py

@@ -1,4 +1,5 @@
 """Enum determining what measurements a Powersensor sensor can report."""
+
 from enum import Enum