[USERGRID-333] Adding convenience methods to post token with query pa…

…rameters
buondoi · Feb 18, 2015 · 07a5714 · 07a5714
1 parent 7865f9a
commit 07a5714
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/.../src/test/java/org/apache/usergrid/rest/test/resource2point0/endpoints/TokenResource.java b/.../src/test/java/org/apache/usergrid/rest/test/resource2point0/endpoints/TokenResource.java
@@ -22,6 +22,8 @@
 
 
 
+import com.sun.jersey.api.client.WebResource;
+import org.apache.usergrid.rest.test.resource2point0.model.QueryParameters;
 import org.apache.usergrid.rest.test.resource2point0.model.Token;
 import org.apache.usergrid.rest.test.resource2point0.state.ClientContext;
 
@@ -36,6 +38,22 @@ public TokenResource(final ClientContext context, final UrlResource parent) {
     }
 
 
+
+    /**
+     * Obtains an access token and sets the token for the context to use in later calls
+     *
+     * @return
+     */
+    public Token post(QueryParameters params) {
+        WebResource resource  = getResource();
+        resource = addParametersToResource(resource, params);
+        Token token = resource.type(MediaType.APPLICATION_JSON_TYPE).accept(MediaType.APPLICATION_JSON)
+            .get(Token.class);
+
+        this.context.setToken(token);
+        return token;
+    }
+
     /**
      * Obtains an access token and sets the token for the context to use in later calls
      *

diff --git a/...test/java/org/apache/usergrid/rest/test/resource2point0/endpoints/mgmt/TokenResource.java b/...test/java/org/apache/usergrid/rest/test/resource2point0/endpoints/mgmt/TokenResource.java
@@ -22,6 +22,7 @@
 import com.sun.jersey.api.client.WebResource;
 import org.apache.usergrid.rest.test.resource2point0.endpoints.NamedResource;
 import org.apache.usergrid.rest.test.resource2point0.endpoints.UrlResource;
+import org.apache.usergrid.rest.test.resource2point0.model.ApiResponse;
 import org.apache.usergrid.rest.test.resource2point0.model.QueryParameters;
 import org.apache.usergrid.rest.test.resource2point0.model.Token;
 import org.apache.usergrid.rest.test.resource2point0.state.ClientContext;
@@ -48,6 +49,21 @@ public Token post() {
         return token;
     }
 
+    /**
+     * Obtains an access token and sets the token for the context to use in later calls
+     *
+     * @return
+     */
+    public Token post(QueryParameters params) {
+        WebResource resource  = getResource();
+        resource = addParametersToResource(resource, params);
+        Token token = resource.type(MediaType.APPLICATION_JSON_TYPE).accept(MediaType.APPLICATION_JSON)
+            .get(Token.class);
+
+        this.context.setToken(token);
+        return token;
+    }
+
     /**
      * Obtains an access token and sets the token for the context to use in later calls
      *