opends/tester/ResponseChecker.java

/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at
 * trunk/proctor/resource/legal-notices/Proctor.LICENSE
 * or https://proctor.dev.java.net/Proctor.LICENSE.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at
 * trunk/opends/resource/legal-notices/OpenDS.LICENSE.  If applicable,
 * add the following below this CDDL HEADER, with the fields enclosed
 * by brackets "[]" replaced with your own identifying information:
 *      Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 *
 *
 *      Portions Copyright 2009 Sun Microsystems, Inc.
 */

package com.ibm.staf.service.opends.tester;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.Reader;
import java.io.StringReader;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.StringTokenizer;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.Unmarshaller;
import javax.xml.soap.MessageFactory;
import javax.xml.soap.MimeHeaders;
import javax.xml.soap.SOAPBody;
import javax.xml.soap.SOAPElement;
import javax.xml.soap.SOAPMessage;
import org.opends.dsml.protocol.BatchResponse;

public class ResponseChecker {

    private Map<String, String> header = new HashMap<String, String>();
    private String httpCode;
    private BatchResponse batchResponse;
    private String body;
    public ResponseChecker(String response) throws Exception {
        this(new StringReader(response));

    }

    public ResponseChecker(Reader response) throws Exception {

        // this.source = new StringBuilder();
        boolean headerDone = false;
        String line = null;
        BufferedReader reader = new BufferedReader(response);
        StringBuilder sb = new StringBuilder();

        // assign first line of header to httpCode
        // then assign all header to <key>:<value> to header map
        // finally fill sb with body of request
        while ((line = reader.readLine()) != null) {
            if (!headerDone) {
                if (line.length() == 0) {
                    headerDone = true;
                }
                if (!headerDone) {
                    if (httpCode == null) {
                        httpCode = line;
                    } else {
                        String[] entry = line.split(":");
                        header.put(entry[0].trim().toLowerCase(), entry[1].trim());
                    }
                }
            } else {
                sb.append(line);
            }
        }
        this.body = sb.toString();

        if (httpCode != null) {
            if (httpCode.split(" ")[1].equals("200")) {

                // reconstruct header for SOAP engine
                MimeHeaders mimeHeaders = new MimeHeaders();
                for (Map.Entry<String, String> entry : header.entrySet()) {
                    String name = entry.getKey();
                    String value = entry.getValue();
                    StringTokenizer token = new StringTokenizer(value, ",");
                    while (token.hasMoreTokens()) {
                        mimeHeaders.addHeader(name, token.nextToken().trim());
                    }
                }
                MessageFactory messageFactory = MessageFactory.newInstance();
                SOAPMessage message = messageFactory.createMessage(mimeHeaders, new ByteArrayInputStream(this.body.getBytes()));

                //DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
                //dbf.setNamespaceAware(true);
                //DocumentBuilder db = dbf.newDocumentBuilder();
                //Document doc = db.newDocument();
                SOAPBody soapBody = message.getSOAPBody();

                Iterator iterator = soapBody.getChildElements();
                while (iterator.hasNext()) {
                    Object object = iterator.next();
                    if (!(object instanceof SOAPElement)) {
                        continue;
                    }
                    SOAPElement soapElement = (SOAPElement) object;

                    JAXBContext jaxbContext = JAXBContext.newInstance(BatchResponse.class);
                    Unmarshaller unmarshaller = jaxbContext.createUnmarshaller();

                    JAXBElement<BatchResponse> batchResponseElement = unmarshaller.unmarshal(soapElement, BatchResponse.class);

                    this.batchResponse = batchResponseElement.getValue();
                }
            }
        }

    }

    public boolean equals(Object o) {
        if (o == this) {
            return true;
        }
        if (!(o instanceof ResponseChecker)) {
            return false;
        }
        ResponseChecker other = (ResponseChecker) o;

        // check code "HTTP/1.1 200 OK".split(" ")[1] = "200"
        if (!this.httpCode.split(" ")[1].equals(other.httpCode.split(" ")[1])) {
            return false;
        }

        if (!Checker.equals(batchResponse, other.batchResponse)) {
            return false;
        }

        return true;
    }
}