1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/test/tools/javac/processing/filer/TestGetResource.java Sat Dec 01 00:00:00 2007 +0000
1.3 @@ -0,0 +1,106 @@
1.4 +/*
1.5 + * Copyright 2006 Sun Microsystems, Inc. All Rights Reserved.
1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
1.7 + *
1.8 + * This code is free software; you can redistribute it and/or modify it
1.9 + * under the terms of the GNU General Public License version 2 only, as
1.10 + * published by the Free Software Foundation.
1.11 + *
1.12 + * This code is distributed in the hope that it will be useful, but WITHOUT
1.13 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
1.14 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
1.15 + * version 2 for more details (a copy is included in the LICENSE file that
1.16 + * accompanied this code).
1.17 + *
1.18 + * You should have received a copy of the GNU General Public License version
1.19 + * 2 along with this work; if not, write to the Free Software Foundation,
1.20 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
1.21 + *
1.22 + * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
1.23 + * CA 95054 USA or visit www.sun.com if you need additional information or
1.24 + * have any questions.
1.25 + */
1.26 +
1.27 +/*
1.28 + * @test
1.29 + * @bug 6380018 6449798
1.30 + * @summary Test Filer.getResource
1.31 + * @author Joseph D. Darcy
1.32 + * @build TestGetResource
1.33 + * @compile -processor TestGetResource -proc:only -Aphase=write TestGetResource.java
1.34 + * @compile -processor TestGetResource -proc:only -Aphase=read TestGetResource.java
1.35 + */
1.36 +
1.37 +import java.util.Set;
1.38 +import java.util.Map;
1.39 +import javax.annotation.processing.*;
1.40 +import javax.lang.model.SourceVersion;
1.41 +import static javax.lang.model.SourceVersion.*;
1.42 +import javax.lang.model.element.*;
1.43 +import javax.lang.model.util.*;
1.44 +import static javax.lang.model.util.ElementFilter.*;
1.45 +import static javax.tools.Diagnostic.Kind.*;
1.46 +import static javax.tools.StandardLocation.*;
1.47 +import java.io.IOException;
1.48 +import java.io.PrintWriter;
1.49 +
1.50 +/**
1.51 + * Test basic functionality of the Filer.getResource method. On the
1.52 + * first run of the annotation processor, write out a resource file
1.53 + * and on the second run read it in.
1.54 + */
1.55 +@SupportedAnnotationTypes("*")
1.56 +@SupportedOptions("phase")
1.57 +public class TestGetResource extends AbstractProcessor {
1.58 + private Messager messager;
1.59 + private Filer filer;
1.60 + private Map<String,String> options;
1.61 +
1.62 + private static String CONTENTS = "Hello World.";
1.63 + private static String PKG = "";
1.64 + private static String RESOURCE_NAME = "Resource1";
1.65 +
1.66 + public boolean process(Set<? extends TypeElement> annotations,
1.67 + RoundEnvironment roundEnv) {
1.68 + try {
1.69 + if (!roundEnv.processingOver()) {
1.70 + String phase = options.get("phase");
1.71 +
1.72 + if (phase.equals("write")) {
1.73 + PrintWriter pw =
1.74 + new PrintWriter(filer.createResource(CLASS_OUTPUT, PKG, RESOURCE_NAME).openWriter());
1.75 + pw.print(CONTENTS);
1.76 + pw.close();
1.77 + } else if (phase.equals("read")) {
1.78 + String contents = filer.getResource(CLASS_OUTPUT,
1.79 + PKG,
1.80 + RESOURCE_NAME).getCharContent(false).toString();
1.81 + if (!contents.equals(CONTENTS))
1.82 + throw new RuntimeException("Expected \n\t" + CONTENTS +
1.83 + "\nbut instead got \n\t" +
1.84 + contents);
1.85 + // Now try to open the file for writing
1.86 + filer.createResource(CLASS_OUTPUT,
1.87 + PKG,
1.88 + RESOURCE_NAME);
1.89 + } else {
1.90 + throw new RuntimeException("Unexpected phase: " + phase);
1.91 + }
1.92 + }
1.93 + } catch(IOException ioe) {
1.94 + throw new RuntimeException(ioe);
1.95 + }
1.96 + return false;
1.97 + }
1.98 +
1.99 + public SourceVersion getSupportedSourceVersion() {
1.100 + return SourceVersion.latest();
1.101 + }
1.102 +
1.103 + public void init(ProcessingEnvironment processingEnv) {
1.104 + super.init(processingEnv);
1.105 + messager = processingEnv.getMessager();
1.106 + filer = processingEnv.getFiler();
1.107 + options = processingEnv.getOptions();
1.108 + }
1.109 +}
