ruletest.java

jsp中 urlRewrite的源代码 很有用的喔

/**
 * Copyright (c) 2005, Paul Tuckey
 * All rights reserved.
 *
 * Each copy or derived work must preserve the copyright notice and this
 * notice unmodified.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 */
package org.tuckey.web.filters.urlrewrite;

import junit.framework.TestCase;
import org.tuckey.web.MockRequest;
import org.tuckey.web.MockResponse;
import org.tuckey.web.filters.urlrewrite.utils.Log;

import javax.servlet.ServletException;
import java.io.IOException;

/**
 * @author Paul Tuckey
 * @version $Revision: 1.18 $ $Date: 2005/12/07 10:22:12 $
 */
public class RuleTest extends TestCase {

    MockResponse response;
    MockRequest request;

    public void setUp() {
        Log.setLevel("TRACE");
        response = new MockResponse();
        request = new MockRequest();
    }

    public void testRule01() throws IOException, ServletException {
        Rule rule = new Rule();
        rule.setFrom("simple(ass)");
        rule.setTo("$1simple");
        rule.initialise(null);
        MockRequest request = new MockRequest("simpleass");
        RewrittenUrl rewrittenUrl = rule.execute(request.getRequestURI(), request, response);

        assertEquals("forward should be default type", "forward", rule.getToType());
        assertEquals("asssimple", rewrittenUrl.getTarget());
        assertTrue("Should be a forward", rewrittenUrl.isForward());
    }

    public void testRuleNullTo() throws IOException, ServletException {
        Rule rule = new Rule();
        rule.setFrom("from");
        rule.setTo("null");
        rule.initialise(null);
        SetAttribute setAttribute1 = new SetAttribute();
        setAttribute1.setType("status");
        setAttribute1.setValue("302");
        rule.addSetAttribute(setAttribute1);
        rule.initialise(null);

        MockRequest request = new MockRequest("from");
        RewrittenUrl rewrittenUrl = rule.execute(request.getRequestURI(), request, response);
        assertEquals(302, response.getStatus());
        assertTrue(rewrittenUrl.isStopFilterChain());

    }

    public void testRuleInclude() throws IOException, ServletException {
        Rule rule = new Rule();
        rule.setFrom("from");
        rule.setTo("to");
        rule.setToType("pre-include");
        rule.initialise(null);
        request.setRequestURI("from");
        RewrittenUrl rewrittenUrl = rule.execute(request.getRequestURI(), request, response);

        assertEquals("pre-include", rule.getToType());
        assertTrue("Should be an pre include", rewrittenUrl.isPreInclude());
    }

    public void testRuleBackRef() throws IOException, ServletException {
        Condition c = new Condition();
        c.setName("hdr");
        c.setValue("aaa([a-z]+)cc(c)");
        Rule rule = new Rule();
        rule.setFrom("([a-z])rom");
        rule.setTo("from match: $1, backref1: %1, backref2: %2, bad backref: %a % %99 %%88, escaped backref: \\%2");
        rule.addCondition(c);
        rule.initialise(null);
        MockRequest request = new MockRequest("from");
        request.setHeader("hdr", "aaafffccc");
        RewrittenUrl rewrittenUrl = rule.execute(request.getRequestURI(), request, response);

        assertEquals("Should have replaced back reference",
                "from match: f, backref1: fff, backref2: c, bad backref: %a % 9 %8, escaped backref: %2", rewrittenUrl.getTarget());

        assertTrue(rule.isToContainsBackReference());
        assertFalse(rule.isToContainsVariable());

    }

    /**
     * here's my urlrewriter.xml :
<condition type="server-name">^([^.]+)\.domain\.com</condition>
<from>^/(.*)</from>
<to type="redirect" last="true">/%1/$1</to>

Now the url with host header 'http://test.domain.com' should be
redirected to 'http://domain.com/test'. But for some reason it doesnt
replace the %1 with 'test'. The redirected url is :
http://domain.com/%1
     */
    public void testRuleBackRefHost() throws IOException, ServletException {
        Condition c = new Condition();
        c.setType("server-name");
        c.setValue("^([^.]+)\\.domain\\.com");
        Rule rule = new Rule();
        rule.setFrom("^/(.*)");
        rule.setTo("/%1/$1");
        rule.setToType("redirect");
        rule.setToLast("true");
        rule.addCondition(c);
        rule.initialise(null);
        MockRequest request = new MockRequest("/from");
        request.setServerName("server.domain.com");
        RewrittenUrl rewrittenUrl = rule.execute(request.getRequestURI(), request, response);

        assertEquals("Should have replaced back reference",
                "/server/from", rewrittenUrl.getTarget());

        assertTrue(rule.isToContainsBackReference());

    }

    public void testRuleBackRefMixed() throws IOException, ServletException {
        Condition c = new Condition();
        c.setName("hdr");
        c.setValue("aaa([a-z]+)ccc");
        Rule rule = new Rule();
        rule.setFrom("([a-z])rom");
        rule.setTo("$1 to %{remote-host} %1 here \\%2");
        rule.addCondition(c);
        rule.initialise(null);
        MockRequest request = new MockRequest("from");
        request.setRemoteHost("server!");
        request.addHeader("hdr", "aaabbbccc");
        RewrittenUrl rewrittenUrl = rule.execute(request.getRequestURI(), request, response);

        assertTrue(rule.isToContainsBackReference());
        assertTrue(rule.isToContainsVariable());

        assertEquals("Should have replaced back reference", "f to server! bbb here %2", rewrittenUrl.getTarget());
    }

    public void testRuleBackRefVar() throws IOException, ServletException {
        Rule rule = new Rule();
        rule.setFrom("from");
        rule.setTo("start ctx: %{context-path}, hdr: %{header:bananna} %{header:}%{::}%{%{}, escaped var: \\%{ignoreme!}, bad var: %{} %{wibble} end");
        rule.initialise(null);
        MockRequest request = new MockRequest("from");
        request.setContextPath("ctxpath");
        request.setRemoteHost("server!");
        request.addHeader("bananna", "bender");
        request.setServerPort(90210);
        RewrittenUrl rewrittenUrl = rule.execute(request.getRequestURI(), request, response);

        assertFalse(rule.isToContainsBackReference());
        assertTrue(rule.isToContainsVariable());

        assertEquals("forward should be default type", "forward", rule.getToType());
        assertEquals("start ctx: ctxpath, hdr: bender , escaped var: %{ignoreme!}, bad var:   end", rewrittenUrl.getTarget());
        assertTrue("Should be a forward", rewrittenUrl.isForward());
    }

    public void testRuleBadCondition() throws IOException, ServletException {
        Rule rule = new Rule();
        rule.setFrom("from");
        rule.setTo("to");
        Condition condition = new Condition();
        condition.setType("port");
        condition.setValue("aaa");
        rule.addCondition(condition);
        rule.initialise(null);
        MockRequest request = new MockRequest("from");
        RewrittenUrl rewrittenUrl = rule.execute(request.getRequestURI(), request, response);
        assertNull(rewrittenUrl);
    }

    public void testRuleConditionsOr() throws IOException, ServletException {
        Rule rule = new Rule();
        rule.setFrom("from");
        rule.setTo("to");
        Condition condition = new Condition();
        condition.setType("port");
        condition.setValue("90");
        condition.setNext("or");
        rule.addCondition(condition);
        Condition condition2 = new Condition();
        condition2.setType("port");
        condition2.setValue("99");
        rule.addCondition(condition2);
        rule.initialise(null);
        MockRequest request = new MockRequest("from");
        request.setServerPort(88);
        RewrittenUrl rewrittenUrl = rule.execute(request.getRequestURI(), request, response);
        assertNull("should not be rewritten", rewrittenUrl);

        request.setServerPort(90);
        RewrittenUrl rewrittenUrl2 = rule.execute(request.getRequestURI(), request, response);
        assertEquals("should not be rewritten", "to", rewrittenUrl2.getTarget());

        request.setServerPort(99);
        RewrittenUrl rewrittenUrl3 = rule.execute(request.getRequestURI(), request, response);
        assertEquals("should not be rewritten", "to", rewrittenUrl3.getTarget());
    }

    public void testRuleConditionsAnd() throws IOException, ServletException {
        Rule rule = new Rule();
        rule.setFrom("from");
        rule.setTo("to");
        Condition condition = new Condition();
        condition.setType("port");
        condition.setValue("90");
        rule.addCondition(condition);
        Condition condition2 = new Condition();
        condition2.setType("character-encoding");
        condition2.setValue("utf8");
        rule.addCondition(condition2);
        rule.initialise(null);
        MockRequest request = new MockRequest("from");
        request.setServerPort(88);
        RewrittenUrl rewrittenUrl = rule.execute(request.getRequestURI(), request, response);
        assertNull("should not be rewritten", rewrittenUrl);

        request.setServerPort(90);
        RewrittenUrl rewrittenUrl2 = rule.execute(request.getRequestURI(), request, response);
        assertNull("should not be rewritten", rewrittenUrl2);

        request.setCharacterEncoding("utf8");
        RewrittenUrl rewrittenUrl3 = rule.execute(request.getRequestURI(), request, response);
        assertEquals("should be rewritten", "to", rewrittenUrl3.getTarget());
    }

    public void testRuleBadRegexp() throws IOException, ServletException {
        Rule rule = new Rule();
        rule.setFrom("fro[m");
        rule.setTo("to");
        rule.initialise(null);
        MockRequest request = new MockRequest("from");
        RewrittenUrl rewrittenUrl = rule.execute(request.getRequestURI(), request, response);
        assertNull(rewrittenUrl);
    }

    public void testRuleCaseInsensitive() throws IOException, ServletException {
        Rule rule = new Rule();
        rule.setFrom("fRom");
        rule.setTo("to");