def test_set_progress(self): """ Test set_progress, which simply checks if the progress_callback is None before calling it """ # Setup mock_callback = mock.Mock() distributor = ISODistributor() # Test distributor.set_progress('id', 'status', mock_callback) mock_callback.assert_called_once_with('id', 'status')
def test_set_progress_no_callback(self): """ Assert that set_progress don't not attempt to call the callback when it is None """ # Setup distributor = ISODistributor() # Test try: distributor.set_progress('id', 'status', None) except AttributeError: self.fail('set_progress should not try to call None')