/******************************************************************************* * Copyright (c) 2008 Institute for Software, HSR Hochschule fuer Technik * Rapperswil, University of applied sciences and others * All rights reserved. This program and the accompanying materials * are made available under the terms of the Eclipse Public License v1.0 * which accompanies this distribution, and is available at * http://www.eclipse.org/legal/epl-v10.html * * Contributors: * Institute for Software - initial API and implementation *******************************************************************************/ package org.eclipse.cdt.core.parser.tests.rewrite.changegenerator.remove; import junit.framework.Test; import org.eclipse.cdt.core.dom.ast.IASTDeclarator; import org.eclipse.cdt.core.dom.ast.IASTParameterDeclaration; import org.eclipse.cdt.core.dom.ast.ASTVisitor; import org.eclipse.cdt.core.parser.tests.rewrite.changegenerator.ChangeGeneratorTest; import org.eclipse.cdt.internal.core.dom.parser.cpp.CPPASTFunctionDeclarator; import org.eclipse.cdt.internal.core.dom.rewrite.ASTModification; import org.eclipse.cdt.internal.core.dom.rewrite.ASTModificationStore; import org.eclipse.cdt.internal.core.dom.rewrite.ASTModification.ModificationKind; public class FirstParameterTest extends ChangeGeneratorTest { public FirstParameterTest(){ super("Remove First Parameter Node"); //$NON-NLS-1$ } @Override protected void setUp() throws Exception { source = "void foo(int a, int b, int c){\n}\n\n"; //$NON-NLS-1$ expectedSource = "void foo(int b, int c){\n}\n\n"; //$NON-NLS-1$ super.setUp(); } public static Test suite() { return new FirstParameterTest(); } @Override protected ASTVisitor createModificator( final ASTModificationStore modStore) { return new ASTVisitor() { { shouldVisitDeclarators = true; } @Override public int visit(IASTDeclarator declarator) { if (declarator instanceof CPPASTFunctionDeclarator) { CPPASTFunctionDeclarator functionDeclarator = (CPPASTFunctionDeclarator)declarator; IASTParameterDeclaration[] parameters = functionDeclarator.getParameters(); for(IASTParameterDeclaration curParam : parameters){ if(String.valueOf(curParam.getDeclarator().getName().toCharArray()).equals("a")){ //$NON-NLS-1$ ASTModification modification = new ASTModification(ModificationKind.REPLACE, curParam, null, null); modStore.storeModification(null, modification); } } } return PROCESS_CONTINUE; } }; } }