TaxCalClient.java

/*
 * @(#)TaxCalClient.java        1.1 2004/05/03
 *
 * Copyright (c) 2004-2005 Sun Microsystems, Inc.
 * 4150,Network Circle, Santa Clara, California, 95054, U.S.A.
 * All rights reserved.
 *
 * This software is the confidential and proprietary information of Sun
 * Microsystems, Inc. ("Confidential Information").  You shall not
 * disclose such Confidential Information and shall use it only in
 * accordance with the terms of the license agreement you entered into
 * with Sun.
 */
package com.sun.appserv.sqe.security.wss.ejbws.taxcal.client;

import com.sun.ejte.ccl.reporter.SimpleReporterAdapter;
import javax.naming.*;
import javax.xml.rpc.Stub;

/**
 * This is AppClient program that access StateTaxEJB and FedTax EJB Webservices.
 * It expects StateTax webservice endpoint and FedTax endpoint URLs.
 * This client is accessed in the ant run target after configuring the webservices
 * message security at system level or applevel.  
 *
 * @version 1.1  03 May 2004
 * @author Jagadesh Munta
 */
public class TaxCalClient {
    /*
     * Tests the getStateTax and getFedTax with expected values. If value 
     * matched,then test PASSED else FAILED.
     */
 	private static SimpleReporterAdapter stat = new SimpleReporterAdapter();
            // J2EE simple reporter for logging the test status.
	private static String stateTaxEndpoint = null;
	private static String fedTaxEndpoint = null;
	private static String testSuite = "Security-wss-ejbws-";
	private static String testCase = null;
	private static TaxCalEjbService taxCalService = null;

    public static void main (String[] args) {

        if(args[0] == null || args[1] == null){
            System.out.println("TaxCal client: Argument missing."+
                    " Please provide target" +
                     "endpoint address as argument");
                System.exit(1);
        } else {
                stateTaxEndpoint = args[0];
                fedTaxEndpoint = args[1];
        }

        stat.addDescription("This is to test message security with sign or username token on request and response with signing using ejb based webservices.");
        try { 
            TaxCalClient client = new TaxCalClient();
            Context ic = new InitialContext();
            taxCalService = (TaxCalEjbService)
               ic.lookup("java:comp/env/service/TaxCalEjbService");

            client.callStateTaxService(); // Uses Binary Token - Sign
            client.callFedTaxService(); // Uses Usename Token
        }catch(Exception e){
                e.printStackTrace();
        }

        stat.printSummary(testSuite);
    }
    
    public void callStateTaxService() {
        double income = 85000.00;
        double deductions = 5000.00;
        double expectedTax = 24000.00;

        try {
            StateTaxIF taxCalIFPort = taxCalService.getStateTaxIFPort();

            ((Stub)taxCalIFPort)._setProperty (Stub.ENDPOINT_ADDRESS_PROPERTY,
                    stateTaxEndpoint);

            double stateTax = taxCalIFPort.getStateTax(income, deductions);
            System.out.println("State tax from ejb based TaxCalService :" +
                    stateTax);

            if(stateTax == expectedTax) {
                stat.addStatus(testSuite + "mesgsign-StateTaxPort", stat.PASS);
            } else {
                stat.addStatus(testSuite + "mesgsign-StateTaxPort", stat.FAIL);
            }

        } catch (Exception ex) {
                System.out.println("State TaxCalEjbWebservice client failed");
                stat.addStatus(testSuite + "mesgsign-StateTaxPort", stat.FAIL);
                ex.printStackTrace();
        } 
    }

    public void callFedTaxService() {
        double income = 97000.00;
        double deductions = 7000.00;
        double expectedTax = 18000.00;

        try {
            FedTaxIF taxCalIFPort = taxCalService.getFedTaxIFPort();
            ((Stub)taxCalIFPort)._setProperty(Stub.ENDPOINT_ADDRESS_PROPERTY,
                    fedTaxEndpoint);

            double fedTax = taxCalIFPort.getFedTax(income, deductions);
            System.out.println("Fed tax from Ejb based TaxCalService :" +
                    fedTax);

            if(fedTax == expectedTax) {
                    stat.addStatus(testSuite + "username-FedTaxPort", stat.PASS);
            } else {
                    stat.addStatus(testSuite + "username-FedTaxPort", stat.FAIL);
            }

        } catch (Exception ex) {
                System.out.println("Fed TaxCalEjbWebService client failed");
                stat.addStatus(testSuite + "username-FedTaxPort", stat.FAIL);
                ex.printStackTrace();
        } 
    }
}