import org.gradle.internal.remote.internal.hub.protocol.EndOfStream;
import org.gradle.internal.remote.internal.hub.protocol.InterHubMessage;

public class QueueInitializer {
    private EndOfStream endOfStream;

    void onStatefulMessage(InterHubMessage message) {
        if (message instanceof EndOfStream) {
            endOfStream = (EndOfStream) message;
        } else {

        then:
        0 * queue._
    }

    def "discards message on end of stream received"() {
        given:
        def closed = new EndOfStream()

        initializer.onStatefulMessage(closed)

        when:
        initializer.onQueueAdded(queue)

        then:
        1 * queue.dispatch(closed)
        0 * queue._
    }

        }
        return queue;
    }

    public void queue(InterHubMessage message) {
        if (message.getDelivery() == InterHubMessage.Delivery.Stateful) {
            initializer.onStatefulMessage(message);
        }
        if (message instanceof Routable) {
            Routable routableMessage = (Routable) message;
            getChannel(routableMessage.getChannel()).dispatch(message);

                    if (endpoints.isEmpty()) {
                        return;
                    }
                    if (message.getDelivery() == InterHubMessage.Delivery.Stateful) {
                        initializer.onStatefulMessage(message);
                    }
                    for (EndPointQueue endpoint : endpoints) {
                        endpoint.dispatch(message);
                    }
                    queue.removeFirst();