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

Thu, 31 Aug 2017 15:17:03 +0800

author
aoqi
date
Thu, 31 Aug 2017 15:17:03 +0800
changeset 2525
2eb010b6cb22
parent 1466
b52a38d4536c
parent 0
959103a6100f
permissions
-rw-r--r--

merge

aoqi@0 1 /*
aoqi@0 2 * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
aoqi@0 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0 4 *
aoqi@0 5 * This code is free software; you can redistribute it and/or modify it
aoqi@0 6 * under the terms of the GNU General Public License version 2 only, as
aoqi@0 7 * published by the Free Software Foundation.
aoqi@0 8 *
aoqi@0 9 * This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
aoqi@0 12 * version 2 for more details (a copy is included in the LICENSE file that
aoqi@0 13 * accompanied this code).
aoqi@0 14 *
aoqi@0 15 * You should have received a copy of the GNU General Public License version
aoqi@0 16 * 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0 18 *
aoqi@0 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0 20 * or visit www.oracle.com if you need additional information or have any
aoqi@0 21 * questions.
aoqi@0 22 */
aoqi@0 23
aoqi@0 24 /*
aoqi@0 25 * @test
aoqi@0 26 * @bug 6999891
aoqi@0 27 * @summary Test valid relative names for Filer.createResource and Filer.getResource
aoqi@0 28 * @library /tools/javac/lib
aoqi@0 29 * @build JavacTestingAbstractProcessor
aoqi@0 30 * @compile TestValidRelativeNames.java
aoqi@0 31 * @compile/process -processor TestValidRelativeNames -Amode=create java.lang.Object
aoqi@0 32 * @compile/process -processor TestValidRelativeNames -Amode=get java.lang.Object
aoqi@0 33 */
aoqi@0 34
aoqi@0 35 import java.io.*;
aoqi@0 36 import java.util.*;
aoqi@0 37 import javax.annotation.processing.*;
aoqi@0 38 import javax.lang.model.*;
aoqi@0 39 import javax.lang.model.element.*;
aoqi@0 40 import javax.tools.Diagnostic;
aoqi@0 41 import javax.tools.StandardLocation;
aoqi@0 42
aoqi@0 43 @SupportedOptions("mode")
aoqi@0 44 public class TestValidRelativeNames extends JavacTestingAbstractProcessor {
aoqi@0 45 enum Kind { READER_WRITER, INPUT_OUTPUT_STREAM };
aoqi@0 46
aoqi@0 47 static final String[] validRelativeNames = {
aoqi@0 48 "foo", "foo.bar", ".foo", ".foo.bar", "foodir/bar", "foodir/.bar"
aoqi@0 49 };
aoqi@0 50
aoqi@0 51 public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
aoqi@0 52 if (roundEnv.processingOver()) {
aoqi@0 53 String mode = options.get("mode");
aoqi@0 54 for (String relativeBase: validRelativeNames) {
aoqi@0 55 for (Kind kind: Kind.values()) {
aoqi@0 56 if (mode.equals("create"))
aoqi@0 57 testCreate(relativeBase, kind);
aoqi@0 58 else
aoqi@0 59 testGet(relativeBase, kind);
aoqi@0 60 }
aoqi@0 61 }
aoqi@0 62 }
aoqi@0 63
aoqi@0 64 return true;
aoqi@0 65 }
aoqi@0 66
aoqi@0 67 void testCreate(String relativeBase, Kind kind) {
aoqi@0 68 String relative = getRelative(relativeBase, kind);
aoqi@0 69 System.out.println("test create relative path: " + relative + ", kind: " + kind);
aoqi@0 70 try {
aoqi@0 71 switch (kind) {
aoqi@0 72 case READER_WRITER:
aoqi@0 73 try (Writer writer = filer.createResource(
aoqi@0 74 StandardLocation.CLASS_OUTPUT, "", relative).openWriter()) {
aoqi@0 75 writer.write(relative);
aoqi@0 76 }
aoqi@0 77 break;
aoqi@0 78
aoqi@0 79 case INPUT_OUTPUT_STREAM:
aoqi@0 80 try (OutputStream out = filer.createResource(
aoqi@0 81 StandardLocation.CLASS_OUTPUT, "", relative).openOutputStream()) {
aoqi@0 82 out.write(relative.getBytes());
aoqi@0 83 }
aoqi@0 84 break;
aoqi@0 85 }
aoqi@0 86 } catch (Exception e) {
aoqi@0 87 messager.printMessage(Diagnostic.Kind.ERROR,
aoqi@0 88 "relative path: " + relative + ", kind: " + kind + ", unexpected exception: " + e);
aoqi@0 89 }
aoqi@0 90 }
aoqi@0 91
aoqi@0 92 void testGet(String relativeBase, Kind kind) {
aoqi@0 93 String relative = getRelative(relativeBase, kind);
aoqi@0 94 System.out.println("test get relative path: " + relative + ", kind: " + kind);
aoqi@0 95 try {
aoqi@0 96 switch (kind) {
aoqi@0 97 case READER_WRITER:
aoqi@0 98 try (Reader reader = new BufferedReader(filer.getResource(
aoqi@0 99 StandardLocation.CLASS_OUTPUT, "", relative).openReader(true))) {
aoqi@0 100 StringBuilder sb = new StringBuilder();
aoqi@0 101 char[] buf = new char[1024];
aoqi@0 102 int n;
aoqi@0 103 while ((n = reader.read(buf, 0, buf.length)) > 0)
aoqi@0 104 sb.append(new String(buf, 0, n));
aoqi@0 105 if (!sb.toString().equals(relative)) {
aoqi@0 106 messager.printMessage(Diagnostic.Kind.ERROR, "unexpected content: " + sb);
aoqi@0 107 }
aoqi@0 108 }
aoqi@0 109 break;
aoqi@0 110
aoqi@0 111 case INPUT_OUTPUT_STREAM:
aoqi@0 112 try (InputStream in = new DataInputStream(filer.getResource(
aoqi@0 113 StandardLocation.CLASS_OUTPUT, "", relative).openInputStream())) {
aoqi@0 114 StringBuilder sb = new StringBuilder();
aoqi@0 115 byte[] buf = new byte[1024];
aoqi@0 116 int n;
aoqi@0 117 while ((n = in.read(buf, 0, buf.length)) > 0)
aoqi@0 118 sb.append(new String(buf, 0, n));
aoqi@0 119 if (!sb.toString().equals(relative)) {
aoqi@0 120 messager.printMessage(Diagnostic.Kind.ERROR, "unexpected content: " + sb);
aoqi@0 121 }
aoqi@0 122 }
aoqi@0 123 break;
aoqi@0 124 }
aoqi@0 125 } catch (Exception e) {
aoqi@0 126 messager.printMessage(Diagnostic.Kind.ERROR,
aoqi@0 127 "relative path: " + relative + ", kind: " + kind + ", unexpected exception: " + e);
aoqi@0 128 }
aoqi@0 129 }
aoqi@0 130
aoqi@0 131 String getRelative(String relativeBase, Kind kind) {
aoqi@0 132 String suffix = (kind == Kind.READER_WRITER ? "RW" : "IOS");
aoqi@0 133 return relativeBase.replace("foo", "foo" + suffix);
aoqi@0 134 }
aoqi@0 135 }
aoqi@0 136

mercurial