/src/test/java/com/atlassian/util/profiling/filters/TestProfilingFilter.java

package com.atlassian.util.profiling.filters;

import javax.servlet.FilterConfig;

import com.atlassian.util.profiling.UtilTimerStack;

import com.mockobjects.dynamic.Mock;

import junit.framework.TestCase;

public class TestProfilingFilter extends TestCase
{
    private Mock mockStatusUpdateStrategy = new Mock(StatusUpdateStrategy.class);
    private StatusUpdateStrategy statusUpdateStrategy = (StatusUpdateStrategy) mockStatusUpdateStrategy.proxy();
    private Mock mockFilterConfig = new Mock(FilterConfig.class);
    private FilterConfig filterConfig = (FilterConfig) mockFilterConfig.proxy();
    private ProfilingFilter profilingFilter;

    public void setUp() throws Exception
    {
        super.setUp();
        mockFilterConfig.expectAndReturn("getFilterName", "profiling");
        profilingFilter = new ProfilingFilter(statusUpdateStrategy);
    }

    public void tearDown() throws Exception
    {
        UtilTimerStack.getDefaultStrategy().setEnabled(false);
        super.tearDown();
    }

    public void testAutoStartTrue()
    {
        mockFilterConfig.expectAndReturn("getInitParameter", ProfilingFilter.AUTOSTART_PARAM, "true");
        profilingFilter.init(filterConfig);
        assertTrue(UtilTimerStack.isActive());
    }

    public void testAutoStartFalse()
    {
        mockFilterConfig.expectAndReturn("getInitParameter", ProfilingFilter.AUTOSTART_PARAM, "false");
        mockFilterConfig.expectAndReturn("getFilterName", "profiling");
        profilingFilter.init(filterConfig);
        assertFalse(UtilTimerStack.isActive());
    }

    public void testAutoStartInvalid()
    {
        mockFilterConfig.expectAndReturn("getInitParameter", ProfilingFilter.AUTOSTART_PARAM, "invalid");
        mockFilterConfig.expectAndReturn("getFilterName", "profiling");
        profilingFilter.init(filterConfig);
        assertFalse(UtilTimerStack.isActive());
    }
}