add support for disk encryption attrs attributes

Signed-off-by: nanli <[email protected]>
avocado-framework · Sep 12, 2024 · f4fe2dc · f4fe2dc
1 parent 9403e81
commit f4fe2dc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/virttest/libvirt_xml/devices/disk.py b/virttest/libvirt_xml/devices/disk.py
@@ -644,7 +644,7 @@ class Encryption(base.base.LibvirtXMLBase):
             dict, keys: type, uuid
         """
 
-        __slots__ = ("encryption", "secret")
+        __slots__ = ("encryption", "secret", "attrs")
 
         def __init__(self, virsh_instance=base.base.virsh):
             accessors.XMLAttribute(
@@ -657,6 +657,9 @@ def __init__(self, virsh_instance=base.base.virsh):
             accessors.XMLElementDict(
                 "secret", self, parent_xpath="/", tag_name="secret"
             )
+            accessors.XMLElementDict(
+                "attrs", self, parent_xpath="/", tag_name="encryption"
+            )
             super(self.__class__, self).__init__(virsh_instance=virsh_instance)
             self.xml = "<encryption/>"