hotspot/test/compiler/aot/jdk.tools.jaotc.test/src/jdk/tools/jaotc/test/collect/FakeFileSupport.java

/*
 * Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */
package jdk.tools.jaotc.test.collect;

import java.net.MalformedURLException;
import java.nio.file.Path;
import java.util.HashSet;
import java.util.Set;

public class FakeFileSupport extends FileSupport {
    private final Set<String> exists = new HashSet<>();
    private final Set<String> directories = new HashSet<>();

    private final Set<String> checkedExists = new HashSet<>();
    private final Set<String> checkedDirectory = new HashSet<>();
    private final Set<String> checkedJarFileSystemRoots = new HashSet<>();
    private final Set<String> classloaderPaths = new HashSet<>();

    private Path jarFileSystemRoot = null;
    private final ClassLoader classLoader;

    public FakeFileSupport(Set<String> existing, Set<String> directories) {
        this.exists.addAll(existing);
        this.directories.addAll(directories);

        classLoader = new ClassLoader() {
            @Override
            public Class<?> loadClass(String name) throws ClassNotFoundException {
                return null;
            }
        };
    }

    public void setJarFileSystemRoot(Path path) {
        jarFileSystemRoot = path;
    }

    @Override
    public boolean exists(Path path) {
        checkedExists.add(path.toString());
        return exists.contains(path.toString());
    }

    @Override
    public boolean isDirectory(Path path) {
        checkedDirectory.add(path.toString());
        return directories.contains(path.toString());
    }

    @Override
    public ClassLoader createClassLoader(Path path) throws MalformedURLException {
        classloaderPaths.add(path.toString());
        return classLoader;
    }

    @Override
    public Path getJarFileSystemRoot(Path jarFile) {
        checkedJarFileSystemRoots.add(jarFile.toString());
        return jarFileSystemRoot;
    }

    @Override
    public boolean isAbsolute(Path entry) {
        return entry.toString().startsWith("/");
    }

    public void addExist(String name) {
        exists.add(name);
    }

    public void addDirectory(String name) {
        directories.add(name);
    }

    public Set<String> getCheckedExists() {
        return checkedExists;
    }

    public Set<String> getCheckedDirectory() {
        return checkedDirectory;
    }

    public Set<String> getCheckedJarFileSystemRoots() {
        return checkedJarFileSystemRoots;
    }

    public Set<String> getClassloaderPaths() {
        return classloaderPaths;
    }
}