WireMonitorServer.java

/*
 * Copyright (c) 2005-2010, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 * 
 * WSO2 Inc. licenses this file to you under the Apache License,
 * Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package org.wso2.carbon.automation.test.utils.common;

/**
 * This class can be used to capture wire messages
 */
public class WireMonitorServer {
    private static final int TIMEOUT_VALUE = 60000;
    int READ_TIME_OUT = 30000;
    private boolean isFinished = false;
    String response;
    int port;

    /**
     * Start listening to a port
     *
     * @param port to be listened
     */
    public WireMonitorServer(int port) {
        this.port = port;
    }

    public void start() {
        response = "";
        Thread thread = new WireMonitor(port, this);
        thread.start();
    }


    /**
     * Wait until response is received and returns
     *
     * @return will return null if response is not received
     */
    public String getCapturedMessage() {
        Long time = System.currentTimeMillis();
        while (!isFinished) {
            // If wire monitor is not responding than 2min this will continue
            if (System.currentTimeMillis() > (time + TIMEOUT_VALUE)) {
                break;
            }
        }
        return response;
    }

    public void setReadTimeOut (int timeout) {
        READ_TIME_OUT = timeout;
    }

    public void setFinished(boolean isFinished) {
        this.isFinished = isFinished;
    }
}