Skip to content

Commit 84512fd

Browse files
committed
fix mock_vhdutil
Signed-off-by: Ronan Abhamon <ronan.abhamon@vates.tech>
1 parent 373a24c commit 84512fd

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/test_blktap2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -494,7 +494,7 @@ def test_setup_cache_success_prt_td_present(self, mock_find_by_path, mock_launch
494494
mock_vdi.path = "prt_path"
495495

496496
# Return some sizes (all the same here)
497-
self.mock_vhdutil.getSizeVirt.return_value = 20 * 1024 * 1024
497+
self.mock_vhdutil.return_value.getSizeVirt.return_value = 20 * 1024 * 1024
498498

499499
prt_tapdisk = blktap2.Tapdisk(
500500
1457, 3, 'vhd', 'dummy', 1)
@@ -528,7 +528,7 @@ def test_setup_cache_success_no_prt_td_present(self, mock_find_by_path, mock_lau
528528
mock_vdi.path = "prt_path"
529529

530530
# Return some sizes (all the same here)
531-
self.mock_vhdutil.getSizeVirt.return_value = 20 * 1024 * 1024
531+
self.mock_vhdutil.return_value.getSizeVirt.return_value = 20 * 1024 * 1024
532532

533533
# No existing tapdisk paths
534534
mock_find_by_path.return_value = {}

0 commit comments

Comments
 (0)