Browse Source

session单元测试

master
richie 5 years ago
parent
commit
68e1e92d6b
  1. 47
      src/test/java/com/fanruan/api/session/MockSession.java
  2. 172
      src/test/java/com/fanruan/api/session/SessionKitTest.java

47
src/test/java/com/fanruan/api/session/MockSession.java

@ -0,0 +1,47 @@
package com.fanruan.api.session;
import com.fr.stable.script.CalculatorProvider;
import com.fr.stable.web.WebContextProvider;
import com.fr.web.session.SessionIDInfo;
/**
* @author richie
* @version 10.0
* Created by richie on 2019-08-30
*/
class MockSession extends SessionIDInfo {
MockSession(String sessionID) {
this.sessionID = sessionID;
}
@Override
public String getWebTitle() {
return null;
}
@Override
public String getRelativePath() {
return null;
}
@Override
public WebContextProvider getWebContext() {
return null;
}
@Override
public Object resolveVariable(Object var, CalculatorProvider cal) {
return null;
}
@Override
public Object getParameterValue(String parameterName) {
return null;
}
@Override
public void release() {
}
}

172
src/test/java/com/fanruan/api/session/SessionKitTest.java

@ -1,18 +1,186 @@
package com.fanruan.api.session; package com.fanruan.api.session;
import com.fanruan.api.Prepare; import com.fanruan.api.Prepare;
import com.fr.general.web.ParameterConstants;
import com.fr.invoke.Reflect;
import com.fr.stable.StringUtils;
import com.fr.stable.web.SessionProvider;
import com.fr.stable.web.SessionSource;
import com.fr.stable.web.Weblet;
import com.fr.web.core.RemoteAddressManager;
import com.fr.web.core.SessionPoolManager;
import com.fr.web.session.SessionIDInfo;
import com.fr.web.utils.WebUtils;
import org.easymock.EasyMock;
import org.easymock.IAnswer;
import org.easymock.IMocksControl;
import org.junit.Assert; import org.junit.Assert;
import org.junit.Test; import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.api.easymock.PowerMock;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor;
import org.powermock.modules.junit4.PowerMockRunner;
import org.powermock.reflect.Whitebox;
import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
/** /**
* @author richie * @author richie
* @version 10.0 * @version 10.0
* Created by richie on 2019-08-09 * Created by richie on 2019-08-09
*/ */
@RunWith(PowerMockRunner.class)
@PrepareForTest({SessionPoolManager.class, WebUtils.class})
@SuppressStaticInitializationFor({"com.fr.web.core.SessionPoolManager"})
public class SessionKitTest extends Prepare { public class SessionKitTest extends Prepare {
private static final long MIN_ACCESS_INTERVAL = 900;
private int taskCount = 0;
private Map<String, SessionProvider> sessions = new HashMap<String, SessionProvider>();
@Test
public void testGenerateSession() throws Exception {
final SessionIDInfo sessionInfo = new MockSession("session");
final Map<String, Object> params = new HashMap<String, Object>();
PowerMock.mockStatic(SessionPoolManager.class);
PowerMock.mockStatic(WebUtils.class);
IMocksControl control = EasyMock.createControl();
final HttpServletRequest request = control.createMock(HttpServletRequest.class);
EasyMock.expect(WebUtils.getIpAddr(request)).andReturn("127.0.0.1").anyTimes();
EasyMock.expect(WebUtils.getHTTPRequestParameter(request, ParameterConstants.OP)).andReturn("view").anyTimes();
EasyMock.expect(WebUtils.parameters4SessionIDInforContainMPCache(request)).andReturn(params).anyTimes();
final Weblet weblet = control.createMock(Weblet.class);
EasyMock.expect(weblet.isSessionOccupy()).andReturn(false).anyTimes();
EasyMock.expect(weblet.createSessionIDInfor(request, "127.0.0.1", params)).andReturn(sessionInfo).anyTimes();
EasyMock.expect(addSessionIDInfo(sessionInfo)).andReturn(sessionInfo.getSessionID()).anyTimes();
EasyMock.expect(SessionPoolManager.generateSessionID(request, null, weblet)).andAnswer(new IAnswer<String>() {
@Override
public String answer() throws Throwable {
SessionProvider sessionIDInfor = weblet.createSessionIDInfor(request, "127.0.0.1", params);
String sessionID = addSessionIDInfo(sessionIDInfor);
if (weblet.isSessionOccupy()) {
assert sessionID != null;
RemoteAddressManager.add("127.0.0.1", sessionID);
}
return sessionID;
}
}).anyTimes();
control.replay();
PowerMock.replayAll();
sessionInfo.setSource(SessionSource.SINGLE);
SessionPoolManager.generateSessionID(request, null, weblet);
Assert.assertSame(sessionInfo.getSource(), SessionSource.SINGLE);
sessionInfo.setSource(SessionSource.INTEGRATION);
SessionPoolManager.generateSessionID(request, null, weblet);
Assert.assertSame(sessionInfo.getSource(), SessionSource.INTEGRATION);
}
@Test
public void testAddDelayTask() {
String sessionID = "test";
sessions.put("test", new MockSession("test"));
for (int i = 0; i < 100; i++) {
accessSession(sessionID);
}
Assert.assertEquals(1, this.taskCount);
for (int i = 0; i < 2; i++) {
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
//do nothing
}
accessSession(sessionID);
}
Assert.assertEquals(3, this.taskCount);
}
private void accessSession(String sessionId) {
if (StringUtils.isBlank(sessionId)) {
return;
}
SessionProvider sessionIDInfor = sessions.get(sessionId);
if (sessionIDInfor != null) {
long accessInterval = System.currentTimeMillis() - sessionIDInfor.getLastAccessTime();
sessionIDInfor.updateTime();
if (accessInterval > MIN_ACCESS_INTERVAL) {
taskCount++;
}
}
}
@Test
public void testGetAllSession() {
Whitebox.setInternalState(
SessionPoolManager.class,
"sessionIDMap",
new ConcurrentHashMap<String, SessionProvider>()
);
Whitebox.setInternalState(SessionPoolManager.class, "sessionToDelete", new HashSet<String>());
Map<String, SessionProvider> sessionIDMap = Reflect.on(SessionPoolManager.class).get("sessionIDMap");
String sessionID1 = "Test1";
sessionIDMap.put(sessionID1, new MockSession(sessionID1));
sessionIDMap.put("Test2", new MockSession("Test2"));
closeSession(sessionID1);
List<SessionProvider> list = SessionPoolManager.getAllSession(SessionProvider.class);
assertEquals(1, list.size());
assertEquals("Test2", list.get(0).getSessionID());
assertNotNull(SessionKit.getSession("Test2"));
}
@Test @Test
public void getSession() throws Exception { public void testHasSessionID() {
Assert.assertNull(SessionKit.getSession("abc")); Whitebox.setInternalState(
SessionPoolManager.class,
"sessionIDMap",
new ConcurrentHashMap<String, SessionProvider>()
);
Whitebox.setInternalState(SessionPoolManager.class, "sessionToDelete", new HashSet<String>());
Map<String, SessionProvider> sessionIDMap = Reflect.on(SessionPoolManager.class).get("sessionIDMap");
String sessionID1 = "Test1";
sessionIDMap.put(sessionID1, new MockSession(sessionID1));
sessionIDMap.put("Test2", new MockSession("Test2"));
closeSession(sessionID1);
assertFalse(SessionPoolManager.hasSessionID(sessionID1));
assertTrue(SessionPoolManager.hasSessionID("Test2"));
}
private void closeSession(final String sessionID) {
final Set<String> sessionToDelete = Reflect.on(SessionPoolManager.class).get("sessionToDelete");
sessionToDelete.add(sessionID);
ExecutorService closeSessionPool = Executors.newFixedThreadPool(1);
closeSessionPool.submit(new Runnable() {
@Override
public void run() {
try {
//模拟删除session对象的过程
Thread.sleep(1000);
} catch (InterruptedException e) {
//do nothing
}
synchronized (SessionPoolManager.class) {
sessionToDelete.remove(sessionID);
}
}
});
}
private String addSessionIDInfo(SessionProvider sessionIDInfo) {
return Reflect.on(SessionPoolManager.class).call("addSessionIDInfor", sessionIDInfo).get();
} }
} }
Loading…
Cancel
Save