Implement Create/Destroy SnapshotStep

chroma_core/models/host.py

@@ -1785,11 +1785,34 @@ class NoNidsPresent(Exception):
     pass
 
 
+class CreateSnapshotStep(Step):
+    def run(self, kwargs):
+        args = {"fsname": kwargs["fsname"], "name": kwargs["name"]}
+        if "comment" in kwargs:
+            args["comment"] = kwargs["comment"]
+        self.invoke_rust_agent_expect_result(
+            kwargs["host"], "snapshot_create", args,
+        )
+
+
+class DestroySnapshotStep(Step):
+    def run(self, kwargs):
+        self.invoke_rust_agent_expect_result(
+            kwargs["host"],
+            "snapshot_destroy",
+            {"fsname": kwargs["fsname"], "name": kwargs["name"], "force": kwargs["force"]},
+        )
+
+
 class MountSnapshotStep(Step):
     def run(self, kwargs):
-        self.invoke_rust_agent_expect_result(kwargs["host"], "snapshot_mount", [kwargs["fsname"], kwargs["name"]])
+        self.invoke_rust_agent_expect_result(
+            kwargs["host"], "snapshot_mount", {"fsname": kwargs["fsname"], "name": kwargs["name"]}
+        )
 
 
 class UnmountSnapshotStep(Step):
     def run(self, kwargs):
-        self.invoke_rust_agent_expect_result(kwargs["host"], "snapshot_unmount", [kwargs["fsname"], kwargs["name"]])
+        self.invoke_rust_agent_expect_result(
+            kwargs["host"], "snapshot_unmount", {"fsname": kwargs["fsname"], "name": kwargs["name"]}
+        )