refactor node state to fully put the agent in charge (#90)

src/api-service/__app__/onefuzzlib/azure/queue.py

@@ -88,6 +88,15 @@ def get_queue(name: QueueNameType, *, account_id: str) -> Optional[QueueServiceC
         return None
 
 
+def clear_queue(name: QueueNameType, *, account_id: str) -> None:
+    queue = get_queue(name, account_id=account_id)
+    if queue:
+        try:
+            queue.clear_messages()
+        except ResourceNotFoundError:
+            return None
+
+
 def send_message(
     name: QueueNameType,
     message: bytes,
@@ -102,6 +111,19 @@ def send_message(
             pass
 
 
+def remove_first_message(name: QueueNameType, *, account_id: str) -> bool:
+    create_queue(name, account_id=account_id)
+    queue = get_queue(name, account_id=account_id)
+    if queue:
+        try:
+            for message in queue.receive_messages():
+                queue.delete_message(message)
+                return True
+        except ResourceNotFoundError:
+            return False
+    return False
+
+
 A = TypeVar("A", bound=BaseModel)