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;
|
||||
|
||||
import io.undertow.Undertow.Builder;
|
||||
|
||||
import java.util.Arrays;
|
||||
|
||||
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 static org.hamcrest.Matchers.equalTo;
|
||||
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
|
||||
* {@link org.springframework.boot.context.embedded.undertow.UndertowEmbeddedServletContainer}.
|
||||
* Tests for {@link UndertowEmbeddedServletContainerFactory} and
|
||||
* {@link UndertowEmbeddedServletContainer} .
|
||||
*
|
||||
* @author Ivan Sopov
|
||||
* @author Andy Wilkinson
|
||||
*/
|
||||
public class UndertowEmbeddedServletContainerFactoryTests extends AbstractEmbeddedServletContainerFactoryTests
|
||||
{
|
||||
|
||||
@Override
|
||||
protected UndertowEmbeddedServletContainerFactory getFactory()
|
||||
{
|
||||
return new UndertowEmbeddedServletContainerFactory();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void errorPage404() throws Exception {
|
||||
AbstractEmbeddedServletContainerFactory factory = getFactory();
|
||||
factory.addErrorPages(new ErrorPage(HttpStatus.NOT_FOUND, "/hello"));
|
||||
this.container = factory.getEmbeddedServletContainer(new ServletRegistrationBean(new ExampleServlet(), "/hello"));
|
||||
this.container.start();
|
||||
assertThat(getResponse("http://localhost:8080/hello"), equalTo("Hello World"));
|
||||
assertThat(getResponse("http://localhost:8080/not-found"), equalTo("Hello World"));
|
||||
}
|
||||
public class UndertowEmbeddedServletContainerFactoryTests extends
|
||||
AbstractEmbeddedServletContainerFactoryTests {
|
||||
|
||||
@Override
|
||||
protected UndertowEmbeddedServletContainerFactory getFactory() {
|
||||
return new UndertowEmbeddedServletContainerFactory(0);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void errorPage404() throws Exception {
|
||||
AbstractEmbeddedServletContainerFactory factory = getFactory();
|
||||
factory.addErrorPages(new ErrorPage(HttpStatus.NOT_FOUND, "/hello"));
|
||||
this.container = factory.getEmbeddedServletContainer(new ServletRegistrationBean(
|
||||
new ExampleServlet(), "/hello"));
|
||||
this.container.start();
|
||||
assertThat(getResponse(getLocalUrl("/hello")), 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