nz.ac.auckland.integration.testing.validator.HeadersValidator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of soaunit Show documentation

Orchestrated Testing of SOA Artifacts Using Apache Camel

The newest version!

package nz.ac.auckland.integration.testing.validator;

import nz.ac.auckland.integration.testing.resource.HeadersTestResource;
import org.apache.camel.Exchange;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Map;

/**
 * Provides a mechanism for validating headers for an expectation or response
 *
 * @author David MacDonald 
 */
public class HeadersValidator implements Validator {

    private static final Logger logger = LoggerFactory.getLogger(HeadersValidator.class);

    private HeadersTestResource resource;

    public HeadersValidator(HeadersTestResource resource) {
        this.resource = resource;
    }

    /**
     * @param exchange The exchange containing the headers we need to validate
     * @return true if each header in the test resource is in input; additional headers in the input will be ignored
     */
    public boolean validate(Exchange exchange) {
        Map value = exchange.getIn().getHeaders();
        return value != null && validate(value);
    }

    public boolean validate(Map value) {
        if (value == null) return false;
        Map expectedHeaders;

        try {
            expectedHeaders = resource.getValue();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }

        //I'm not interested if the input has any additional headers
        for (String expectedKey : expectedHeaders.keySet()) {
            if (!value.containsKey(expectedKey)) {
                logger.warn("The key: {} was not found", expectedKey);
                return false;
            }

            if (!value.get(expectedKey).equals(expectedHeaders.get(expectedKey))) {
                logger.warn("The key: {} has an unexpected value", expectedKey);
                return false;
            }
        }

        return true;
    }
}