URIBrokerConfigTests.java example

Explorer
citrus-master
/*
 * Copyright (c) 2002-2012 Alibaba Group Holding Limited.
 * All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.alibaba.citrus.service.uribroker;

import static com.alibaba.citrus.test.TestUtil.*;
import static com.alibaba.citrus.util.CollectionUtil.*;
import static java.util.Collections.*;
import static org.junit.Assert.*;

import java.util.List;
import javax.servlet.http.HttpServletRequest;

import com.alibaba.citrus.service.pull.PullService;
import com.alibaba.citrus.service.uribroker.impl.URIBrokerServiceImpl;
import com.alibaba.citrus.service.uribroker.interceptor.URIBrokerInterceptor;
import com.alibaba.citrus.service.uribroker.interceptor.URIBrokerPathInterceptor;
import com.alibaba.citrus.service.uribroker.uri.GenericURIBroker;
import com.alibaba.citrus.service.uribroker.uri.URIBroker;
import com.alibaba.citrus.service.uribroker.uri.URIBroker.URIType;
import com.alibaba.citrus.util.i18n.LocaleUtil;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.springframework.context.ApplicationContext;

/**
 * 测试uribroker的配置。
 *
 * @author Michael Zhou
 */
public class URIBrokerConfigTests extends AbstractURIBrokerServiceTests {
    private GenericURIBroker uri;

    @Before
    public void init() throws Exception {
        LocaleUtil.setContext(null, "UTF-8");
    }

    @After
    public void dispose() {
        LocaleUtil.resetContext();
    }

    @Test
    public void withoutRequest() {
        ApplicationContext factory = createContext("services.xml", null);
        service = (URIBrokerServiceImpl) factory.getBean("uriBrokerService");

        assertNotNull(service);
        assertNull(getFieldValue(service, "request", HttpServletRequest.class));
        assertSame(service, factory.getBean("uris"));

        uri = (GenericURIBroker) service.getURIBroker("link");
        assertEquals("http:///", uri.toString());
    }

    @Test
    public void link_URIBroker() {
        uri = (GenericURIBroker) service.getURIBroker("link");

        assertEquals(null, uri.getURIType());

        assertEquals("http://localhost/", uri.render());
        assertEquals("http://user:pass@www.alibaba.com/aa/bb",
                     uri.setServerURI("http://user:pass@www.alibaba.com/aa/bb").render());
        assertEquals("http://www.alibaba.com/", uri.setServerName("www.alibaba.com").render());
        assertEquals("http://localhost:8080/", uri.setServerPort(8080).render());
        assertEquals("http://localhost/", uri.setServerPort(80).render());
        assertEquals("https://localhost:80/", uri.setServerScheme("https").render());
        assertEquals("http://localhost/aa/bb", uri.addPath("aa").addPath("bb").render());

        // fork
        GenericURIBroker newUri = (GenericURIBroker) uri.setServerURI("http://www.taobao.com/aa/bb").fork();

        assertEquals("http://www.taobao.com/aa/bb/cc", newUri.addPath("cc").render());
        assertEquals("http://www.taobao.com/aa/bb/dd", newUri.addPath("dd").render());
    }

    @Test
    public void link2_URIBroker() {
        uri = (GenericURIBroker) service.getURIBroker("link2");

        assertEquals(URIType.relative, uri.getURIType());

        assertEquals("http://myuser2:mypass2@myservername2:1234/aaa/a1/bbb/ccc/ddd?"
                     + "aaa=1111&bbb=2222&ccc=3333#myreference2", uri.render());

        assertEquals("http://myuser2:mypass2@myservername2:1234/aaa/a1/bbb/ccc/ddd?"
                     + "aaa=1111&bbb=2222&ccc=3333#myreference2", uri.toString());
    }

    @Test
    public void link3_URIBroker_extendsLink2_overrideWithEmpty() {
        uri = (GenericURIBroker) service.getURIBroker("link3");
        assertEquals(URIType.relative, uri.getURIType());
        assertEquals("https:///aaa/a1/bbb/ccc/ddd?aaa=1111&bbb=2222&ccc=3333", uri.toString());
    }

    @Test
    public void link4_serverURI() {
        uri = (GenericURIBroker) service.getURIBroker("link4");
        assertEquals("http://user:pass@server:1234/a/b/c?a=1&b=+&c=#ref", uri.toString());
        assertEquals("http://user:pass@server:1234/a/b/c", uri.getServerURI());
    }

    @Test
    public void mylink_URIBoker_variableReplacement() {
        uri = (GenericURIBroker) service.getURIBroker("mylink");
        assertEquals("http://www.taobao.com/", uri.render());
    }

    @Test
    public void linkCharset_URLEncoding() {
        // 未指明charset的，使用context中指定的charset进行URL encoding。
        uri = (GenericURIBroker) service.getURIBroker("link");
        assertEquals("http://localhost/?str=%E4%B8%AD%E5%9B%BD", uri.addQueryData("str", "中国").render());

        // 指明charset的，使用指定的charset进行URL encoding。
        uri = (GenericURIBroker) service.getURIBroker("linkCharset");
        assertEquals("http://localhost/?str=%D6%D0%B9%FA", uri.addQueryData("str", "中国").render());
    }

    @Test
    public void linkWithInterceptor_URIBroker() {
        PullService pull = (PullService) factory.getBean("pullService");

        // 确保uri interceptor在render之前没有被执行过
        // 曾经的bug：pull service在logging为debug时，会提前执行uri.interceptors。
        uri = (GenericURIBroker) pull.getContext().pull("linkWithInterceptor");
        assertTrue(uri.addQueryData("path", "hello").render().startsWith("http://www.mydomain.com/hello?path=hello&r="));

        assertTrue(uri.toString().startsWith("http://localhost/"));
        assertTrue(uri.render().startsWith("http://www.mydomain.com/abc?r="));

        // 添加另一个interceptor
        uri.addInterceptor(new URIBrokerInterceptor() {
            public void perform(URIBroker broker) {
                broker.addQueryData("id", 1);
            }
        });

        assertTrue(uri.render().startsWith("http://www.mydomain.com/abc?id=1&r="));
    }

    @Test
    public void linkWithPathInterceptor_URIBroker() {
        uri = (GenericURIBroker) service.getURIBroker("linkWithPathInterceptor");

        assertEquals("http://www.mydomain.com/abc/def", uri.render());

        // 添加另一个interceptor
        uri.addInterceptor(new URIBrokerPathInterceptor() {
            public void perform(URIBroker broker) {
                broker.setServerName("www.mydomain1.com");
            }

            public String perform(URIBroker broker, String path) {
                broker.addQueryData("id", 1);
                return path;
            }
        });

        assertEquals("http://www.mydomain1.com/abc/def?id=1", uri.render());
    }

    @Test
    public void not_requestAware() {
        service = (URIBrokerServiceImpl) factory.getBean("not-requestAware");

        for (String name : service.getNames()) {
            URIBroker uri = service.getURIBroker(name);

            if ("link-request".equals(name)) {
                assertEquals("http://localhost/", uri.render());
            } else {
                assertEquals("http:///", uri.render());
                assertEquals("http://www.alibaba.com:9999/", uri.setServerName("www.alibaba.com").setServerPort(9999)
                                                                .render());
            }
        }
    }

    @Test
    public void standalone() {
        URIBroker broker = (URIBroker) factory.getBean("standalone_uri");
        assertEquals("http://www.taobao.com/hello", broker.toString());
    }

    @Test
    public void noCharset() {
        service = (URIBrokerServiceImpl) factory.getBean("noDefaultCharset");

        URIBroker link1 = service.getURIBroker("a");
        assertEquals(null, link1.getCharset());
        assertEquals("http://localhost/?name=%E4%B8%AD%E5%9B%BD", link1.addQueryData("name", "中国").render()); // from context

        URIBroker link2 = service.getURIBroker("b");
        assertEquals("GBK", link2.getCharset());
        assertEquals("http://localhost/?name=%D6%D0%B9%FA", link2.addQueryData("name", "中国").render());
    }

    @Test
    public void defaultCharset() {
        service = (URIBrokerServiceImpl) factory.getBean("withDefaultCharset");

        URIBroker link1 = service.getURIBroker("x");
        assertEquals("UTF-8", link1.getCharset());
        assertEquals("http://localhost/?name=%E4%B8%AD%E5%9B%BD", link1.addQueryData("name", "中国").render());

        URIBroker link2 = service.getURIBroker("y");
        assertEquals("GBK", link2.getCharset());
        assertEquals("http://localhost/?name=%D6%D0%B9%FA", link2.addQueryData("name", "中国").render());
    }

    @Test
    public void uriBean() {
        service = (URIBrokerServiceImpl) factory.getBean("uri_bean");

        URIBroker bean = service.getURIBroker("mybean");
        assertEquals("https://myserver/aa/bb/cc", bean.render());
    }

    @Test
    public void importAndOverride() {
        service = (URIBrokerServiceImpl) factory.getBean("import");
        URIBrokerServiceImpl imported = (URIBrokerServiceImpl) factory.getBean("uris");

        // check exposed names
        List<String> importedNames = createArrayList(imported.getExposedNames());
        List<String> names = createArrayList(service.getExposedNames());

        names.add("link2");

        sort(importedNames);
        sort(names);

        assertEquals(importedNames, names);

        // check all names
        importedNames = createArrayList(imported.getNames());
        names = createArrayList(service.getNames());

        importedNames.add("newlink");

        sort(importedNames);
        sort(names);

        assertEquals(importedNames, names);

        // check uris
        for (String name : names) {
            URIBroker uri = service.getURIBroker(name);
            URIBroker importedUri = imported.getURIBroker(name);

            if (uri != null) {
                uri = uri.getParent();
                assertFalse(uri.isAutoReset());
            }

            if (importedUri != null) {
                importedUri = importedUri.getParent();
                assertFalse(importedUri.isAutoReset());
            }

            if ("link2".equals(name)) {
                assertEquals("http:///", uri.toString());
                assertEquals(
                        "http://myuser2:mypass2@myservername2:1234/aaa/a1/bbb/ccc/ddd?aaa=1111&bbb=2222&ccc=3333#myreference2",
                        importedUri.toString());
            } else if ("newlink".equals(name)) {
                assertEquals("http://www.taobao.com/newlink", uri.toString());
                assertNull(importedUri);
            } else {
                assertNotNull(uri);
                assertSame(importedUri, uri);
            }
        }
    }
}