1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 package org.apache.struts2.util;
23
24 import java.util.HashMap;
25 import java.util.Map;
26 import java.util.TreeMap;
27
28 import junit.framework.TestCase;
29
30 import com.opensymphony.xwork2.ActionContext;
31
32
33 /***
34 * TokenHelperTest
35 *
36 */
37 public class TokenHelperTest extends TestCase {
38
39 private Map session;
40
41
42 public void testSetToken() {
43 String token = TokenHelper.setToken();
44 assertEquals(token, session.get(TokenHelper.DEFAULT_TOKEN_NAME));
45 }
46
47 public void testSetTokenWithName() {
48 String tokenName = "myTestToken";
49 String token = TokenHelper.setToken(tokenName);
50 assertEquals(token, session.get(tokenName));
51 }
52
53 public void testValidToken() {
54 String tokenName = "validTokenTest";
55 String token = TokenHelper.setToken(tokenName);
56 assertEquals(token, session.get(tokenName));
57 ActionContext.getContext().getParameters().put(TokenHelper.TOKEN_NAME_FIELD, new String[]{tokenName});
58 ActionContext.getContext().getParameters().put(tokenName, new String[]{token});
59 assertTrue(TokenHelper.validToken());
60 }
61
62 public void testGetTokenDoesNotNpe() {
63 String token = TokenHelper.getToken(null);
64 assertTrue(token == null);
65
66 String token2 = TokenHelper.getToken("");
67 assertTrue(token2 == null);
68 }
69
70 protected void setUp() throws Exception {
71 session = new HashMap();
72 Map params = new TreeMap();
73 Map ctxMap = new TreeMap();
74 ctxMap.put(ActionContext.SESSION, session);
75 ctxMap.put(ActionContext.PARAMETERS, params);
76 ActionContext ctx = new ActionContext(ctxMap);
77 ActionContext.setContext(ctx);
78 }
79
80 protected void tearDown() {
81 ActionContext.setContext(null);
82 }
83 }
84