allocation-manager/airavata-allocation-manager/airavata-allocation-manager-stubs/src/main/java/org/apache/airavata/allocation/manager/client/NotificationTestClient.java

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package org.apache.airavata.allocation.manager.client;

import java.util.List;
import org.apache.airavata.allocation.manager.models.ReviewerAllocationDetail;
import org.apache.thrift.TException;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TProtocol;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport;
import org.apache.thrift.transport.TTransportException;
import org.apache.airavata.allocation.manager.models.UserAllocationDetail;
import org.apache.airavata.allocation.manager.models.UserDetail;
import org.apache.airavata.allocation.manager.service.cpi.AllocationRegistryService;

public class NotificationTestClient {

    public static void main(String[] args) {

        try {
            TTransport transport;

            transport = new TSocket("localhost", 9040);
            transport.open();

            TProtocol protocol = new TBinaryProtocol(transport);
            AllocationRegistryService.Client client = new AllocationRegistryService.Client(protocol);
            System.out.println("Started client");
            
//            System.out.println("Testing createAllocationRequest() ");
//
//            UserAllocationDetail requestDetails = new UserAllocationDetail();
//            requestDetails.setProjectId("1115");
//            requestDetails.setUsername("nikitha");
//            requestDetails.setRequestedDate(101L);
//            requestDetails.setTitle("Test");
//            requestDetails.setProjectDescription("Test");
//            requestDetails.setTypeOfAllocation("Comm");
//            requestDetails.setStatus("pending");
//
//            System.out.println(client.createAllocationRequest(requestDetails));

//            UserAllocationDetail rDetails1 = new UserAllocationDetail();
//            rDetails1.setProjectId("1234");
//            rDetails1.setUsername("madrina");
//            rDetails1.setRequestedDate(101L);
//            rDetails1.setTitle("Test");
//            rDetails1.setProjectDescription("Test");
//            rDetails1.setTypeOfAllocation("Comm");
//            rDetails1.setStatus("pending");
//
//            System.out.println(client.createAllocationRequest(rDetails1));
//            
//            System.out.println("######################");
//            
//            System.out.println("Testing deleteAllocationRequest() ");
//
//            System.out.println(client.deleteAllocationRequest("123456"));
//            
//            System.out.println("######################");
//            
//            System.out.println("Testing assignReviewers() ");
//
//            System.out.println(client.assignReviewers("123456", "reviewer1", "admin"));
//            System.out.println(client.assignReviewers("123456", "reviewer2", "admin"));
            System.out.println(client.assignReviewers("1234", "reviewer2", "admin"));
            System.out.println(client.assignReviewers("1234", "reviewer3", "admin"));
//            
//            System.out.println("######################");
//
//
//
//            System.out.println("Testing updateRequestByReviewer() ");
//            ReviewerAllocationDetail reviewerAllocationDetail = new ReviewerAllocationDetail();
//            reviewerAllocationDetail.setProjectId("123456");
//            reviewerAllocationDetail.setUsername("reviewer2");
//            reviewerAllocationDetail.setMaxMemoryPerCpu(15L);
//            System.out.println(client.updateRequestByReviewer(reviewerAllocationDetail));
//
//            ReviewerAllocationDetail reviewerAllocationDetail1 = new ReviewerAllocationDetail();
//            reviewerAllocationDetail1.setProjectId("123456");
//            reviewerAllocationDetail1.setUsername("reviewer1");
//            reviewerAllocationDetail1.setMaxMemoryPerCpu(5L);
//            System.out.println(client.updateRequestByReviewer(reviewerAllocationDetail1));
//            
//
//
            System.out.println("Testing approveRequest() ");

            System.out.println(client.approveRequest("1234", "admin", 1l, 2l, 50l));
//            
//            System.out.println("######################");
//
            System.out.println("Testing rejectRequest() ");

            System.out.println(client.rejectRequest("1234", "admin"));
//            
//            System.out.println("######################");

            transport.close();
        } catch (TTransportException e) {
            e.printStackTrace();
        } catch (TException x) {
            x.printStackTrace();
        } finally {
            //transport.close();
        }
    }

}