test/java/nio/file/attribute/UserDefinedFileAttributeView/Basic.java - toolchain/jdk/jdk9_jdk - Git at Google

 /*
  * Copyright (c) 2008, 2009, 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.
  */

 /* @test
  * @bug 4313887 6838333
  * @summary Unit test for java.nio.file.attribute.UserDefinedFileAttributeView
  * @library ../..
  */

 import java.nio.ByteBuffer;
 import java.nio.charset.Charset;
 import java.nio.file.*;
 import static java.nio.file.LinkOption.*;
 import java.nio.file.attribute.*;
 import java.util.Arrays;
 import java.util.Map;
 import java.util.Random;
 import java.io.IOException;

 public class Basic {

     private static Random rand = new Random();

     private static final String ATTR_NAME = "mime_type";
     private static final String ATTR_VALUE = "text/plain";
     private static final String ATTR_VALUE2 = "text/html";

     static interface Task {
         void run() throws Exception;
     }

     static void tryCatch(Class<? extends Throwable> ex, Task task) {
         boolean caught = false;
         try {
             task.run();
         } catch (Throwable x) {
             if (ex.isAssignableFrom(x.getClass())) {
                 caught = true;
             } else {
                 throw new RuntimeException(x);
             }
         }
         if (!caught)
             throw new RuntimeException(ex.getName() + " expected");
     }

     static void expectNullPointerException(Task task) {
         tryCatch(NullPointerException.class, task);
     }

     static boolean hasAttribute(UserDefinedFileAttributeView view, String attr)
         throws IOException
     {
         for (String name: view.list()) {
             if (name.equals(ATTR_NAME))
                 return true;
         }
         return false;
     }

     static void test(Path file, LinkOption... options) throws IOException {
         final UserDefinedFileAttributeView view = file
             .getFileAttributeView(UserDefinedFileAttributeView.class, options);
         ByteBuffer buf = rand.nextBoolean() ?
             ByteBuffer.allocate(100) : ByteBuffer.allocateDirect(100);

         // Test: write
         buf.put(ATTR_VALUE.getBytes()).flip();
         int size = buf.remaining();
         int nwrote = view.write(ATTR_NAME, buf);
         if (nwrote != size)
             throw new RuntimeException("Unexpected number of bytes written");

         // Test: size
         if (view.size(ATTR_NAME) != size)
             throw new RuntimeException("Unexpected size");

         // Test: read
         buf.clear();
         int nread = view.read(ATTR_NAME, buf);
         if (nread != size)
             throw new RuntimeException("Unexpected number of bytes read");
         buf.flip();
         String value = Charset.defaultCharset().decode(buf).toString();
         if (!value.equals(ATTR_VALUE))
             throw new RuntimeException("Unexpected attribute value");

         // Test: read with insufficient space
         tryCatch(IOException.class, new Task() {
             public void run() throws IOException {
                 view.read(ATTR_NAME, ByteBuffer.allocateDirect(1));
             }});

         // Test: replace value
         buf.clear();
         buf.put(ATTR_VALUE2.getBytes()).flip();
         size = buf.remaining();
         view.write(ATTR_NAME, buf);
         if (view.size(ATTR_NAME) != size)
             throw new RuntimeException("Unexpected size");

         // Test: list
         if (!hasAttribute(view, ATTR_NAME))
             throw new RuntimeException("Attribute name not in list");

         // Test: delete
         view.delete(ATTR_NAME);
         if (hasAttribute(view, ATTR_NAME))
             throw new RuntimeException("Attribute name in list");

         // Test: dynamic access
         String name = "user:" + ATTR_NAME;
         byte[] valueAsBytes = ATTR_VALUE.getBytes();
         file.setAttribute(name, valueAsBytes);
         byte[] actualAsBytes = (byte[])file.getAttribute(name);
         if (!Arrays.equals(valueAsBytes, actualAsBytes))
             throw new RuntimeException("Unexpected attribute value");
         Map<String,?> map = file.readAttributes(name);
         if (!Arrays.equals(valueAsBytes, (byte[])map.get(ATTR_NAME)))
             throw new RuntimeException("Unexpected attribute value");
         map = file.readAttributes("user:*");
         if (!Arrays.equals(valueAsBytes, (byte[])map.get(ATTR_NAME)))
             throw new RuntimeException("Unexpected attribute value");
         map = file.readAttributes("user:DoesNotExist");
         if (!map.isEmpty())
             throw new RuntimeException("Map expected to be empty");
     }

     static void miscTests(final Path file) throws IOException {
         final UserDefinedFileAttributeView view = file
             .getFileAttributeView(UserDefinedFileAttributeView.class);
         view.write(ATTR_NAME, ByteBuffer.wrap(ATTR_VALUE.getBytes()));

         // NullPointerException
         final ByteBuffer buf = ByteBuffer.allocate(100);

         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 view.read(null, buf);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 view.read(ATTR_NAME, null);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 view.write(null, buf);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                view.write(ATTR_NAME, null);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 view.size(null);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 view.delete(null);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 file.getAttribute(null);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 file.getAttribute("user:" + ATTR_NAME, (LinkOption[])null);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 file.setAttribute("user:" + ATTR_NAME, null);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 file.setAttribute(null, new byte[0]);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 file.setAttribute("user: " + ATTR_NAME, new byte[0], (LinkOption[])null);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 file.readAttributes((String)null);
             }});
         expectNullPointerException(new Task() {
             public void run() throws IOException {
                 file.readAttributes("*", (LinkOption[])null);
             }});

         // Read-only buffer
         tryCatch(IllegalArgumentException.class, new Task() {
             public void run() throws IOException {
                 ByteBuffer buf = ByteBuffer.wrap(ATTR_VALUE.getBytes()).asReadOnlyBuffer();
                 view.write(ATTR_NAME, buf);
                 buf.flip();
                 view.read(ATTR_NAME, buf);
             }});

         // Zero bytes remaining
         tryCatch(IOException.class, new Task() {
             public void run() throws IOException {
                 ByteBuffer buf = buf = ByteBuffer.allocateDirect(100);
                 buf.position(buf.capacity());
                 view.read(ATTR_NAME, buf);
             }});
     }

     public static void main(String[] args) throws IOException {
         // create temporary directory to run tests
         Path dir = TestUtil.createTemporaryDirectory();
         try {
             if (!dir.getFileStore().supportsFileAttributeView("user")) {
                 System.out.println("UserDefinedFileAttributeView not supported - skip test");
                 return;
             }

             // test access to user defined attributes of regular file
             Path file = dir.resolve("foo.html").createFile();
             try {
                 test(file);
             } finally {
                 file.delete();
             }

             // test access to user define attributes of directory
             file = dir.resolve("foo").createDirectory();
             try {
                 test(file);
             } finally {
                 file.delete();
             }

             // test access to user defined attributes of sym link
             if (TestUtil.supportsLinks(dir)) {
                 Path target = dir.resolve("doesnotexist");
                 Path link = dir.resolve("link").createSymbolicLink(target);
                 try {
                     test(link, NOFOLLOW_LINKS);
                 } catch (IOException x) {
                     // access to attributes of sym link may not be supported
                 } finally {
                     link.delete();
                 }
             }

             // misc. tests
             try {
                 file = dir.resolve("foo.txt").createFile();
                 miscTests(dir);
             } finally {
                 file.delete();
             }

         } finally {
             TestUtil.removeAll(dir);
         }
     }
  }
	/*
	* Copyright (c) 2008, 2009, 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.
	*/

	/* @test
	* @bug 4313887 6838333
	* @summary Unit test for java.nio.file.attribute.UserDefinedFileAttributeView
	* @library ../..
	*/

	import java.nio.ByteBuffer;
	import java.nio.charset.Charset;
	import java.nio.file.*;
	import static java.nio.file.LinkOption.*;
	import java.nio.file.attribute.*;
	import java.util.Arrays;
	import java.util.Map;
	import java.util.Random;
	import java.io.IOException;

	public class Basic {

	private static Random rand = new Random();

	private static final String ATTR_NAME = "mime_type";
	private static final String ATTR_VALUE = "text/plain";
	private static final String ATTR_VALUE2 = "text/html";

	static interface Task {
	void run() throws Exception;
	}

	static void tryCatch(Class<? extends Throwable> ex, Task task) {
	boolean caught = false;
	try {
	task.run();
	} catch (Throwable x) {
	if (ex.isAssignableFrom(x.getClass())) {
	caught = true;
	} else {
	throw new RuntimeException(x);
	}
	}
	if (!caught)
	throw new RuntimeException(ex.getName() + " expected");
	}

	static void expectNullPointerException(Task task) {
	tryCatch(NullPointerException.class, task);
	}

	static boolean hasAttribute(UserDefinedFileAttributeView view, String attr)
	throws IOException
	{
	for (String name: view.list()) {
	if (name.equals(ATTR_NAME))
	return true;
	}
	return false;
	}

	static void test(Path file, LinkOption... options) throws IOException {
	final UserDefinedFileAttributeView view = file
	.getFileAttributeView(UserDefinedFileAttributeView.class, options);
	ByteBuffer buf = rand.nextBoolean() ?
	ByteBuffer.allocate(100) : ByteBuffer.allocateDirect(100);

	// Test: write
	buf.put(ATTR_VALUE.getBytes()).flip();
	int size = buf.remaining();
	int nwrote = view.write(ATTR_NAME, buf);
	if (nwrote != size)
	throw new RuntimeException("Unexpected number of bytes written");

	// Test: size
	if (view.size(ATTR_NAME) != size)
	throw new RuntimeException("Unexpected size");

	// Test: read
	buf.clear();
	int nread = view.read(ATTR_NAME, buf);
	if (nread != size)
	throw new RuntimeException("Unexpected number of bytes read");
	buf.flip();
	String value = Charset.defaultCharset().decode(buf).toString();
	if (!value.equals(ATTR_VALUE))
	throw new RuntimeException("Unexpected attribute value");

	// Test: read with insufficient space
	tryCatch(IOException.class, new Task() {
	public void run() throws IOException {
	view.read(ATTR_NAME, ByteBuffer.allocateDirect(1));
	}});

	// Test: replace value
	buf.clear();
	buf.put(ATTR_VALUE2.getBytes()).flip();
	size = buf.remaining();
	view.write(ATTR_NAME, buf);
	if (view.size(ATTR_NAME) != size)
	throw new RuntimeException("Unexpected size");

	// Test: list
	if (!hasAttribute(view, ATTR_NAME))
	throw new RuntimeException("Attribute name not in list");

	// Test: delete
	view.delete(ATTR_NAME);
	if (hasAttribute(view, ATTR_NAME))
	throw new RuntimeException("Attribute name in list");

	// Test: dynamic access
	String name = "user:" + ATTR_NAME;
	byte[] valueAsBytes = ATTR_VALUE.getBytes();
	file.setAttribute(name, valueAsBytes);
	byte[] actualAsBytes = (byte[])file.getAttribute(name);
	if (!Arrays.equals(valueAsBytes, actualAsBytes))
	throw new RuntimeException("Unexpected attribute value");
	Map<String,?> map = file.readAttributes(name);
	if (!Arrays.equals(valueAsBytes, (byte[])map.get(ATTR_NAME)))
	throw new RuntimeException("Unexpected attribute value");
	map = file.readAttributes("user:*");
	if (!Arrays.equals(valueAsBytes, (byte[])map.get(ATTR_NAME)))
	throw new RuntimeException("Unexpected attribute value");
	map = file.readAttributes("user:DoesNotExist");
	if (!map.isEmpty())
	throw new RuntimeException("Map expected to be empty");
	}

	static void miscTests(final Path file) throws IOException {
	final UserDefinedFileAttributeView view = file
	.getFileAttributeView(UserDefinedFileAttributeView.class);
	view.write(ATTR_NAME, ByteBuffer.wrap(ATTR_VALUE.getBytes()));

	// NullPointerException
	final ByteBuffer buf = ByteBuffer.allocate(100);

	expectNullPointerException(new Task() {
	public void run() throws IOException {
	view.read(null, buf);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	view.read(ATTR_NAME, null);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	view.write(null, buf);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	view.write(ATTR_NAME, null);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	view.size(null);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	view.delete(null);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	file.getAttribute(null);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	file.getAttribute("user:" + ATTR_NAME, (LinkOption[])null);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	file.setAttribute("user:" + ATTR_NAME, null);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	file.setAttribute(null, new byte[0]);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	file.setAttribute("user: " + ATTR_NAME, new byte[0], (LinkOption[])null);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	file.readAttributes((String)null);
	}});
	expectNullPointerException(new Task() {
	public void run() throws IOException {
	file.readAttributes("*", (LinkOption[])null);
	}});

	// Read-only buffer
	tryCatch(IllegalArgumentException.class, new Task() {
	public void run() throws IOException {
	ByteBuffer buf = ByteBuffer.wrap(ATTR_VALUE.getBytes()).asReadOnlyBuffer();
	view.write(ATTR_NAME, buf);
	buf.flip();
	view.read(ATTR_NAME, buf);
	}});

	// Zero bytes remaining
	tryCatch(IOException.class, new Task() {
	public void run() throws IOException {
	ByteBuffer buf = buf = ByteBuffer.allocateDirect(100);
	buf.position(buf.capacity());
	view.read(ATTR_NAME, buf);
	}});
	}

	public static void main(String[] args) throws IOException {
	// create temporary directory to run tests
	Path dir = TestUtil.createTemporaryDirectory();
	try {
	if (!dir.getFileStore().supportsFileAttributeView("user")) {
	System.out.println("UserDefinedFileAttributeView not supported - skip test");
	return;
	}

	// test access to user defined attributes of regular file
	Path file = dir.resolve("foo.html").createFile();
	try {
	test(file);
	} finally {
	file.delete();
	}

	// test access to user define attributes of directory
	file = dir.resolve("foo").createDirectory();
	try {
	test(file);
	} finally {
	file.delete();
	}

	// test access to user defined attributes of sym link
	if (TestUtil.supportsLinks(dir)) {
	Path target = dir.resolve("doesnotexist");
	Path link = dir.resolve("link").createSymbolicLink(target);
	try {
	test(link, NOFOLLOW_LINKS);
	} catch (IOException x) {
	// access to attributes of sym link may not be supported
	} finally {
	link.delete();
	}
	}

	// misc. tests
	try {
	file = dir.resolve("foo.txt").createFile();
	miscTests(dir);
	} finally {
	file.delete();
	}

	} finally {
	TestUtil.removeAll(dir);
	}
	}
	}