1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from unittest.mock import MagicMock, call
from pyipmi.helper import clear_repository_helper
from pyipmi.msgs.constants import (REPOSITORY_ERASURE_COMPLETED,
REPOSITORY_ERASURE_IN_PROGRESS,
REPOSITORY_INITIATE_ERASE,
REPOSITORY_GET_ERASE_STATUS)
def test_clear_repository_helper():
reserve_fn = MagicMock()
reserve_fn.return_value = (0x1234)
clear_fn = MagicMock()
clear_fn.side_effect = [
REPOSITORY_ERASURE_COMPLETED,
REPOSITORY_ERASURE_IN_PROGRESS,
REPOSITORY_ERASURE_COMPLETED,
]
clear_repository_helper(reserve_fn, clear_fn)
clear_calls = [
call(REPOSITORY_INITIATE_ERASE, 0x1234),
call(REPOSITORY_GET_ERASE_STATUS, 0x1234),
call(REPOSITORY_GET_ERASE_STATUS, 0x1234),
]
clear_fn.assert_has_calls(clear_calls)
assert clear_fn.call_count == 3
|
