com.mockrunner.base.GenericHTMLOutputModule Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mockrunner-servlet Show documentation

Mock classes for servlets

There is a newer version: 2.0.7

package com.mockrunner.base;

import com.mockrunner.mock.web.WebMockObjectFactory;

/**
 * Generic implementation of {@link HTMLOutputModule}.
 * Can be used to to test HTML output in modules that
 * do not extend {@link HTMLOutputModule}.
 */
public class GenericHTMLOutputModule extends HTMLOutputModule
{
    private WebMockObjectFactory factory;
    
    public GenericHTMLOutputModule(WebMockObjectFactory factory)
    {
        super(factory);
        this.factory = factory;
    }
    
    public String getOutput()
    {
        try
        {
            factory.getMockResponse().getWriter().flush();    
        }
        catch(Exception ignored)
        {
        
        }
        return factory.getMockResponse().getOutputStreamContent();
    }
}