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
015package org.apache.tapestry5.internal.test;
016
017import org.apache.tapestry5.services.MarkupRendererFilter;
018import org.apache.tapestry5.services.MarkupRenderer;
019import 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 */
027public 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}