Search in sources :

Example 1 with StorageResource

use of org.apache.airavata.registry.cpi.StorageResource in project airavata by apache.

the class StorageResourceAppCatalogTest method testAddStorageResource.

@Test
public void testAddStorageResource() {
    try {
        StorageResource storageResource = appcatalog.getStorageResource();
        StorageResourceDescription description = new StorageResourceDescription();
        description.setHostName("localhost");
        description.setEnabled(true);
        description.setStorageResourceDescription("testDescription");
        String scpDataMoveId = addSCPDataMovement();
        System.out.println("**** SCP DataMoveId****** :" + scpDataMoveId);
        String gridFTPDataMoveId = addGridFTPDataMovement();
        System.out.println("**** grid FTP DataMoveId****** :" + gridFTPDataMoveId);
        List<DataMovementInterface> dataMovementInterfaces = new ArrayList<DataMovementInterface>();
        DataMovementInterface scpInterface = new DataMovementInterface();
        scpInterface.setDataMovementInterfaceId(scpDataMoveId);
        scpInterface.setDataMovementProtocol(DataMovementProtocol.SCP);
        scpInterface.setPriorityOrder(1);
        DataMovementInterface gridFTPMv = new DataMovementInterface();
        gridFTPMv.setDataMovementInterfaceId(gridFTPDataMoveId);
        gridFTPMv.setDataMovementProtocol(DataMovementProtocol.GridFTP);
        gridFTPMv.setPriorityOrder(2);
        dataMovementInterfaces.add(scpInterface);
        dataMovementInterfaces.add(gridFTPMv);
        description.setDataMovementInterfaces(dataMovementInterfaces);
        String resourceId = storageResource.addStorageResource(description);
        System.out.println("**********Resource id ************* : " + resourceId);
        StorageResourceDescription storageResourceDescription = null;
        if (storageResource.isStorageResourceExists(resourceId)) {
            storageResourceDescription = storageResource.getStorageResource(resourceId);
            List<DataMovementInterface> movementInterfaces = storageResourceDescription.getDataMovementInterfaces();
            if (movementInterfaces != null && !movementInterfaces.isEmpty()) {
                for (DataMovementInterface dataMovementInterface : movementInterfaces) {
                    System.out.println("Data Movement Interface Id :" + dataMovementInterface.getDataMovementInterfaceId());
                    System.out.println("Data Movement Protocol :" + dataMovementInterface.getDataMovementProtocol().toString());
                }
            }
        }
        description.setHostName("localhost2");
        storageResource.updateStorageResource(resourceId, description);
        if (storageResource.isStorageResourceExists(resourceId)) {
            storageResourceDescription = storageResource.getStorageResource(resourceId);
            System.out.println("**********Updated Resource name ************* : " + storageResourceDescription.getHostName());
        }
        assertTrue("Compute resource save successfully", storageResourceDescription != null);
    } catch (AppCatalogException e) {
        logger.error(e.getMessage(), e);
    }
}
Also used : StorageResource(org.apache.airavata.registry.cpi.StorageResource) AppCatalogException(org.apache.airavata.registry.cpi.AppCatalogException) StorageResourceDescription(org.apache.airavata.model.appcatalog.storageresource.StorageResourceDescription) ArrayList(java.util.ArrayList) Test(org.junit.Test)

Aggregations

ArrayList (java.util.ArrayList)1 StorageResourceDescription (org.apache.airavata.model.appcatalog.storageresource.StorageResourceDescription)1 AppCatalogException (org.apache.airavata.registry.cpi.AppCatalogException)1 StorageResource (org.apache.airavata.registry.cpi.StorageResource)1 Test (org.junit.Test)1