From 1d85435fe8d947585ace80ae25969f70a975124f Mon Sep 17 00:00:00 2001
From: Adrian Cole <adrian@jclouds.org>
Date: Mon, 16 Apr 2012 11:14:13 -0700
Subject: [PATCH] missing unwrap class arg

---
 core/src/main/java/org/jclouds/Wrapper.java            | 10 ++++++++--
 .../main/java/org/jclouds/internal/BaseWrapper.java    |  7 ++++++-
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/core/src/main/java/org/jclouds/Wrapper.java b/core/src/main/java/org/jclouds/Wrapper.java
index e5a740a46c..d80839f894 100644
--- a/core/src/main/java/org/jclouds/Wrapper.java
+++ b/core/src/main/java/org/jclouds/Wrapper.java
@@ -51,8 +51,14 @@ public interface Wrapper {
     *            if the wrapped context is not assignable from the specified class.
     * @see #getWrappedType()
     */
-   <C extends Closeable> C unwrap(TypeToken<C> type);
-
+   <C extends Closeable> C unwrap(TypeToken<C> type) throws IllegalArgumentException;
+   
+   /**
+    * shortcut for {@code unwrap(TypeToken.of(clazz))}
+    * @see #unwrap(TypeToken)
+    */
+   <C extends Closeable> C unwrap(Class<C> clazz) throws IllegalArgumentException;
+   
    /**
     * shortcut for {@code unwrap(getWrappedType())}
     * 
diff --git a/core/src/main/java/org/jclouds/internal/BaseWrapper.java b/core/src/main/java/org/jclouds/internal/BaseWrapper.java
index b28208c5f9..d93ac915a7 100644
--- a/core/src/main/java/org/jclouds/internal/BaseWrapper.java
+++ b/core/src/main/java/org/jclouds/internal/BaseWrapper.java
@@ -53,7 +53,12 @@ public abstract class BaseWrapper implements Wrapper {
       checkArgument(checkNotNull(type, "type").isAssignableFrom(wrappedType), "wrapped type: %s not assignable from %s", wrappedType, type);
       return (C) wrapped;
    }
-
+   
+   @Override
+   public <C extends Closeable> C unwrap(Class<C> clazz) {
+      return unwrap (TypeToken.of(checkNotNull(clazz, "clazz")));
+   }
+   
    @Override
    public TypeToken<? extends Closeable> getWrappedType() {
       return wrappedType;