9aae2bb3b7501ab4b8a48a3b5fcdb50c49bb4442
[controller.git] / opendaylight / samples / clustersession / src / test / java / org / opendaylight / controller / clustersession / ClusterSessionManagerTest.java
1 package org.opendaylight.controller.clustersession;
2
3 import static org.junit.Assert.assertEquals;
4 import static org.junit.Assert.assertNotNull;
5 import static org.junit.Assert.assertNull;
6 import static org.mockito.Mockito.mock;
7 import static org.mockito.Mockito.when;
8
9 import java.io.IOException;
10 import java.util.HashMap;
11 import java.util.concurrent.ConcurrentHashMap;
12 import java.util.concurrent.ConcurrentMap;
13
14 import org.apache.catalina.Context;
15 import org.apache.catalina.LifecycleException;
16 import org.apache.catalina.Session;
17 import org.junit.AfterClass;
18 import org.junit.BeforeClass;
19 import org.junit.Test;
20 import org.junit.runner.RunWith;
21 import org.opendaylight.controller.clustering.services.IClusterGlobalServices;
22 import org.opendaylight.controller.clustersession.impl.ClusterSessionServiceImpl;
23 import org.osgi.framework.Bundle;
24 import org.osgi.framework.BundleContext;
25 import org.osgi.framework.FrameworkUtil;
26 import org.osgi.framework.ServiceReference;
27 import org.powermock.api.mockito.PowerMockito;
28 import org.powermock.core.classloader.annotations.PrepareForTest;
29 import org.powermock.modules.junit4.PowerMockRunner;
30
31 @RunWith(PowerMockRunner.class)
32 @PrepareForTest({FrameworkUtil.class})
33 public class ClusterSessionManagerTest {
34   static ClusterSessionManager manager = null;
35   static ClusterSessionServiceImpl sessionService = null;
36   private static final String SESSION_CACHE = "customSessionManager.sessionData";
37   static ConcurrentMap<String, ClusterSessionData> sessions = new ConcurrentHashMap<String, ClusterSessionData>();
38   private String sessionId = "1234567";
39   final String AUTH_TYPE = "FORM";
40   final String ATTRIBUTE_NAME = "AuthType";
41   final int SESSION_ID_LENGTH = 7;
42   @SuppressWarnings("unchecked")
43   @BeforeClass
44   public static void init(){
45     Bundle bundle = mock(Bundle.class);
46     BundleContext context = mock(BundleContext.class);
47     IClusterGlobalServices clusterGlobalService = mock(IClusterGlobalServices.class);
48     ServiceReference<IClusterGlobalServices> serviceReference = mock(ServiceReference.class);
49     PowerMockito.mockStatic(FrameworkUtil.class);
50     when(FrameworkUtil.getBundle(ClusterSessionManager.class)).thenReturn(bundle);
51     when(bundle.getBundleContext()).thenReturn(context);
52     when(context.getService(serviceReference)).thenReturn(clusterGlobalService);
53     when((ConcurrentMap<String, ClusterSessionData>)clusterGlobalService.getCache(SESSION_CACHE)).thenReturn(sessions);
54     Context containerContext = mock(Context.class);
55     manager = new ClusterSessionManager();
56     manager.setContainer(containerContext);
57     try {
58       manager.startInternal();
59     } catch (LifecycleException e) {
60     }
61     sessionService = (ClusterSessionServiceImpl) manager.getSessionService();
62     sessionService.addingService(serviceReference);
63   }
64
65   @Test
66   public void checkSessionManagerCreated(){
67     assertEquals("session manager info does not match", "ClusterSessionManager/1.0", manager.getInfo());
68     assertEquals("session manager name does not match", "ClusterSessionManager", manager.getName());
69   }
70
71   @Test
72   public void testCreateEmptySession(){
73     Session session = manager.createEmptySession();
74     assertEquals("session manager does not match", manager, session.getManager());
75   }
76
77   @Test
78   public void testCreateRandomSessionId(){
79     Session session = manager.createSession(null);
80     assertEquals("Session should be valid", true, session.isValid());
81     manager.remove(session);
82   }
83
84   @Test
85   public void testCreateSession(){
86     Session session = manager.createSession(sessionId);
87     assertEquals("Session should be valid", true, session.isValid());
88     assertEquals("Session id does not match", sessionId, session.getId());
89     manager.remove(session);
90   }
91
92   @Test
93   public void testReCreateSession(){
94     Session session = manager.createSession(sessionId);
95     assertEquals("Session should be valid", true, session.isValid());
96     assertEquals("Session id does not match", sessionId, session.getId());
97     manager.createSession(sessionId);
98     manager.remove(session);
99   }
100
101   @Test
102   public void testSessionCRUD() throws IOException{
103     Session foundSession = manager.findSession(sessionId);
104     assertNull("Session should not exist here", foundSession);
105     Session session = manager.createSession(sessionId);
106     manager.add(session);
107     foundSession = manager.findSession(sessionId);
108     assertEquals("Session was not found, id does not match", sessionId, foundSession.getId());
109     manager.remove(session);
110     foundSession = manager.findSession(sessionId);
111     assertEquals("Session was not removed", null, foundSession);
112   }
113
114   @Test
115   public void testExpireSession() throws IOException{
116     Session session = manager.createSession(sessionId);
117     session.setAuthType(AUTH_TYPE);
118     manager.add(session);
119     Session foundSession = manager.findSession(sessionId);
120     assertEquals("Session was not found", sessionId, foundSession.getId());
121     manager.expireSession(sessionId);
122     foundSession = manager.findSession(sessionId);
123     assertEquals("Session was not expired", null, foundSession);
124   }
125
126   @Test
127   public void testFindSessions(){
128     Session session = manager.createSession(sessionId);
129     session.setAuthType(AUTH_TYPE);
130     manager.add(session);
131     Session[] sessions = manager.findSessions();
132     assertEquals("Session array size does not match", 1, sessions.length);
133     assertEquals("Session array size does not match", sessionId, sessions[0].getId());
134     manager.remove(session);
135   }
136
137   @Test
138   public void testGetSession(){
139     ClusterSession session = (ClusterSession) manager.createSession(sessionId);
140     session.setAttribute(ATTRIBUTE_NAME, AUTH_TYPE);
141     manager.add(session);
142     HashMap<String, String> sessionAttributes = manager.getSession(sessionId);
143     assertNotNull("Session attribute should not be null", sessionAttributes);
144     assertEquals("Session attribute size does not match", 1, sessionAttributes.size());
145     assertEquals("Session attribute size does not match", AUTH_TYPE, sessionAttributes.get(ATTRIBUTE_NAME));
146     manager.remove(session);
147   }
148
149   @AfterClass
150   public static void cleanup(){
151     try {
152       manager.stopInternal();
153     } catch (LifecycleException e) {
154     }
155   }
156
157 }

©2013 OpenDaylight, A Linux Foundation Collaborative Project. All Rights Reserved.
OpenDaylight is a registered trademark of The OpenDaylight Project, Inc.
Linux Foundation and OpenDaylight are registered trademarks of the Linux Foundation.
Linux is a registered trademark of Linus Torvalds.