def test_project_deletion_exception(self, mock_db, mock_print): mock_db().get_project_members_list.side_effect = Exception bdo = BDO() bdo.project_deletion() mock_print.assert_called_once() mock_db().get_project_members_list.assert_called_once() mock_db().rollback_data.assert_called_once()
def test_project_deletion(self, mock_db): mock_db().get_project_members_list.return_value = [['dummy', 'dummy', '2020-02-23']] mock_db().get_bdo_approvals_list.return_value = [['dummy_id', 'MEMBER|sample_member|ea32d0e8-298a-4967-' '9fbd-72dc595294fb|334aabf5-8663-492d-bf75-' '6a5d7ece032d|project1']] bdo = BDO() bdo.project_id = '334aabf5-8663-492d-bf75-6a5d7ece032d' bdo.project_deletion() mock_db().get_project_members_list.assert_called_once() mock_db().register_project_completion.assert_called_once() mock_db().remove_project_all_members.assert_called_once() mock_db().get_bdo_approvals_list.assert_called_once() mock_db().commit_data.assert_called_once() mock_db().resolve_request.assert_called_once_with("'True'", mock.ANY) mock_db().update_project.assert_called_once_with('is_deleted', 'True', mock.ANY)