import org.gradle.internal.build.event.BuildEventSubscriptions;
import org.gradle.internal.invocation.BuildAction;

public abstract class SubscribableBuildAction implements BuildAction {
    private final BuildEventSubscriptions clientSubscriptions;

    public SubscribableBuildAction(BuildEventSubscriptions clientSubscriptions) {
        this.clientSubscriptions = clientSubscriptions;
    }

    public BuildActionRunner.Result execute(BuildAction action, BuildSessionContext buildSession) {
        if (action instanceof SubscribableBuildAction) {
            SubscribableBuildAction subscribableBuildAction = (SubscribableBuildAction) action;
            registerListenersForClientSubscriptions(subscribableBuildAction.getClientSubscriptions(), eventConsumer);
        }
        try {
            return delegate.execute(action, buildSession);

import org.gradle.internal.session.BuildSessionContext
import org.gradle.internal.session.BuildSessionActionExecutor
import org.gradle.tooling.events.OperationType
import org.gradle.tooling.internal.provider.action.SubscribableBuildAction
import spock.lang.Specification

class SubscribableBuildActionExecutorSpec extends Specification {

    def "removes listeners after executing build action"() {
        given:

import org.gradle.api.internal.StartParameterInternal;
import org.gradle.internal.build.event.BuildEventSubscriptions;
import org.gradle.tooling.internal.protocol.ModelIdentifier;

public class BuildModelAction extends SubscribableBuildAction {
    private final StartParameterInternal startParameter;
    private final String modelName;
    private final boolean runTasks;

import org.gradle.internal.build.event.BuildEventSubscriptions;
import org.gradle.tooling.internal.provider.serialization.SerializedPayload;

public class ClientProvidedBuildAction extends SubscribableBuildAction {
    private final StartParameterInternal startParameter;
    private final SerializedPayload action;
    private final boolean runTasks;

import org.gradle.internal.build.event.BuildEventSubscriptions;
import org.gradle.tooling.internal.provider.serialization.SerializedPayload;

public class ClientProvidedPhasedAction extends SubscribableBuildAction {
    private final StartParameterInternal startParameter;
    private final SerializedPayload phasedAction;
    private final boolean runTasks;

import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

public class TestExecutionRequestAction extends SubscribableBuildAction {
    private final StartParameterInternal startParameter;
    private final Set<InternalTestDescriptor> testDescriptors;
    private final Set<String> classNames;

Class <org.gradle.tooling.internal.provider.action.SubscribableBuildAction> is not annotated (directly or via its package) with @org.gradle.api.NonNullApi in (SubscribableBuildAction.java:0)