001: /*
002: * Copyright (c) 2002-2008 Gargoyle Software Inc. All rights reserved.
003: *
004: * Redistribution and use in source and binary forms, with or without
005: * modification, are permitted provided that the following conditions are met:
006: *
007: * 1. Redistributions of source code must retain the above copyright notice,
008: * this list of conditions and the following disclaimer.
009: * 2. Redistributions in binary form must reproduce the above copyright notice,
010: * this list of conditions and the following disclaimer in the documentation
011: * and/or other materials provided with the distribution.
012: * 3. The end-user documentation included with the redistribution, if any, must
013: * include the following acknowledgment:
014: *
015: * "This product includes software developed by Gargoyle Software Inc.
016: * (http://www.GargoyleSoftware.com/)."
017: *
018: * Alternately, this acknowledgment may appear in the software itself, if
019: * and wherever such third-party acknowledgments normally appear.
020: * 4. The name "Gargoyle Software" must not be used to endorse or promote
021: * products derived from this software without prior written permission.
022: * For written permission, please contact info@GargoyleSoftware.com.
023: * 5. Products derived from this software may not be called "HtmlUnit", nor may
024: * "HtmlUnit" appear in their name, without prior written permission of
025: * Gargoyle Software Inc.
026: *
027: * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES,
028: * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
029: * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GARGOYLE
030: * SOFTWARE INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
031: * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
032: * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
033: * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
034: * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
035: * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
036: * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
037: */
038: package com.gargoylesoftware.htmlunit.util;
039:
040: import java.io.IOException;
041: import java.net.URL;
042:
043: import com.gargoylesoftware.htmlunit.WebClient;
044: import com.gargoylesoftware.htmlunit.WebConnection;
045: import com.gargoylesoftware.htmlunit.WebRequestSettings;
046: import com.gargoylesoftware.htmlunit.WebResponse;
047: import com.gargoylesoftware.htmlunit.WebResponseData;
048: import com.gargoylesoftware.htmlunit.WebResponseImpl;
049:
050: /**
051: * Extension of {@link WebConnectionWrapper} providing facility methods to deliver something else than
052: * what the wrapped connection would deliver.
053: *
054: * @version $Revision: 2132 $
055: * @author Marc Guillemot
056: */
057: public abstract class FalsifyingWebConnection extends
058: WebConnectionWrapper {
059:
060: /**
061: * Constructs a WebConnection object wrapping provided WebConnection.
062: * @param webConnection the webConnection that does the real work
063: * @throws IllegalArgumentException if the connection is <code>null</code>
064: */
065: public FalsifyingWebConnection(final WebConnection webConnection)
066: throws IllegalArgumentException {
067: super (webConnection);
068: }
069:
070: /**
071: * Constructs an instance and places itself as connection of the WebClient.
072: * @param webClient the WebClient which WebConnection should be wrapped
073: * @throws IllegalArgumentException if the WebClient is <code>null</code>
074: */
075: public FalsifyingWebConnection(final WebClient webClient)
076: throws IllegalArgumentException {
077: super (webClient);
078: }
079:
080: /**
081: * Delivers the content for an alternate url as if it would come from the requested one
082: * @param webRequestSettings the original web request settings
083: * @param url the url from which the content should be retrieved
084: * @return the response
085: * @throws IOException if a problem occurred
086: */
087: protected WebResponse deliverFromAlternateUrl(
088: final WebRequestSettings webRequestSettings, final URL url)
089: throws IOException {
090: final URL originalUrl = webRequestSettings.getURL();
091: webRequestSettings.setURL(url);
092: final WebResponse resp = super .getResponse(webRequestSettings);
093: return new WebResponseWrapper(resp) {
094: public URL getUrl() {
095: return originalUrl;
096: }
097: };
098: }
099:
100: /**
101: * Builds a WebResponse with a new content preserving the other informations
102: * @param webResponse the web response to adapt
103: * @param newContent the new content to place in the response
104: * @return a web response with the new content
105: * @throws IOException if an encoding problem occurred
106: */
107: protected WebResponse replaceContent(final WebResponse webResponse,
108: final String newContent) throws IOException {
109: final byte[] body = newContent.getBytes(webResponse
110: .getContentCharSet());
111: final WebResponseData wrd = new WebResponseData(body,
112: webResponse.getStatusCode(), webResponse
113: .getStatusMessage(), webResponse
114: .getResponseHeaders());
115: return new WebResponseImpl(wrd, webResponse.getUrl(),
116: webResponse.getRequestMethod(), webResponse
117: .getLoadTimeInMilliSeconds());
118:
119: }
120: }
|