def test_pending_transaction_in_transaction():
"""Test transactional_update.pending_transaction when in transaction"""
tu_in_transaction = "salt.modules.transactional_update.in_transaction"
with patch(tu_in_transaction) as in_transaction_mock:
in_transaction_mock.return_value = True
with pytest.raises(CommandExecutionError):
tu.pending_transaction()
def test_pending_transaction(self):
"""Test transactional_update.pending_transaction"""
matrix = (
(False, ["1", "2+", "3-"], True),
(False, ["1", "2-", "3+"], True),
(False, ["1", "2", "3*"], False),
)
for in_transaction, snapshots, result in matrix:
salt_mock = {
"cmd.run_all":
MagicMock(return_value={
"stdout": snapshots,
"retcode": 0
})
}
tu_in_transaction = "salt.modules.transactional_update.in_transaction"
with patch(tu_in_transaction) as in_transaction_mock:
in_transaction_mock.return_value = in_transaction
with patch.dict(tu.__salt__, salt_mock):
assert tu.pending_transaction() is result
salt_mock["cmd.run_all"].assert_called_with([
"snapper", "--no-dbus", "list", "--columns", "number"
])