1/*
2 * Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24/*
25 * @test 8173845
26 * @summary test custom file managers
27 * @build KullaTesting TestingInputStream
28 * @run testng FileManagerTest
29 */
30
31
32import java.io.File;
33import java.io.IOException;
34
35import java.util.Set;
36import javax.tools.ForwardingJavaFileManager;
37import javax.tools.JavaFileObject;
38import javax.tools.JavaFileObject.Kind;
39import javax.tools.StandardJavaFileManager;
40import org.testng.annotations.BeforeMethod;
41import org.testng.annotations.Test;
42
43import static org.testng.Assert.assertTrue;
44
45@Test
46public class FileManagerTest extends KullaTesting {
47
48    boolean encountered;
49
50    class MyFileManager extends ForwardingJavaFileManager<StandardJavaFileManager>
51            implements StandardJavaFileManager {
52
53        protected MyFileManager(StandardJavaFileManager fileManager) {
54            super(fileManager);
55        }
56
57        @Override
58        public Iterable<JavaFileObject> list(Location location,
59                String packageName,
60                Set<Kind> kinds,
61                boolean recurse)
62                throws IOException {
63            //System.out.printf("list(%s, %s, %s, %b)\n",
64            //        location, packageName, kinds, recurse);
65            if (packageName.equals("java.lang.reflect")) {
66                encountered = true;
67            }
68            return fileManager.list(location, packageName, kinds, recurse);
69        }
70
71        @Override
72        public Iterable<? extends JavaFileObject> getJavaFileObjectsFromFiles(Iterable<? extends File> files) {
73            return fileManager.getJavaFileObjectsFromFiles(files);
74        }
75
76        @Override
77        public Iterable<? extends JavaFileObject> getJavaFileObjects(File... files) {
78            return fileManager.getJavaFileObjects(files);
79        }
80
81        @Override
82        public Iterable<? extends JavaFileObject> getJavaFileObjectsFromStrings(Iterable<String> names) {
83            return fileManager.getJavaFileObjectsFromStrings(names);
84        }
85
86        @Override
87        public Iterable<? extends JavaFileObject> getJavaFileObjects(String... names) {
88            return fileManager.getJavaFileObjects(names);
89        }
90
91        @Override
92        public void setLocation(Location location, Iterable<? extends File> files) throws IOException {
93            fileManager.setLocation(location, files);
94        }
95
96        @Override
97        public Iterable<? extends File> getLocation(Location location) {
98            return fileManager.getLocation(location);
99        }
100
101    }
102
103    @BeforeMethod
104    @Override
105    public void setUp() {
106        setUp(b -> b.fileManager(fm -> new MyFileManager(fm)));
107    }
108
109    public void testSnippetMemberAssignment() {
110        assertEval("java.lang.reflect.Array.get(new String[1], 0) == null");
111        assertTrue(encountered, "java.lang.reflect not encountered");
112    }
113
114}
115