Example usage for org.apache.commons.collections.iterators IteratorEnumeration IteratorEnumeration

Introduction

In this page you can find the example usage for org.apache.commons.collections.iterators IteratorEnumeration IteratorEnumeration.

Prototype

public IteratorEnumeration(Iterator iterator)

Source Link

Document

Constructs a new IteratorEnumeration that will use the given iterator.

Usage

From source file:org.wso2.carbon.identity.webfinger.DefaultWebFingerProcessorTest.java

@Test
public void testGetResponse() throws Exception {
    DefaultWebFingerProcessor defaultWebFingerProcessor = DefaultWebFingerProcessor.getInstance();
    HttpServletRequest request = Mockito.mock(HttpServletRequest.class);
    final Map<String, String> parameterMap = new HashMap<>();
    parameterMap.put(WebFingerConstants.RESOURCE, "TestResource1");
    Mockito.doAnswer(new Answer() {

        @Override//  w w  w .  ja  va  2 s  . c om
        public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
            return parameterMap.get(invocationOnMock.getArguments()[0]);
        }
    }).when(request).getParameter(Matchers.anyString());
    Mockito.doAnswer(new Answer() {

        @Override
        public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
            return new IteratorEnumeration(parameterMap.keySet().iterator());
        }
    }).when(request).getParameterNames();
    try {
        WebFingerResponse response = defaultWebFingerProcessor.getResponse(request);
        fail("WebFingerEndpointException should have been thrown");
    } catch (WebFingerEndpointException e) {
        //Expected exception
    }

    parameterMap.put(WebFingerConstants.REL, "TestRelates");

    try {
        WebFingerResponse response = defaultWebFingerProcessor.getResponse(request);
        fail("WebFingerEndpointException should have been thrown");
    } catch (WebFingerEndpointException e) {
        //Expected exception
    }

    parameterMap.put(WebFingerConstants.RESOURCE, "http://test.t/TestResource1");
    WebFingerResponse response = defaultWebFingerProcessor.getResponse(request);
    assertNotNull(response);
}

From source file:org.wso2.carbon.identity.webfinger.servlet.WebFingerServletTest.java

@Test
public void testDoGet() throws Exception {
    final Map<String, String> parameters = new HashMap<>();

    HttpServletRequest request = Mockito.mock(HttpServletRequest.class);
    Mockito.when(request.getParameterNames())
            .thenReturn(new IteratorEnumeration(parameters.keySet().iterator()));
    Mockito.when(request.getParameter(Matchers.anyString())).thenAnswer(new Answer<String>() {

        @Override//w  w w .j  av a 2s .co  m
        public String answer(InvocationOnMock invocationOnMock) throws Throwable {
            return parameters.get(invocationOnMock.getArgumentAt(0, String.class));
        }
    });

    HttpServletResponse response = Mockito.mock(HttpServletResponse.class);

    Mockito.doAnswer(new Answer() {

        @Override
        public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
            return null;
        }
    }).when(response).setStatus(Matchers.anyInt());

    WebFingerServlet webFingerServlet = new WebFingerServlet();
    webFingerServlet.doGet(request, response);
}

From source file:test.net.jawr.web.servlet.mock.MockServletContext.java

@SuppressWarnings("unchecked")
public Enumeration<String> getAttributeNames() {
    return new IteratorEnumeration(attributes.keySet().iterator());
}

From source file:test.net.jawr.web.servlet.mock.MockServletContext.java

@SuppressWarnings("unchecked")
public Enumeration<String> getInitParameterNames() {
    return new IteratorEnumeration(attributes.keySet().iterator());
}