fix: fix agent group command by adding scheduler_cron_time

Update the create and modify agent group command to include
the mandatory scheduler_cron_time parameter.

Author: ozgen

gvm/protocols/gmp/_gmpnext.py

@@ -228,6 +228,7 @@ def create_agent_group(
         self,
         name: str,
         agent_ids: list[str],
+        scheduler_cron_time: str,
         *,
         comment: str | None = None,
     ) -> T:
@@ -236,6 +237,7 @@ def create_agent_group(
         Args:
             name: Name of the new agent group.
             agent_ids: List of agent UUIDs to include in the group (required).
+            scheduler_cron_time: Scheduler cron to use.
             comment: Optional comment for the group.
 
         Raises:
@@ -246,12 +248,14 @@ def create_agent_group(
                 name=name,
                 comment=comment,
                 agent_ids=agent_ids,
+                scheduler_cron_time=scheduler_cron_time,
             )
         )
 
     def modify_agent_group(
         self,
         agent_group_id: EntityID,
+        scheduler_cron_time: str,
         *,
         name: str | None = None,
         comment: str | None = None,
@@ -262,6 +266,7 @@ def modify_agent_group(
         Args:
             agent_group_id: UUID of the group to modify.
             name: Optional new name for the group.
+            scheduler_cron_time: Scheduler cron to use.
             comment: Optional comment for the group.
             agent_ids: Optional list of agent UUIDs to set for the group.
 
@@ -274,6 +279,7 @@ def modify_agent_group(
                 name=name,
                 comment=comment,
                 agent_ids=agent_ids,
+                scheduler_cron_time=scheduler_cron_time,
             )
         )
 

gvm/protocols/gmp/requests/next/_agent_groups.py

@@ -16,6 +16,7 @@ def create_agent_group(
         cls,
         name: str,
         agent_ids: list[str],
+        scheduler_cron_time: str,
         *,
         comment: str | None = None,
     ) -> Request:
@@ -24,6 +25,7 @@ def create_agent_group(
         Args:
             name: Name of the new agent group.
             agent_ids: List of agent UUIDs to include in the group (required).
+            scheduler_cron_time: Cron-like time to schedule new agent groups (required).
             comment: Optional comment for the group.
 
         Raises:
@@ -38,9 +40,15 @@ def create_agent_group(
             raise RequiredArgument(
                 function=cls.create_agent_group.__name__, argument="agent_ids"
             )
+        if not scheduler_cron_time:
+            raise RequiredArgument(
+                function=cls.create_agent_group.__name__,
+                argument="scheduler_cron_time",
+            )
 
         cmd = XmlCommand("create_agent_group")
         cmd.add_element("name", name)
+        cmd.add_element("scheduler_cron_time", scheduler_cron_time)
 
         if comment:
             cmd.add_element("comment", comment)
@@ -76,6 +84,7 @@ def clone_agent_group(cls, agent_group_id: EntityID) -> Request:
     def modify_agent_group(
         cls,
         agent_group_id: EntityID,
+        scheduler_cron_time: str,
         *,
         name: str | None = None,
         comment: str | None = None,
@@ -85,6 +94,7 @@ def modify_agent_group(
 
         Args:
             agent_group_id: UUID of the group to modify.
+            scheduler_cron_time: Cron-like time to schedule the agent groups.
             name: Optional new name for the group.
             comment: Optional comment for the group.
             agent_ids: Optional list of agent UUIDs to set for the group.
@@ -98,9 +108,15 @@ def modify_agent_group(
                 argument="agent_group_id",
             )
 
+        if not scheduler_cron_time:
+            raise RequiredArgument(
+                function=cls.modify_agent_group.__name__,
+                argument="scheduler_cron_time",
+            )
+
         cmd = XmlCommand("modify_agent_group")
         cmd.set_attribute("agent_group_id", str(agent_group_id))
-
+        cmd.add_element("scheduler_cron_time", scheduler_cron_time)
         if name:
             cmd.add_element("name", name)
 

tests/protocols/gmpnext/entities/agent_groups/test_create_agent_group.py

@@ -11,30 +11,49 @@ def test_create_agent_group(self):
             name="ExampleGroup",
             agent_ids=["agent-1", "agent-2"],
             comment="Sample comment",
+            scheduler_cron_time="0 */5 * * *",
         )
 
         self.connection.send.has_been_called_with(
             b"<create_agent_group>"
             b"<name>ExampleGroup</name>"
+            b"<scheduler_cron_time>0 */5 * * *</scheduler_cron_time>"
             b"<comment>Sample comment</comment>"
             b'<agents><agent id="agent-1"/><agent id="agent-2"/></agents>'
             b"</create_agent_group>"
         )
 
     def test_create_agent_group_without_name(self):
         with self.assertRaises(RequiredArgument):
-            self.gmp.create_agent_group(name="", agent_ids=["agent-1"])
+            self.gmp.create_agent_group(
+                name="",
+                agent_ids=["agent-1"],
+                scheduler_cron_time="0 */5 * * *",
+            )
 
     def test_create_agent_group_without_agents(self):
         with self.assertRaises(RequiredArgument):
-            self.gmp.create_agent_group(name="Group", agent_ids=[])
+            self.gmp.create_agent_group(
+                name="Group", agent_ids=[], scheduler_cron_time="0 */5 * * *"
+            )
+
+    def test_create_agent_group_without_scheduler_cron_time(self):
+        with self.assertRaises(RequiredArgument):
+            self.gmp.create_agent_group(
+                name="Group", agent_ids=["agent-1"], scheduler_cron_time=None
+            )
 
     def test_create_agent_group_without_comment(self):
-        self.gmp.create_agent_group(name="GroupX", agent_ids=["a1", "a2"])
+        self.gmp.create_agent_group(
+            name="GroupX",
+            agent_ids=["a1", "a2"],
+            scheduler_cron_time="0 */5 * * *",
+        )
 
         self.connection.send.has_been_called_with(
             b"<create_agent_group>"
             b"<name>GroupX</name>"
+            b"<scheduler_cron_time>0 */5 * * *</scheduler_cron_time>"
             b'<agents><agent id="a1"/><agent id="a2"/></agents>'
             b"</create_agent_group>"
         )

tests/protocols/gmpnext/entities/agent_groups/test_modify_agent_group.py

@@ -11,11 +11,13 @@ def test_modify_agent_group(self):
             agent_group_id="group-1",
             name="NewName",
             comment="Updated comment",
+            scheduler_cron_time="0 */5 * * *",
             agent_ids=["agent-1", "agent-2"],
         )
 
         self.connection.send.has_been_called_with(
             b'<modify_agent_group agent_group_id="group-1">'
+            b"<scheduler_cron_time>0 */5 * * *</scheduler_cron_time>"
             b"<name>NewName</name>"
             b"<comment>Updated comment</comment>"
             b'<agents><agent id="agent-1"/><agent id="agent-2"/></agents>'
@@ -24,17 +26,27 @@ def test_modify_agent_group(self):
 
     def test_modify_agent_group_without_id(self):
         with self.assertRaises(RequiredArgument):
-            self.gmp.modify_agent_group(agent_group_id=None)
+            self.gmp.modify_agent_group(
+                agent_group_id=None, scheduler_cron_time="0 */5 * * *"
+            )
+
+    def test_modify_agent_group_without_scheduler_cron_time(self):
+        with self.assertRaises(RequiredArgument):
+            self.gmp.modify_agent_group(
+                agent_group_id="group-1", scheduler_cron_time=None
+            )
 
     def test_modify_agent_group_without_name(self):
         self.gmp.modify_agent_group(
             agent_group_id="group-1",
             comment="Updated comment",
+            scheduler_cron_time="0 */5 * * *",
             agent_ids=["agent-1", "agent-2"],
         )
 
         self.connection.send.has_been_called_with(
             b'<modify_agent_group agent_group_id="group-1">'
+            b"<scheduler_cron_time>0 */5 * * *</scheduler_cron_time>"
             b"<comment>Updated comment</comment>"
             b'<agents><agent id="agent-1"/><agent id="agent-2"/></agents>'
             b"</modify_agent_group>"
@@ -44,11 +56,13 @@ def test_modify_agent_group_without_comment(self):
         self.gmp.modify_agent_group(
             agent_group_id="group-1",
             name="NewName",
+            scheduler_cron_time="0 */5 * * *",
             agent_ids=["agent-1", "agent-2"],
         )
 
         self.connection.send.has_been_called_with(
             b'<modify_agent_group agent_group_id="group-1">'
+            b"<scheduler_cron_time>0 */5 * * *</scheduler_cron_time>"
             b"<name>NewName</name>"
             b'<agents><agent id="agent-1"/><agent id="agent-2"/></agents>'
             b"</modify_agent_group>"
@@ -59,10 +73,12 @@ def test_modify_agent_group_without_agent_ids(self):
             agent_group_id="group-1",
             name="NewName",
             comment="Updated comment",
+            scheduler_cron_time="0 */5 * * *",
         )
 
         self.connection.send.has_been_called_with(
             b'<modify_agent_group agent_group_id="group-1">'
+            b"<scheduler_cron_time>0 */5 * * *</scheduler_cron_time>"
             b"<name>NewName</name>"
             b"<comment>Updated comment</comment>"
             b"</modify_agent_group>"