KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > com > sslexplorer > core > CoreUtilTest


1 /*
2  * SSL-Explorer
3  *
4  * Copyright (C) 2003-2006 3SP LTD. All Rights Reserved
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2 of
9  * the License, or (at your option) any later version.
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public
16  * License along with this program; if not, write to the Free Software
17  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18  */
19             
20 package com.sslexplorer.core;
21
22 import static org.junit.Assert.assertEquals;
23
24 import org.junit.Test;
25
26 public class CoreUtilTest {
27
28     @Test
29     public void removeParameterFromPath() throws Exception JavaDoc {
30         assertEquals("First parm in a list",
31             "/blah.do?parm1=val1&parm2=val2",
32             CoreUtil.removeParameterFromPath("/blah.do?msg=123&parm1=val1&parm2=val2", "msg"));
33         assertEquals("Second parm in a list",
34             "/blah.do?parm1=1&parm2=val2",
35             CoreUtil.removeParameterFromPath("/blah.do?parm1=1&msg=123&parm2=val2", "msg"));
36         assertEquals("Last parm in a list",
37             "/blah.do?parm1=1&parm2=val2",
38             CoreUtil.removeParameterFromPath("/blah.do?parm1=1&parm2=val2&msg=123", "msg"));
39         assertEquals("Not in a list",
40             "/blah.do?parm1=1&parm2=val2",
41             CoreUtil.removeParameterFromPath("/blah.do?parm1=1&parm2=val2&msg=123", "msg"));
42     }
43     
44     @Test
45     public void filterJavascript() throws Exception JavaDoc {
46         assertEquals("/error.jsp", CoreUtil.filterSafeURI("javascript:alert('hello')", "/error.jsp"));
47         assertEquals("/ok.do", CoreUtil.filterSafeURI("/ok.do", "/error.jsp"));
48         assertEquals("ok.do", CoreUtil.filterSafeURI("ok.do", "/error.jsp"));
49         assertEquals("http://3sp.com/ok.do", CoreUtil.filterSafeURI("http://3sp.com/ok.do", "/error.jsp"));
50         assertEquals("https://3sp.com/ok.do", CoreUtil.filterSafeURI("https://3sp.com/ok.do", "/error.jsp"));
51     }
52 }
Popular Tags