Allow to chain ContextConsumer implementations
Closes gh-26723
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright 2012-2019 the original author or authors.
|
||||
* Copyright 2012-2021 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.
|
||||
@@ -17,6 +17,7 @@
|
||||
package org.springframework.boot.test.context.runner;
|
||||
|
||||
import org.springframework.context.ApplicationContext;
|
||||
import org.springframework.util.Assert;
|
||||
|
||||
/**
|
||||
* Callback interface used to process an {@link ApplicationContext} with the ability to
|
||||
@@ -38,4 +39,20 @@ public interface ContextConsumer<C extends ApplicationContext> {
|
||||
*/
|
||||
void accept(C context) throws Throwable;
|
||||
|
||||
/**
|
||||
* Returns a composed {@code ContextConsumer} that performs, in sequence, this
|
||||
* operation followed by the {@code after} operation.
|
||||
* @param after the operation to perform after this operation
|
||||
* @return a composed {@code ContextConsumer} that performs in sequence this operation
|
||||
* followed by the {@code after} operation
|
||||
* @since 2.6.0
|
||||
*/
|
||||
default ContextConsumer<C> andThen(ContextConsumer<? super C> after) {
|
||||
Assert.notNull(after, "After must not be null");
|
||||
return (context) -> {
|
||||
accept(context);
|
||||
after.accept(context);
|
||||
};
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user