1 files changed, 227 insertions, 0 deletions
diff --git a/jimfs/src/test/java/com/google/common/jimfs/TestAttributeProvider.java b/jimfs/src/test/java/com/google/common/jimfs/TestAttributeProvider.java
new file mode 100644
index 0000000..4518132
--- /dev/null
+++ b/jimfs/src/test/java/com/google/common/jimfs/TestAttributeProvider.java
@@ -0,0 +1,227 @@
+/*
+ * Copyright 2013 Google Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.common.jimfs;
+
+import static com.google.common.base.Preconditions.checkNotNull;
+
+import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.ImmutableSet;
+import java.io.IOException;
+import java.nio.file.attribute.BasicFileAttributeView;
+import java.nio.file.attribute.FileAttributeView;
+import java.nio.file.attribute.FileTime;
+import java.util.HashMap;
+import java.util.Map;
+import org.checkerframework.checker.nullness.compatqual.NullableDecl;
+
+/** @author Colin Decker */
+public final class TestAttributeProvider extends AttributeProvider {
+
+  private static final ImmutableSet<String> ATTRIBUTES = ImmutableSet.of("foo", "bar", "baz");
+
+  @Override
+  public String name() {
+    return "test";
+  }
+
+  @Override
+  public ImmutableSet<String> inherits() {
+    return ImmutableSet.of("basic");
+  }
+
+  @Override
+  public ImmutableSet<String> fixedAttributes() {
+    return ATTRIBUTES;
+  }
+
+  @Override
+  public ImmutableMap<String, ?> defaultValues(Map<String, ?> userDefaults) {
+    Map<String, Object> result = new HashMap<>();
+
+    Long bar = 0L;
+    Integer baz = 1;
+    if (userDefaults.containsKey("test:bar")) {
+      bar = checkType("test", "bar", userDefaults.get("test:bar"), Number.class).longValue();
+    }
+    if (userDefaults.containsKey("test:baz")) {
+      baz = checkType("test", "baz", userDefaults.get("test:baz"), Integer.class);
+    }
+
+    result.put("test:bar", bar);
+    result.put("test:baz", baz);
+    return ImmutableMap.copyOf(result);
+  }
+
+  @Override
+  public void set(File file, String view, String attribute, Object value, boolean create) {
+    switch (attribute) {
+      case "bar":
+        checkNotCreate(view, attribute, create);
+        file.setAttribute(
+            "test", "bar", checkType(view, attribute, value, Number.class).longValue());
+        break;
+      case "baz":
+        file.setAttribute("test", "baz", checkType(view, attribute, value, Integer.class));
+        break;
+      default:
+        throw unsettable(view, attribute, create);
+    }
+  }
+
+  @Override
+  public Object get(File file, String attribute) {
+    if (attribute.equals("foo")) {
+      return "hello";
+    }
+    return file.getAttribute("test", attribute);
+  }
+
+  @Override
+  public Class<TestAttributeView> viewType() {
+    return TestAttributeView.class;
+  }
+
+  @Override
+  public TestAttributeView view(
+      FileLookup lookup, ImmutableMap<String, FileAttributeView> inheritedViews) {
+    return new View(lookup, (BasicFileAttributeView) inheritedViews.get("basic"));
+  }
+
+  @Override
+  public Class<TestAttributes> attributesType() {
+    return TestAttributes.class;
+  }
+
+  @Override
+  public TestAttributes readAttributes(File file) {
+    return new Attributes(file);
+  }
+
+  static final class View implements TestAttributeView {
+
+    private final FileLookup lookup;
+    private final BasicFileAttributeView basicView;
+
+    public View(FileLookup lookup, BasicFileAttributeView basicView) {
+      this.lookup = checkNotNull(lookup);
+      this.basicView = checkNotNull(basicView);
+    }
+
+    @Override
+    public String name() {
+      return "test";
+    }
+
+    @Override
+    public Attributes readAttributes() throws IOException {
+      return new Attributes(lookup.lookup());
+    }
+
+    @Override
+    public void setTimes(
+        @NullableDecl FileTime lastModifiedTime,
+        @NullableDecl FileTime lastAccessTime,
+        @NullableDecl FileTime createTime)
+        throws IOException {
+      basicView.setTimes(lastModifiedTime, lastAccessTime, createTime);
+    }
+
+    @Override
+    public void setBar(long bar) throws IOException {
+      lookup.lookup().setAttribute("test", "bar", bar);
+    }
+
+    @Override
+    public void setBaz(int baz) throws IOException {
+      lookup.lookup().setAttribute("test", "baz", baz);
+    }
+  }
+
+  static final class Attributes implements TestAttributes {
+
+    private final Long bar;
+    private final Integer baz;
+
+    public Attributes(File file) {
+      this.bar = (Long) file.getAttribute("test", "bar");
+      this.baz = (Integer) file.getAttribute("test", "baz");
+    }
+
+    @Override
+    public String foo() {
+      return "hello";
+    }
+
+    @Override
+    public long bar() {
+      return bar;
+    }
+
+    @Override
+    public int baz() {
+      return baz;
+    }
+
+    // BasicFileAttributes is just implemented here because readAttributes requires a subtype of
+    // BasicFileAttributes -- methods are not implemented
+
+    @Override
+    public FileTime lastModifiedTime() {
+      return null;
+    }
+
+    @Override
+    public FileTime lastAccessTime() {
+      return null;
+    }
+
+    @Override
+    public FileTime creationTime() {
+      return null;
+    }
+
+    @Override
+    public boolean isRegularFile() {
+      return false;
+    }
+
+    @Override
+    public boolean isDirectory() {
+      return false;
+    }
+
+    @Override
+    public boolean isSymbolicLink() {
+      return false;
+    }
+
+    @Override
+    public boolean isOther() {
+      return false;
+    }
+
+    @Override
+    public long size() {
+      return 0;
+    }
+
+    @Override
+    public Object fileKey() {
+      return null;
+    }
+  }
+}