Here are the examples of the python api mock.call.set_attribute taken from open source projects. By voting up you can indicate which examples are most useful and appropriate.
1 Examples
0
Example 1
Project: barman Source File: test_backup.py
@patch('barman.backup.BackupInfo')
def test_keyboard_interrupt(self, mock_infofile):
"""
Unit test for a quick check on exception catching
during backup operations
Test case 1: raise a general exception, backup status in
BackupInfo should be FAILED.
Test case 2: raise a KeyboardInterrupt exception, simulating
a user pressing CTRL + C while a backup is in progress,
backup status in BackupInfo should be FAILED.
"""
# BackupManager setup
backup_manager = build_backup_manager()
instance = mock_infofile.return_value
# Instruct the patched method to raise a general exception
backup_manager.executor.start_backup = Mock(
side_effect=Exception('abc'))
# invoke backup method
backup_manager.backup()
# verify that mock status is FAILED
assert mock.call.set_attribute(
'status', 'FAILED') in instance.mock_calls
# Instruct the patched method to raise a KeyboardInterrupt
backup_manager.executor.start_backup = Mock(
side_effect=KeyboardInterrupt())
# invoke backup method
backup_manager.backup()
# verify that mock status is FAILED
assert mock.call.set_attribute(
'status', 'FAILED') in instance.mock_calls