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
27 import junit.framework.TestCase;
28
29 import com.opensymphony.xwork2.ActionContext;
30
31
32 /***
33 * TokenHelperTest
34 *
35 */
36 public class TokenHelperTest extends TestCase {
37
38 private Map session;
39
40
41 public void testSetToken() {
42 String token = TokenHelper.setToken();
43 assertEquals(token, session.get(TokenHelper.DEFAULT_TOKEN_NAME));
44 }
45
46 public void testSetTokenWithName() {
47 String tokenName = "myTestToken";
48 String token = TokenHelper.setToken(tokenName);
49 assertEquals(token, session.get(tokenName));
50 }
51
52 public void testValidToken() {
53 String tokenName = "validTokenTest";
54 String token = TokenHelper.setToken(tokenName);
55 assertEquals(token, session.get(tokenName));
56 ActionContext.getContext().getParameters().put(TokenHelper.TOKEN_NAME_FIELD, new String[]{tokenName});
57 ActionContext.getContext().getParameters().put(tokenName, new String[]{token});
58 assertTrue(TokenHelper.validToken());
59 }
60
61 protected void setUp() throws Exception {
62 session = new HashMap();
63 Map params = new HashMap();
64 Map ctxMap = new HashMap();
65 ctxMap.put(ActionContext.SESSION, session);
66 ctxMap.put(ActionContext.PARAMETERS, params);
67 ActionContext ctx = new ActionContext(ctxMap);
68 ActionContext.setContext(ctx);
69 }
70
71 protected void tearDown() {
72 ActionContext.setContext(null);
73 }
74 }
75