package org.wamblee.xmlrouter.impl;

import static junit.framework.Assert.*;
import static org.mockito.Matchers.*;
import static org.mockito.Mockito.*;

import javax.xml.transform.dom.DOMSource;

import org.junit.Before;
import org.junit.Test;
import org.wamblee.xmlrouter.common.Id;
import org.wamblee.xmlrouter.config.Filter;

public class RobustFilterTest {

    private Filter filter;
    private Filter robust;
    private DOMSource source;

    @Before
    public void setUp() {
        filter = mock(Filter.class);
        robust = new RobustFilter(new Id<Filter>(10), filter);
        source = mock(DOMSource.class);
    }

    @Test
    public void testNoEzception() {
        when(filter.isAllowed(anyString(), any(DOMSource.class))).thenReturn(
            true);
        assertTrue(robust.isAllowed("xx", source));
        verify(filter).isAllowed(eq("xx"), same(source));

        reset(filter);
        when(filter.isAllowed(anyString(), any(DOMSource.class))).thenReturn(
            false);
        assertFalse(robust.isAllowed("xx", source));
        verify(filter).isAllowed(eq("xx"), same(source));
    }

    @Test
    public void testException() {
        doThrow(new RuntimeException("bla")).when(filter).isAllowed(
            anyString(), any(DOMSource.class));
        assertTrue(robust.isAllowed("xx", source));
        verify(filter).isAllowed(eq("xx"), same(source));

    }
}