/*
* $Id: PortletSessionMapTest.java 418521 2006-07-01 23:36:50Z mrdon $
*
* Copyright 2006 The Apache Software Foundation.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.struts2.portlet;
import java.util.Arrays;
import java.util.Enumeration;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import javax.portlet.PortletRequest;
import javax.portlet.PortletSession;
import org.jmock.Mock;
import org.jmock.MockObjectTestCase;
import org.jmock.core.Constraint;
/**
* PortletSessionMapTest. Insert description.
*
*/
public class PortletSessionMapTest extends MockObjectTestCase {
public void testPut() {
Mock mockSession = mock(PortletSession.class);
Mock mockRequest = mock(PortletRequest.class);
PortletRequest req = (PortletRequest)mockRequest.proxy();
PortletSession session = (PortletSession)mockSession.proxy();
mockRequest.expects(once()).method("getPortletSession").will(returnValue(session));
Constraint[] params = new Constraint[]{eq("testAttribute1"), eq("testValue1")};
mockSession.expects(once()).method("setAttribute").with(params);
mockSession.expects(once()).method("getAttribute").with(eq("testAttribute1")).will(returnValue("testValue1"));
params = new Constraint[]{eq("testAttribute2"), eq("testValue2")};
mockSession.expects(once()).method("setAttribute").with(params);
mockSession.expects(once()).method("getAttribute").with(eq("testAttribute2")).will(returnValue("testValue2"));
PortletSessionMap map = new PortletSessionMap(req);
map.put("testAttribute1", "testValue1");
map.put("testAttribute2", "testValue2");
}
public void testGet() {
Mock mockSession = mock(PortletSession.class);
Mock mockRequest = mock(PortletRequest.class);
PortletRequest req = (PortletRequest)mockRequest.proxy();
PortletSession session = (PortletSession)mockSession.proxy();
mockRequest.expects(once()).method("getPortletSession").will(returnValue(session));
mockSession.expects(once()).method("getAttribute").with(eq("testAttribute1")).will(returnValue("testValue1"));
mockSession.expects(once()).method("getAttribute").with(eq("testAttribute2")).will(returnValue("testValue2"));
PortletSessionMap map = new PortletSessionMap(req);
Object val1 = map.get("testAttribute1");
Object val2 = map.get("testAttribute2");
assertEquals("testValue1", val1);
assertEquals("testValue2", val2);
}
public void testClear() {
Mock mockSession = mock(PortletSession.class);
Mock mockRequest = mock(PortletRequest.class);
PortletRequest req = (PortletRequest)mockRequest.proxy();
PortletSession session = (PortletSession)mockSession.proxy();
mockRequest.expects(once()).method("getPortletSession").will(returnValue(session));
mockSession.expects(once()).method("invalidate");
PortletSessionMap map = new PortletSessionMap(req);
map.clear();
}
public void testRemove() {
Mock mockSession = mock(PortletSession.class);
Mock mockRequest = mock(PortletRequest.class);
PortletRequest req = (PortletRequest)mockRequest.proxy();
PortletSession session = (PortletSession)mockSession.proxy();
mockRequest.expects(once()).method("getPortletSession").will(returnValue(session));
mockSession.stubs().method("getAttribute").with(eq("dummyKey")).will(returnValue("dummyValue"));
mockSession.expects(once()).method("removeAttribute").with(eq("dummyKey"));
PortletSessionMap map = new PortletSessionMap(req);
Object ret = map.remove("dummyKey");
assertEquals("dummyValue", ret);
}
public void testEntrySet() {
Mock mockSession = mock(PortletSession.class);
Mock mockRequest = mock(PortletRequest.class);
PortletRequest req = (PortletRequest)mockRequest.proxy();
PortletSession session = (PortletSession)mockSession.proxy();
Enumeration names = new Enumeration() {
List keys = Arrays.asList(new Object[]{"key1", "key2"});
Iterator it = keys.iterator();
public boolean hasMoreElements() {
return it.hasNext();
}
public Object nextElement() {
return it.next();
}
};
mockSession.stubs().method("getAttributeNames").will(returnValue(names));
mockSession.stubs().method("getAttribute").with(eq("key1")).will(returnValue("value1"));
mockSession.stubs().method("getAttribute").with(eq("key2")).will(returnValue("value2"));
mockRequest.expects(once()).method("getPortletSession").will(returnValue(session));
PortletSessionMap map = new PortletSessionMap(req);
Set entries = map.entrySet();
assertEquals(2, entries.size());
Iterator it = entries.iterator();
Map.Entry entry = (Map.Entry)it.next();
assertEquals("key2", entry.getKey());
assertEquals("value2", entry.getValue());
entry = (Map.Entry)it.next();
assertEquals("key1", entry.getKey());
assertEquals("value1", entry.getValue());
}
}