Andy Wilkinson
10 years ago
parent
c501b889af
commit
1864d79077
@ -0,0 +1,35 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2012-2014 the original author or authors.
|
||||||
|
*
|
||||||
|
* 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 org.springframework.boot.context.embedded.undertow;
|
||||||
|
|
||||||
|
import io.undertow.Undertow.Builder;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Callback interface that can be used to customize an Undertow {@link Builder}.
|
||||||
|
*
|
||||||
|
* @author Andy Wilkinson
|
||||||
|
* @since 1.2.0
|
||||||
|
* @see UndertowEmbeddedServletContainerFactory
|
||||||
|
*/
|
||||||
|
public interface UndertowBuilderCustomizer {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @param builder the {@code Builder} to customize
|
||||||
|
*/
|
||||||
|
void customize(Builder builder);
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,23 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2012-2014 the original author or authors.
|
||||||
|
*
|
||||||
|
* 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Support for Undertow {@link org.springframework.boot.context.embedded.EmbeddedServletContainer EmbeddedServletContainers}.
|
||||||
|
*
|
||||||
|
* @see org.springframework.boot.context.embedded.undertow.UndertowEmbeddedServletContainerFactory
|
||||||
|
*/
|
||||||
|
package org.springframework.boot.context.embedded.undertow;
|
||||||
|
|
||||||
@ -1,34 +1,96 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2012-2014 the original author or authors.
|
||||||
|
*
|
||||||
|
* 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 org.springframework.boot.context.embedded.undertow;
|
package org.springframework.boot.context.embedded.undertow;
|
||||||
|
|
||||||
|
import io.undertow.Undertow.Builder;
|
||||||
|
|
||||||
|
import java.util.Arrays;
|
||||||
|
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import org.springframework.boot.context.embedded.*;
|
import org.mockito.InOrder;
|
||||||
|
import org.springframework.boot.context.embedded.AbstractEmbeddedServletContainerFactory;
|
||||||
|
import org.springframework.boot.context.embedded.AbstractEmbeddedServletContainerFactoryTests;
|
||||||
|
import org.springframework.boot.context.embedded.ErrorPage;
|
||||||
|
import org.springframework.boot.context.embedded.ExampleServlet;
|
||||||
|
import org.springframework.boot.context.embedded.ServletRegistrationBean;
|
||||||
import org.springframework.http.HttpStatus;
|
import org.springframework.http.HttpStatus;
|
||||||
|
|
||||||
import static org.hamcrest.Matchers.equalTo;
|
import static org.hamcrest.Matchers.equalTo;
|
||||||
import static org.junit.Assert.assertThat;
|
import static org.junit.Assert.assertThat;
|
||||||
|
import static org.mockito.Matchers.anyObject;
|
||||||
|
import static org.mockito.Mockito.inOrder;
|
||||||
|
import static org.mockito.Mockito.mock;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Tests for {@link org.springframework.boot.context.embedded.undertow.UndertowEmbeddedServletContainerFactory} and
|
* Tests for {@link UndertowEmbeddedServletContainerFactory} and
|
||||||
* {@link org.springframework.boot.context.embedded.undertow.UndertowEmbeddedServletContainer}.
|
* {@link UndertowEmbeddedServletContainer} .
|
||||||
*
|
*
|
||||||
* @author Ivan Sopov
|
* @author Ivan Sopov
|
||||||
|
* @author Andy Wilkinson
|
||||||
*/
|
*/
|
||||||
public class UndertowEmbeddedServletContainerFactoryTests extends AbstractEmbeddedServletContainerFactoryTests
|
public class UndertowEmbeddedServletContainerFactoryTests extends
|
||||||
{
|
AbstractEmbeddedServletContainerFactoryTests {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected UndertowEmbeddedServletContainerFactory getFactory()
|
protected UndertowEmbeddedServletContainerFactory getFactory() {
|
||||||
{
|
return new UndertowEmbeddedServletContainerFactory(0);
|
||||||
return new UndertowEmbeddedServletContainerFactory();
|
}
|
||||||
}
|
|
||||||
|
@Test
|
||||||
@Test
|
public void errorPage404() throws Exception {
|
||||||
public void errorPage404() throws Exception {
|
AbstractEmbeddedServletContainerFactory factory = getFactory();
|
||||||
AbstractEmbeddedServletContainerFactory factory = getFactory();
|
factory.addErrorPages(new ErrorPage(HttpStatus.NOT_FOUND, "/hello"));
|
||||||
factory.addErrorPages(new ErrorPage(HttpStatus.NOT_FOUND, "/hello"));
|
this.container = factory.getEmbeddedServletContainer(new ServletRegistrationBean(
|
||||||
this.container = factory.getEmbeddedServletContainer(new ServletRegistrationBean(new ExampleServlet(), "/hello"));
|
new ExampleServlet(), "/hello"));
|
||||||
this.container.start();
|
this.container.start();
|
||||||
assertThat(getResponse("http://localhost:8080/hello"), equalTo("Hello World"));
|
assertThat(getResponse(getLocalUrl("/hello")), equalTo("Hello World"));
|
||||||
assertThat(getResponse("http://localhost:8080/not-found"), equalTo("Hello World"));
|
assertThat(getResponse(getLocalUrl("/not-found")), equalTo("Hello World"));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void setNullUndertowBuilderCustomizersThrows() {
|
||||||
|
UndertowEmbeddedServletContainerFactory factory = getFactory();
|
||||||
|
this.thrown.expect(IllegalArgumentException.class);
|
||||||
|
this.thrown.expectMessage("undertowBuilderCustomizers must not be null");
|
||||||
|
factory.setUndertowBuilderCustomizers(null);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void addNullContextCustomizersThrows() {
|
||||||
|
UndertowEmbeddedServletContainerFactory factory = getFactory();
|
||||||
|
this.thrown.expect(IllegalArgumentException.class);
|
||||||
|
this.thrown.expectMessage("undertowBuilderCustomizers must not be null");
|
||||||
|
factory.addUndertowBuilderCustomizers((UndertowBuilderCustomizer[]) null);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void builderCustomizers() throws Exception {
|
||||||
|
UndertowEmbeddedServletContainerFactory factory = getFactory();
|
||||||
|
UndertowBuilderCustomizer[] customizers = new UndertowBuilderCustomizer[4];
|
||||||
|
for (int i = 0; i < customizers.length; i++) {
|
||||||
|
customizers[i] = mock(UndertowBuilderCustomizer.class);
|
||||||
|
}
|
||||||
|
factory.setUndertowBuilderCustomizers(Arrays.asList(customizers[0],
|
||||||
|
customizers[1]));
|
||||||
|
factory.addUndertowBuilderCustomizers(customizers[2], customizers[3]);
|
||||||
|
this.container = factory.getEmbeddedServletContainer();
|
||||||
|
InOrder ordered = inOrder((Object[]) customizers);
|
||||||
|
for (UndertowBuilderCustomizer customizer : customizers) {
|
||||||
|
ordered.verify(customizer).customize((Builder) anyObject());
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue