| /******************************************************************************* |
| * Copyright (c) 2009 Shane Clarke. |
| * 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: |
| * Shane Clarke - initial API and implementation |
| *******************************************************************************/ |
| package org.eclipse.jst.ws.jaxws.core.tests; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| import org.eclipse.core.runtime.CoreException; |
| import org.eclipse.core.runtime.NullProgressMonitor; |
| import org.eclipse.jdt.core.IMethod; |
| import org.eclipse.jdt.core.dom.Annotation; |
| import org.eclipse.jdt.core.dom.MemberValuePair; |
| import org.eclipse.jst.ws.annotations.core.AnnotationsCore; |
| import org.eclipse.jst.ws.annotations.core.utils.AnnotationUtils; |
| |
| /** |
| * |
| * @author sclarke |
| * |
| */ |
| public class AddAnnotationToMethodTest extends AbstractAnnotationTest { |
| |
| @Override |
| public String getPackageName() { |
| return "com.example"; |
| } |
| |
| @Override |
| public String getClassName() { |
| return "Calculator.java"; |
| } |
| |
| @Override |
| public String getClassContents() { |
| StringBuilder classContents = new StringBuilder("package com.example;\n\n"); |
| classContents.append("public class Calculator {\n\n\tpublic int add(int i, int k) {"); |
| classContents.append("\n\t\treturn i + k;\n\t}\n}"); |
| return classContents.toString(); |
| } |
| |
| @Override |
| public Annotation getAnnotation() { |
| List<MemberValuePair> memberValuePairs = new ArrayList<MemberValuePair>(); |
| |
| MemberValuePair operationValuePair = AnnotationsCore.createStringMemberValuePair(ast, |
| "operationName", "add"); |
| |
| memberValuePairs.add(operationValuePair); |
| |
| return AnnotationsCore.createAnnotation(ast, javax.jws.WebMethod.class, javax.jws.WebMethod.class |
| .getSimpleName(), memberValuePairs); |
| } |
| |
| public void testAddAnnotationToMethod() { |
| try { |
| assertNotNull(annotation); |
| assertEquals("WebMethod", AnnotationUtils.getAnnotationName(annotation)); |
| |
| IMethod method = source.findPrimaryType().getMethod("add", new String[] { "I", "I" }); |
| assertNotNull(method); |
| |
| AnnotationUtils.getImportChange(compilationUnit, javax.jws.WebMethod.class, textFileChange, true); |
| |
| AnnotationUtils.createMethodAnnotationChange(source, compilationUnit, rewriter, method, |
| annotation, textFileChange); |
| |
| assertTrue(executeChange(new NullProgressMonitor(), textFileChange)); |
| |
| assertTrue(AnnotationUtils.isAnnotationPresent(method, AnnotationUtils |
| .getAnnotationName(annotation))); |
| } catch (CoreException ce) { |
| fail(ce.getLocalizedMessage()); |
| } |
| } |
| } |