Arvados-DCO-1.1-Signed-off-by: Radhika Chippada <radhika@curoverse.com>
[arvados.git] / sdk / java / src / test / java / org / arvados / sdk / java / ArvadosTest.java
index 9931d183265da558cfd39ccf3147bca57bc82b4d..9de894669d847e62a8d4074cb2426f35da79e7fb 100644 (file)
@@ -1,47 +1,30 @@
-package org.arvados.sdk.java;
+// Copyright (C) The Arvados Authors. All rights reserved.
+//
+// SPDX-License-Identifier: Apache-2.0
+
+package org.arvados.sdk;
 
 import java.io.File;
+import java.io.FileInputStream;
+import java.math.BigDecimal;
+import java.util.ArrayList;
+import java.util.Calendar;
+import java.util.Date;
+import java.util.GregorianCalendar;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
 
 import org.junit.Test;
 
 import static org.junit.Assert.*;
 
-import com.google.api.services.discovery.model.RestDescription;
-import com.google.api.services.discovery.model.RestResource;
-
-import org.json.simple.JSONObject;
-import org.json.simple.parser.JSONParser;
-
 /**
  * Unit test for Arvados.
  */
 public class ArvadosTest {
 
-  /**
-   * test discover method
-   * @throws Exception
-   */
-  @Test
-  public void testDiscover() throws Exception {
-    Arvados arv = new Arvados("arvados", "v1");
-
-    RestDescription restDescription = arv.discover();
-
-    // The discover method returns the supported methods
-    Map<String, RestResource> resources = restDescription.getResources();
-    assertNotNull("Expected resources", resources);
-
-    Object users = resources.get("users");
-    assertNotNull ("Expected users.list method", users);
-    assertEquals("Exepcted users.list to be a RestResource type", RestResource.class, users.getClass());
-
-    assertTrue("Root URL expected to match ARVADOS_API_HOST env paramdeter", 
-        restDescription.getRootUrl().contains(System.getenv().get("ARVADOS_API_HOST")));
-  }
-
   /**
    * Test users.list api
    * @throws Exception
@@ -52,21 +35,14 @@ public class ArvadosTest {
 
     Map<String, Object> params = new HashMap<String, Object>();
 
-    String response = arv.call("users", "list", params);
-    assertTrue("Expected users.list in response", response.contains("arvados#userList"));
-    assertTrue("Expected users.list in response", response.contains("uuid"));
-
-    JSONParser parser = new JSONParser();
-    Object obj = parser.parse(response);
-    JSONObject jsonObject = (JSONObject) obj;
+    Map response = arv.call("users", "list", params);
+    assertEquals("Expected kind to be users.list", "arvados#userList", response.get("kind"));
 
-    assertEquals("Expected kind to be users.list", "arvados#userList", jsonObject.get("kind"));
-
-    List items = (List)jsonObject.get("items");
+    List items = (List)response.get("items");
     assertNotNull("expected users list items", items);
     assertTrue("expected at least one item in users list", items.size()>0);
 
-    JSONObject firstUser = (JSONObject)items.get(0);
+    Map firstUser = (Map)items.get(0);
     assertNotNull ("Expcted at least one user", firstUser);
 
     assertEquals("Expected kind to be user", "arvados#user", firstUser.get("kind"));
@@ -84,15 +60,13 @@ public class ArvadosTest {
     // call user.system and get uuid of this user
     Map<String, Object> params = new HashMap<String, Object>();
 
-    String response = arv.call("users", "list", params);
-    JSONParser parser = new JSONParser();
-    Object obj = parser.parse(response);
-    JSONObject jsonObject = (JSONObject) obj;
-    assertNotNull("expected users list", jsonObject);
-    List items = (List)jsonObject.get("items");
+    Map response = arv.call("users", "list", params);
+
+    assertNotNull("expected users list", response);
+    List items = (List)response.get("items");
     assertNotNull("expected users list items", items);
 
-    JSONObject firstUser = (JSONObject)items.get(0);
+    Map firstUser = (Map)items.get(0);
     String userUuid = (String)firstUser.get("uuid");
 
     // invoke users.get with the system user uuid
@@ -101,10 +75,8 @@ public class ArvadosTest {
 
     response = arv.call("users", "get", params);
 
-    //JSONParser parser = new JSONParser();
-    jsonObject = (JSONObject) parser.parse(response);;
-    assertNotNull("Expected uuid for first user", jsonObject.get("uuid"));
-    assertEquals("Expected system user uuid", userUuid, jsonObject.get("uuid"));
+    assertNotNull("Expected uuid for first user", response.get("uuid"));
+    assertEquals("Expected system user uuid", userUuid, response.get("uuid"));
   }
 
   /**
@@ -115,17 +87,33 @@ public class ArvadosTest {
   public void testCreateUser() throws Exception {
     Arvados arv = new Arvados("arvados", "v1");
 
-    File file = new File(getClass().getResource( "/create_user.json" ).toURI());
-    String filePath = file.getPath();
-
     Map<String, Object> params = new HashMap<String, Object>();
     params.put("user", "{}");
-    String response = arv.call("users", "create", params);
+    Map response = arv.call("users", "create", params);
+
+    assertEquals("Expected kind to be user", "arvados#user", response.get("kind"));
+
+    Object uuid = response.get("uuid");
+    assertNotNull("Expected uuid for first user", uuid);
+
+    // delete the object
+    params = new HashMap<String, Object>();
+    params.put("uuid", uuid);
+    response = arv.call("users", "delete", params);
+
+    // invoke users.get with the system user uuid
+    params = new HashMap<String, Object>();
+    params.put("uuid", uuid);
 
-    JSONParser parser = new JSONParser();
-    JSONObject jsonObject = (JSONObject) parser.parse(response);
-    assertEquals("Expected kind to be user", "arvados#user", jsonObject.get("kind"));
-    assertNotNull("Expected uuid for first user", jsonObject.get("uuid"));
+    Exception caught = null;
+    try {
+      arv.call("users", "get", params);
+    } catch (Exception e) {
+      caught = e;
+    }
+
+    assertNotNull ("expected exception", caught);
+    assertTrue ("Expected 404", caught.getMessage().contains("Path not found"));
   }
 
   @Test
@@ -156,27 +144,28 @@ public class ArvadosTest {
 
     Map<String, Object> params = new HashMap<String, Object>();
     params.put("user", "{}");
-    String response = arv.call("users", "create", params);
+    Map response = arv.call("users", "create", params);
 
-    JSONParser parser = new JSONParser();
-    JSONObject jsonObject = (JSONObject) parser.parse(response);
-    assertEquals("Expected kind to be user", "arvados#user", jsonObject.get("kind"));
-    
-    Object uuid = jsonObject.get("uuid");
+    assertEquals("Expected kind to be user", "arvados#user", response.get("kind"));
+
+    Object uuid = response.get("uuid");
     assertNotNull("Expected uuid for first user", uuid);
-    
+
     // update this user
     params = new HashMap<String, Object>();
     params.put("user", "{}");
     params.put("uuid", uuid);
     response = arv.call("users", "update", params);
 
-    parser = new JSONParser();
-    jsonObject = (JSONObject) parser.parse(response);
-    assertEquals("Expected kind to be user", "arvados#user", jsonObject.get("kind"));
-    
-    uuid = jsonObject.get("uuid");
+    assertEquals("Expected kind to be user", "arvados#user", response.get("kind"));
+
+    uuid = response.get("uuid");
     assertNotNull("Expected uuid for first user", uuid);
+
+    // delete the object
+    params = new HashMap<String, Object>();
+    params.put("uuid", uuid);
+    response = arv.call("users", "delete", params);
   }
 
   /**
@@ -185,11 +174,9 @@ public class ArvadosTest {
    */
   @Test
   public void testUnsupportedApiName() throws Exception {
-    Arvados arv = new Arvados("not_arvados", "v1");
-
     Exception caught = null;
     try {
-      arv.call("users", "list", null);
+      Arvados arv = new Arvados("not_arvados", "v1");
     } catch (Exception e) {
       caught = e;
     }
@@ -204,11 +191,9 @@ public class ArvadosTest {
    */
   @Test
   public void testUnsupportedVersion() throws Exception {
-    Arvados arv = new Arvados("arvados", "v2");
-
     Exception caught = null;
     try {
-      arv.call("users", "list", null);
+      Arvados arv = new Arvados("arvados", "v2");
     } catch (Exception e) {
       caught = e;
     }
@@ -216,7 +201,7 @@ public class ArvadosTest {
     assertNotNull ("expected exception", caught);
     assertTrue ("Expected 404 when unsupported version is used", caught.getMessage().contains("404 Not Found"));
   }
-  
+
   /**
    * Test unsupported api version api
    * @throws Exception
@@ -235,7 +220,7 @@ public class ArvadosTest {
     assertNotNull ("expected exception", caught);
     assertTrue ("Expected ERROR: 404 not found", caught.getMessage().contains("ERROR: resource not found"));
   }
-  
+
   /**
    * Test unsupported api version api
    * @throws Exception
@@ -264,27 +249,34 @@ public class ArvadosTest {
     Arvados arv = new Arvados("arvados", "v1");
 
     File file = new File(getClass().getResource( "/first_pipeline.json" ).toURI());
-    String filePath = file.getPath();
+    byte[] data = new byte[(int)file.length()];
+    try {
+      FileInputStream is = new FileInputStream(file);
+      is.read(data);
+      is.close();
+    }catch(Exception e) {
+      e.printStackTrace();
+    }
 
     Map<String, Object> params = new HashMap<String, Object>();
-    params.put("pipeline_template", "{}");                          // TBD - read file and send
-    String response = arv.call("pipeline_templates", "create", params);
-
-    JSONParser parser = new JSONParser();
-    JSONObject jsonObject = (JSONObject) parser.parse(response);
-    assertEquals("Expected kind to be user", "arvados#pipelineTemplate", jsonObject.get("kind"));
-    String uuid = (String)jsonObject.get("uuid");
+    params.put("pipeline_template", new String(data));
+    Map response = arv.call("pipeline_templates", "create", params);
+    assertEquals("Expected kind to be user", "arvados#pipelineTemplate", response.get("kind"));
+    String uuid = (String)response.get("uuid");
     assertNotNull("Expected uuid for pipeline template", uuid);
-    
+
     // get the pipeline
     params = new HashMap<String, Object>();
     params.put("uuid", uuid);
     response = arv.call("pipeline_templates", "get", params);
 
-    parser = new JSONParser();
-    jsonObject = (JSONObject) parser.parse(response);
-    assertEquals("Expected kind to be user", "arvados#pipelineTemplate", jsonObject.get("kind"));
-    assertEquals("Expected uuid for pipeline template", uuid, jsonObject.get("uuid"));
+    assertEquals("Expected kind to be user", "arvados#pipelineTemplate", response.get("kind"));
+    assertEquals("Expected uuid for pipeline template", uuid, response.get("uuid"));
+
+    // delete the object
+    params = new HashMap<String, Object>();
+    params.put("uuid", uuid);
+    response = arv.call("pipeline_templates", "delete", params);
   }
 
   /**
@@ -301,14 +293,8 @@ public class ArvadosTest {
 
     Map<String, Object> params = new HashMap<String, Object>();
 
-    String response = arv.call("users", "list", params);
-    assertTrue("Expected users.list in response", response.contains("arvados#userList"));
-    assertTrue("Expected users.list in response", response.contains("uuid"));
-
-    JSONParser parser = new JSONParser();
-    Object obj = parser.parse(response);
-    JSONObject jsonObject = (JSONObject) obj;
-    assertEquals("Expected kind to be users.list", "arvados#userList", jsonObject.get("kind"));
+    Map response = arv.call("users", "list", params);
+    assertEquals("Expected kind to be users.list", "arvados#userList", response.get("kind"));
   }
 
   /**
@@ -321,17 +307,10 @@ public class ArvadosTest {
 
     Map<String, Object> params = new HashMap<String, Object>();
 
-    String response = arv.call("users", "list", params);
-    assertTrue("Expected users.list in response", response.contains("arvados#userList"));
-    assertTrue("Expected users.list in response", response.contains("uuid"));
-
-    JSONParser parser = new JSONParser();
-    Object obj = parser.parse(response);
-    JSONObject jsonObject = (JSONObject) obj;
-
-    assertEquals("Expected kind to be users.list", "arvados#userList", jsonObject.get("kind"));
+    Map response = arv.call("users", "list", params);
+    assertEquals("Expected users.list in response", "arvados#userList", response.get("kind"));
 
-    List items = (List)jsonObject.get("items");
+    List items = (List)response.get("items");
     assertNotNull("expected users list items", items);
     assertTrue("expected at least one item in users list", items.size()>0);
 
@@ -340,16 +319,12 @@ public class ArvadosTest {
     // make the request again with limit
     params = new HashMap<String, Object>();
     params.put("limit", numUsersListItems-1);
-    
-    response = arv.call("users", "list", params);
 
-    parser = new JSONParser();
-    obj = parser.parse(response);
-    jsonObject = (JSONObject) obj;
+    response = arv.call("users", "list", params);
 
-    assertEquals("Expected kind to be users.list", "arvados#userList", jsonObject.get("kind"));
+    assertEquals("Expected kind to be users.list", "arvados#userList", response.get("kind"));
 
-    items = (List)jsonObject.get("items");
+    items = (List)response.get("items");
     assertNotNull("expected users list items", items);
     assertTrue("expected at least one item in users list", items.size()>0);
 
@@ -357,4 +332,136 @@ public class ArvadosTest {
     assertEquals ("Got more users than requested", numUsersListItems-1, numUsersListItems2);
   }
 
-}
\ No newline at end of file
+  @Test
+  public void testGetLinksWithFilters() throws Exception {
+    Arvados arv = new Arvados("arvados", "v1");
+
+    Map<String, Object> params = new HashMap<String, Object>();
+
+    Map response = arv.call("links", "list", params);
+    assertEquals("Expected links.list in response", "arvados#linkList", response.get("kind"));
+
+    String[][] filters = new String[1][];
+    String[] condition = new String[3];
+    condition[0] = "name";
+    condition[1] = "=";
+    condition[2] = "can_manage";
+    filters[0] = condition;
+    params.put("filters", filters);
+    
+    response = arv.call("links", "list", params);
+    
+    assertEquals("Expected links.list in response", "arvados#linkList", response.get("kind"));
+    assertFalse("Expected no can_manage in response", response.toString().contains("\"name\":\"can_manage\""));
+  }
+
+  @Test
+  public void testGetLinksWithFiltersAsList() throws Exception {
+    Arvados arv = new Arvados("arvados", "v1");
+
+    Map<String, Object> params = new HashMap<String, Object>();
+
+    Map response = arv.call("links", "list", params);
+    assertEquals("Expected links.list in response", "arvados#linkList", response.get("kind"));
+
+    List<List> filters = new ArrayList<List>();
+    List<String> condition = new ArrayList<String>();
+    condition.add("name");
+    condition.add("is_a");
+    condition.add("can_manage");
+    filters.add(condition);
+    params.put("filters", filters);
+    
+    response = arv.call("links", "list", params);
+    
+    assertEquals("Expected links.list in response", "arvados#linkList", response.get("kind"));
+    assertFalse("Expected no can_manage in response", response.toString().contains("\"name\":\"can_manage\""));
+  }
+
+  @Test
+  public void testGetLinksWithTimestampFilters() throws Exception {
+    Arvados arv = new Arvados("arvados", "v1");
+
+    Map<String, Object> params = new HashMap<String, Object>();
+
+    Map response = arv.call("links", "list", params);
+    assertEquals("Expected links.list in response", "arvados#linkList", response.get("kind"));
+
+    // get links created "tomorrow". Expect none in response
+    Calendar calendar = new GregorianCalendar();
+    calendar.setTime(new Date());
+    calendar.add(Calendar.DAY_OF_MONTH, 1);
+    
+    Object[][] filters = new Object[1][];
+    Object[] condition = new Object[3];
+    condition[0] = "created_at";
+    condition[1] = ">";
+    condition[2] = calendar.get(Calendar.YEAR) + "-" + (calendar.get(Calendar.MONTH)+1) + "-" + calendar.get(Calendar.DAY_OF_MONTH);
+    filters[0] = condition;
+    params.put("filters", filters);
+    
+    response = arv.call("links", "list", params);
+    
+    assertEquals("Expected links.list in response", "arvados#linkList", response.get("kind"));
+    int items_avail = ((BigDecimal)response.get("items_available")).intValue();
+    assertEquals("Expected zero links", items_avail, 0);
+  }
+
+  @Test
+  public void testGetLinksWithWhereClause() throws Exception {
+    Arvados arv = new Arvados("arvados", "v1");
+
+    Map<String, Object> params = new HashMap<String, Object>();
+
+    Map<String, String> where = new HashMap<String, String>();
+    where.put("where", "updated_at > '2014-05-01'");
+    
+    params.put("where", where);
+    
+    Map response = arv.call("links", "list", params);
+    
+    assertEquals("Expected links.list in response", "arvados#linkList", response.get("kind"));
+  }
+
+  @Test
+  public void testGetAvailableResources() throws Exception {
+    Arvados arv = new Arvados("arvados", "v1");
+    Set<String> resources = arv.getAvailableResourses();
+    assertNotNull("Expected resources", resources);
+    assertTrue("Excected users in resrouces", resources.contains("users"));
+  }
+
+  @Test
+  public void testGetAvailableMethodsResources() throws Exception {
+    Arvados arv = new Arvados("arvados", "v1");
+    Set<String> methods = arv.getAvailableMethodsForResourse("users");
+    assertNotNull("Expected resources", methods);
+    assertTrue("Excected create method for users", methods.contains("create"));
+  }
+
+  @Test
+  public void testGetAvailableParametersForUsersGetMethod() throws Exception {
+    Arvados arv = new Arvados("arvados", "v1");
+    Map<String,List<String>> parameters = arv.getAvailableParametersForMethod("users", "get");
+    assertNotNull("Expected parameters", parameters);
+    assertTrue("Excected uuid parameter for get method for users", parameters.get("required").contains("uuid"));
+  }
+
+  @Test
+  public void testGetAvailableParametersForUsersCreateMethod() throws Exception {
+    Arvados arv = new Arvados("arvados", "v1");
+    Map<String,List<String>> parameters = arv.getAvailableParametersForMethod("users", "create");
+    assertNotNull("Expected parameters", parameters);
+    assertTrue("Excected user parameter for get method for users", parameters.get("required").contains("user"));
+  }
+
+  @Test
+  public void testGetAvailableParametersForUsersListMethod() throws Exception {
+    Arvados arv = new Arvados("arvados", "v1");
+    Map<String,List<String>> parameters = arv.getAvailableParametersForMethod("users", "list");
+    assertNotNull("Expected parameters", parameters);
+    assertTrue("Excected no required parameter for list method for users", parameters.get("required").size() == 0);
+    assertTrue("Excected some optional parameters for list method for users", parameters.get("optional").contains("filters"));
+  }
+
+}