1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/test/tools/javac/processing/filer/TestGetResource.java Wed Apr 27 01:34:52 2016 +0800 1.3 @@ -0,0 +1,91 @@ 1.4 +/* 1.5 + * Copyright (c) 2006, 2010, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 1.23 + * or visit www.oracle.com if you need additional information or have any 1.24 + * 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 + * @library /tools/javac/lib 1.33 + * @build JavacTestingAbstractProcessor TestGetResource 1.34 + * @compile -processor TestGetResource -proc:only -Aphase=write TestGetResource.java 1.35 + * @compile -processor TestGetResource -proc:only -Aphase=read TestGetResource.java 1.36 + */ 1.37 + 1.38 +import java.util.Set; 1.39 +import java.util.Map; 1.40 +import javax.annotation.processing.*; 1.41 +import javax.lang.model.SourceVersion; 1.42 +import static javax.lang.model.SourceVersion.*; 1.43 +import javax.lang.model.element.*; 1.44 +import javax.lang.model.util.*; 1.45 +import static javax.lang.model.util.ElementFilter.*; 1.46 +import static javax.tools.Diagnostic.Kind.*; 1.47 +import static javax.tools.StandardLocation.*; 1.48 +import java.io.IOException; 1.49 +import java.io.PrintWriter; 1.50 + 1.51 +/** 1.52 + * Test basic functionality of the Filer.getResource method. On the 1.53 + * first run of the annotation processor, write out a resource file 1.54 + * and on the second run read it in. 1.55 + */ 1.56 +@SupportedOptions("phase") 1.57 +public class TestGetResource extends JavacTestingAbstractProcessor { 1.58 + private static String CONTENTS = "Hello World."; 1.59 + private static String PKG = ""; 1.60 + private static String RESOURCE_NAME = "Resource1"; 1.61 + 1.62 + public boolean process(Set<? extends TypeElement> annotations, 1.63 + RoundEnvironment roundEnv) { 1.64 + try { 1.65 + if (!roundEnv.processingOver()) { 1.66 + String phase = options.get("phase"); 1.67 + 1.68 + if (phase.equals("write")) { 1.69 + PrintWriter pw = 1.70 + new PrintWriter(filer.createResource(CLASS_OUTPUT, PKG, RESOURCE_NAME).openWriter()); 1.71 + pw.print(CONTENTS); 1.72 + pw.close(); 1.73 + } else if (phase.equals("read")) { 1.74 + String contents = filer.getResource(CLASS_OUTPUT, 1.75 + PKG, 1.76 + RESOURCE_NAME).getCharContent(false).toString(); 1.77 + if (!contents.equals(CONTENTS)) 1.78 + throw new RuntimeException("Expected \n\t" + CONTENTS + 1.79 + "\nbut instead got \n\t" + 1.80 + contents); 1.81 + // Now try to open the file for writing 1.82 + filer.createResource(CLASS_OUTPUT, 1.83 + PKG, 1.84 + RESOURCE_NAME); 1.85 + } else { 1.86 + throw new RuntimeException("Unexpected phase: " + phase); 1.87 + } 1.88 + } 1.89 + } catch(IOException ioe) { 1.90 + throw new RuntimeException(ioe); 1.91 + } 1.92 + return false; 1.93 + } 1.94 +}