001    // Copyright 2009 The Apache Software Foundation
002    //
003    // Licensed under the Apache License, Version 2.0 (the "License");
004    // you may not use this file except in compliance with the License.
005    // You may obtain a copy of the License at
006    //
007    //     http://www.apache.org/licenses/LICENSE-2.0
008    //
009    // Unless required by applicable law or agreed to in writing, software
010    // distributed under the License is distributed on an "AS IS" BASIS,
011    // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
012    // See the License for the specific language governing permissions and
013    // limitations under the License.
014    
015    package org.apache.tapestry5.internal.test;
016    
017    import org.apache.tapestry5.services.MarkupRendererFilter;
018    import org.apache.tapestry5.services.MarkupRenderer;
019    import org.apache.tapestry5.MarkupWriter;
020    
021    /**
022     * Used to capture the rendered document from a traditional page render. Invokes {@link
023     * org.apache.tapestry5.internal.test.TestableResponse#setRenderedDocument(org.apache.tapestry5.dom.Document)}.
024     *
025     * @since 5.1.0.0
026     */
027    public class CaptureRenderedDocument implements MarkupRendererFilter
028    {
029        private final TestableResponse testableResponse;
030    
031        public CaptureRenderedDocument(TestableResponse testableResponse)
032        {
033            this.testableResponse = testableResponse;
034        }
035    
036    
037        public void renderMarkup(MarkupWriter writer, MarkupRenderer renderer)
038        {
039            renderer.renderMarkup(writer);
040    
041            testableResponse.setRenderedDocument(writer.getDocument());
042        }
043    }