test/tools/javac/processing/filer/TestGetResource.java

Tue, 25 May 2010 15:54:51 -0700

author
ohair
date
Tue, 25 May 2010 15:54:51 -0700
changeset 554
9d9f26857129
parent 1
9a66ca7c79fa
child 699
d2aaaec153e8
permissions
-rw-r--r--

6943119: Rebrand source copyright notices
Reviewed-by: darcy

duke@1 1 /*
ohair@554 2 * Copyright (c) 2006, Oracle and/or its affiliates. All rights reserved.
duke@1 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
duke@1 4 *
duke@1 5 * This code is free software; you can redistribute it and/or modify it
duke@1 6 * under the terms of the GNU General Public License version 2 only, as
duke@1 7 * published by the Free Software Foundation.
duke@1 8 *
duke@1 9 * This code is distributed in the hope that it will be useful, but WITHOUT
duke@1 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
duke@1 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
duke@1 12 * version 2 for more details (a copy is included in the LICENSE file that
duke@1 13 * accompanied this code).
duke@1 14 *
duke@1 15 * You should have received a copy of the GNU General Public License version
duke@1 16 * 2 along with this work; if not, write to the Free Software Foundation,
duke@1 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
duke@1 18 *
ohair@554 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@554 20 * or visit www.oracle.com if you need additional information or have any
ohair@554 21 * questions.
duke@1 22 */
duke@1 23
duke@1 24 /*
duke@1 25 * @test
duke@1 26 * @bug 6380018 6449798
duke@1 27 * @summary Test Filer.getResource
duke@1 28 * @author Joseph D. Darcy
duke@1 29 * @build TestGetResource
duke@1 30 * @compile -processor TestGetResource -proc:only -Aphase=write TestGetResource.java
duke@1 31 * @compile -processor TestGetResource -proc:only -Aphase=read TestGetResource.java
duke@1 32 */
duke@1 33
duke@1 34 import java.util.Set;
duke@1 35 import java.util.Map;
duke@1 36 import javax.annotation.processing.*;
duke@1 37 import javax.lang.model.SourceVersion;
duke@1 38 import static javax.lang.model.SourceVersion.*;
duke@1 39 import javax.lang.model.element.*;
duke@1 40 import javax.lang.model.util.*;
duke@1 41 import static javax.lang.model.util.ElementFilter.*;
duke@1 42 import static javax.tools.Diagnostic.Kind.*;
duke@1 43 import static javax.tools.StandardLocation.*;
duke@1 44 import java.io.IOException;
duke@1 45 import java.io.PrintWriter;
duke@1 46
duke@1 47 /**
duke@1 48 * Test basic functionality of the Filer.getResource method. On the
duke@1 49 * first run of the annotation processor, write out a resource file
duke@1 50 * and on the second run read it in.
duke@1 51 */
duke@1 52 @SupportedAnnotationTypes("*")
duke@1 53 @SupportedOptions("phase")
duke@1 54 public class TestGetResource extends AbstractProcessor {
duke@1 55 private Messager messager;
duke@1 56 private Filer filer;
duke@1 57 private Map<String,String> options;
duke@1 58
duke@1 59 private static String CONTENTS = "Hello World.";
duke@1 60 private static String PKG = "";
duke@1 61 private static String RESOURCE_NAME = "Resource1";
duke@1 62
duke@1 63 public boolean process(Set<? extends TypeElement> annotations,
duke@1 64 RoundEnvironment roundEnv) {
duke@1 65 try {
duke@1 66 if (!roundEnv.processingOver()) {
duke@1 67 String phase = options.get("phase");
duke@1 68
duke@1 69 if (phase.equals("write")) {
duke@1 70 PrintWriter pw =
duke@1 71 new PrintWriter(filer.createResource(CLASS_OUTPUT, PKG, RESOURCE_NAME).openWriter());
duke@1 72 pw.print(CONTENTS);
duke@1 73 pw.close();
duke@1 74 } else if (phase.equals("read")) {
duke@1 75 String contents = filer.getResource(CLASS_OUTPUT,
duke@1 76 PKG,
duke@1 77 RESOURCE_NAME).getCharContent(false).toString();
duke@1 78 if (!contents.equals(CONTENTS))
duke@1 79 throw new RuntimeException("Expected \n\t" + CONTENTS +
duke@1 80 "\nbut instead got \n\t" +
duke@1 81 contents);
duke@1 82 // Now try to open the file for writing
duke@1 83 filer.createResource(CLASS_OUTPUT,
duke@1 84 PKG,
duke@1 85 RESOURCE_NAME);
duke@1 86 } else {
duke@1 87 throw new RuntimeException("Unexpected phase: " + phase);
duke@1 88 }
duke@1 89 }
duke@1 90 } catch(IOException ioe) {
duke@1 91 throw new RuntimeException(ioe);
duke@1 92 }
duke@1 93 return false;
duke@1 94 }
duke@1 95
duke@1 96 public SourceVersion getSupportedSourceVersion() {
duke@1 97 return SourceVersion.latest();
duke@1 98 }
duke@1 99
duke@1 100 public void init(ProcessingEnvironment processingEnv) {
duke@1 101 super.init(processingEnv);
duke@1 102 messager = processingEnv.getMessager();
duke@1 103 filer = processingEnv.getFiler();
duke@1 104 options = processingEnv.getOptions();
duke@1 105 }
duke@1 106 }

mercurial