implement cbt_tests on lvmohba sr

Signed-off-by: Goulven Riou <goulven.riou@vates.tech>

Author: Lankou66

tests/storage/lvmohba/test_lvmohba_sr.py

@@ -2,14 +2,18 @@
 
 from lib.commands import SSHCommandFailed
 from lib.common import Defer, vm_image, wait_for
+from lib.host import Host
 from lib.sr import SR
-from lib.vdi import VDI, ImageFormat
+from lib.vdi import VDI
 from lib.vm import VM
 from tests.storage import (
     MAX_VDI_SIZE,
+    CBTTest,
     CoalesceOperation,
     ImageFormat,
     XVACompression,
+    assert_cbt_log_does_not_exist_lvm_sr,
+    assert_cbt_log_exists_lvm_sr,
     coalesce_integrity,
     full_vdi_write,
     vdi_export_import,
@@ -93,3 +97,63 @@ def test_reboot(self, host, lvmohba_sr, vm_on_lvmohba_sr):
         vm.shutdown(verify=True)
 
     # *** End of tests with reboots
+
+
+class TestLVMoHBACBT(CBTTest):
+    """Test CBT functionality on LVMOHBA SR"""
+
+    @staticmethod
+    def assert_cbt_log_exists(host: Host, sr: SR, vdi: VDI) -> None:
+        assert_cbt_log_exists_lvm_sr(host, sr, vdi)
+
+    @staticmethod
+    def assert_cbt_log_does_not_exist(host: Host, sr: SR, vdi: VDI) -> None:
+        assert_cbt_log_does_not_exist_lvm_sr(host, sr, vdi)
+
+    def test_enable_disable_cbt(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI) -> None:
+        self._test_enable_disable_cbt(host, lvmohba_sr, vdi_on_lvmohba_sr)
+
+    def test_cbt_log_creation(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI) -> None:
+        self._test_cbt_log_creation(host, lvmohba_sr, vdi_on_lvmohba_sr)
+
+    def test_snapshot_with_cbt(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI) -> None:
+        self._test_snapshot_with_cbt(host, lvmohba_sr, vdi_on_lvmohba_sr)
+
+    @pytest.mark.small_vm
+    def test_changed_blocks_tracking(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI,
+                                     vm_on_lvmohba_sr: VM) -> None:
+        self._test_changed_blocks_tracking(host, lvmohba_sr, vdi_on_lvmohba_sr, vm_on_lvmohba_sr)
+
+    @pytest.mark.small_vm
+    def test_cbt_after_coalesce(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI,
+                                vm_on_lvmohba_sr: VM) -> None:
+        self._test_cbt_after_coalesce(host, lvmohba_sr, vdi_on_lvmohba_sr, vm_on_lvmohba_sr)
+
+    @pytest.mark.small_vm
+    def test_incremental_snap_scenario(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI,
+                                       vm_on_lvmohba_sr: VM) -> None:
+        self._test_incremental_snap_scenario(host, lvmohba_sr, vdi_on_lvmohba_sr, vm_on_lvmohba_sr)
+
+    def test_disable_cbt_removes_log(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI) -> None:
+        self._test_disable_cbt_removes_log(host, lvmohba_sr, vdi_on_lvmohba_sr)
+
+    def test_destroy_vdi_removes_cbt_log(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI) -> None:
+        self._test_destroy_vdi_removes_cbt_log(host, lvmohba_sr, vdi_on_lvmohba_sr)
+
+    def test_cbt_persist_after_sr_reboot(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI) -> None:
+        self._test_cbt_persist_after_sr_reboot(host, lvmohba_sr, vdi_on_lvmohba_sr)
+
+    def test_cbt_on_snapshot_chain(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI) -> None:
+        self._test_cbt_on_snapshot_chain(host, lvmohba_sr, vdi_on_lvmohba_sr)
+
+    def test_cbt_parent_disable_does_not_affect_snapshot(self, host: Host, lvmohba_sr: SR,
+                                                         vdi_on_lvmohba_sr: VDI) -> None:
+        self._test_cbt_parent_disable_does_not_affect_snapshot(host, lvmohba_sr, vdi_on_lvmohba_sr)
+
+    @pytest.mark.small_vm
+    def test_cbt_bitmap_non_zero_after_write(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI,
+                                             vm_on_lvmohba_sr: VM) -> None:
+        self._test_cbt_bitmap_non_zero_after_write(host, lvmohba_sr, vdi_on_lvmohba_sr, vm_on_lvmohba_sr)
+
+    def test_cbt_data_destroy(self, host: Host, lvmohba_sr: SR, vdi_on_lvmohba_sr: VDI) -> None:
+        self._test_cbt_data_destroy(host, lvmohba_sr, vdi_on_lvmohba_sr)