/*

 * Copyright 2008-2009 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.com if you need additional information or

 * have any questions.

 */

/* @test

 * @bug 4313887

 * @summary Unit test for java.nio.file.DirectoryStream

 * @library ..

 */

import java.nio.file.*;

import java.util.*;

import java.io.IOException;

public class Basic {

    static boolean found;

    static void doTest(final Path dir) throws IOException {

        DirectoryStream<Path> stream;

        // test that directory is empty

        Files.withDirectory(dir, new FileAction<FileRef>() {

            public void invoke(FileRef entry) {

                throw new RuntimeException("directory not empty");

            }

        });

        // create file in directory

        final Path foo = Paths.get("foo");

        dir.resolve(foo).createFile();

        // iterate over directory and check there is one entry

        found = false;

        Files.withDirectory(dir, new FileAction<Path>() {

            public void invoke(Path entry) {

                if (entry.getName().equals(foo)) {

                    if (found)

                        throw new RuntimeException("entry already found");

                    found = true;

                } else {

                    throw new RuntimeException("entry " + entry.getName() +

                        " not expected");

                }

            }

        });

        if (!found)

            throw new RuntimeException("entry not found");

        // check filtering: f* should match foo

        DirectoryStream.Filter<Path> filter = new DirectoryStream.Filter<Path>() {

            private PathMatcher matcher =

                dir.getFileSystem().getPathMatcher("glob:f*");

            public boolean accept(Path file) {

                return matcher.matches(file);

            }

        };

        Files.withDirectory(dir, filter, new FileAction<Path>() {

            public void invoke(Path entry) {

                if (!entry.getName().equals(foo))

                    throw new RuntimeException("entry not expected");

            }

        });

        // check filtering: z* should not match any files

        filter = new DirectoryStream.Filter<Path>() {

            private PathMatcher matcher =

                dir.getFileSystem().getPathMatcher("glob:z*");

            public boolean accept(Path file) {

                return matcher.matches(file);

            }

        };

        Files.withDirectory(dir, filter, new FileAction<FileRef>() {

            public void invoke(FileRef entry) {

                throw new RuntimeException("no matching entries expected");

            }

        });

        // check that exception or error thrown by filter is not thrown

        // by newDirectoryStream or iterator method.

        stream = dir.newDirectoryStream(new DirectoryStream.Filter<Path>() {

            public boolean accept(Path file) {

                throw new RuntimeException("Should not be visible");

            }

        });

        try {

            stream.iterator();

        } finally {

            stream.close();

        }

        // test NotDirectoryException

        try {

            dir.resolve(foo).newDirectoryStream();

            throw new RuntimeException("NotDirectoryException not thrown");

        } catch (NotDirectoryException x) {

        }

        // test iterator remove method

        stream = dir.newDirectoryStream();

        Iterator<Path> i = stream.iterator();

        while (i.hasNext()) {

            Path entry = i.next();

            if (!entry.getName().equals(foo))

                throw new RuntimeException("entry not expected");

            i.remove();

        }

        stream.close();

        // test IllegalStateException

        stream =  dir.newDirectoryStream();

        i = stream.iterator();

        try {

            stream.iterator();

            throw new RuntimeException("IllegalStateException not thrown as expected");

        } catch (IllegalStateException x) {

        }

        stream.close();

        try {

            stream.iterator();

            throw new RuntimeException("IllegalStateException not thrown as expected");

        } catch (IllegalStateException x) {

        }

        try {

            i.hasNext();

            throw new RuntimeException("ConcurrentModificationException not thrown as expected");

        } catch (ConcurrentModificationException x) {

            Throwable t = x.getCause();

            if (!(t instanceof IllegalStateException))

                throw new RuntimeException("Cause is not IllegalStateException as expected");

        }

        try {

            i.next();

            throw new RuntimeException("IllegalStateException not thrown as expected");

        } catch (ConcurrentModificationException x) {

            Throwable t = x.getCause();

            if (!(t instanceof IllegalStateException))

                throw new RuntimeException("Cause is not IllegalStateException as expected");

        }

    }

    public static void main(String[] args) throws IOException {

        Path dir = TestUtil.createTemporaryDirectory();

        try {

            doTest(dir);

        } finally {

            TestUtil.removeAll(dir);

        }

    }

}