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