def test_removal_of_updates_with_unknown_object_exception(self): def mock_delete(unused_self): raise github.UnknownObjectException(status='', data='') delete_swap = self.swap(github.GitRef.GitRef, 'delete', mock_delete) with self.run_cmd_swap, self.get_git_ref_swap, delete_swap: update_changelog_and_credits.remove_updates_and_delete_branch( self.mock_repo, 'target_branch')
def test_removal_of_updates_with_no_exception(self): def mock_delete(unused_self): pass delete_swap = self.swap(github.GitRef.GitRef, 'delete', mock_delete) with self.run_cmd_swap, self.get_git_ref_swap, delete_swap: update_changelog_and_credits.remove_updates_and_delete_branch( self.mock_repo, 'target_branch')
def test_removal_of_updates_with_valid_exception(self): def mock_delete(unused_self): raise Exception('Error') delete_swap = self.swap(github.GitRef.GitRef, 'delete', mock_delete) with self.run_cmd_swap, self.get_git_ref_swap, delete_swap: with self.assertRaisesRegexp( Exception, ('Please ensure that target_branch branch is deleted before ' 're-running the script')): update_changelog_and_credits.remove_updates_and_delete_branch( self.mock_repo, 'target_branch')