1 /*

     2  * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.

     8  *

     9  * This code is distributed in the hope that it will be useful, but WITHOUT

    10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    12  * version 2 for more details (a copy is included in the LICENSE file that

    13  * accompanied this code).

    14  *

    15  * You should have received a copy of the GNU General Public License version

    16  * 2 along with this work; if not, write to the Free Software Foundation,

    17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    18  *

    19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    20  * or visit www.oracle.com if you need additional information or have any

    21  * questions.

    22  */

    23 

    24 import org.reactivestreams.tck.TestEnvironment;

    25 import org.reactivestreams.tck.flow.FlowSubscriberBlackboxVerification;

    26 

    27 import java.net.http.HttpResponse.BodySubscriber;

    28 import java.net.http.HttpResponse.BodySubscribers;

    29 import java.nio.ByteBuffer;

    30 import java.util.List;

    31 import java.util.concurrent.CompletionStage;

    32 import java.util.concurrent.Flow.Publisher;

    33 import java.util.concurrent.Flow.Subscriber;

    34 import java.util.concurrent.Flow.Subscription;

    35 

    36 /* See TckDriver.java for more information */

    37 public class BodySubscribersOfPublisher

    38         extends FlowSubscriberBlackboxVerification<List<ByteBuffer>> {

    39 

    40     public BodySubscribersOfPublisher() {

    41         super(new TestEnvironment(450L));

    42     }

    43 

    44     /* The reason for overriding this method is that BodySubscribers.ofPublisher

    45        is somewhat tricky. It is not an independent Subscriber, but rather

    46        an adaptor from Subscriber to Publisher. Until the Subscriber that

    47        subscribed to that resulting Publisher requests anything, nothing

    48        happens. */

    49     @Override

    50     public void triggerFlowRequest(

    51             Subscriber<? super List<ByteBuffer>> subscriber)

    52     {

    53         BodySubscriber<Publisher<List<ByteBuffer>>> sub =

    54                 (BodySubscriber<Publisher<List<ByteBuffer>>>) subscriber;

    55         CompletionStage<Publisher<List<ByteBuffer>>> body = sub.getBody();

    56         Publisher<List<ByteBuffer>> pub = body.toCompletableFuture().join();

    57         pub.subscribe(new Subscriber<>() {

    58 

    59             @Override

    60             public void onSubscribe(Subscription subscription) {

    61                 subscription.request(Integer.MAX_VALUE);

    62             }

    63 

    64             @Override public void onNext(List<ByteBuffer> item) { }

    65             @Override public void onError(Throwable throwable) { }

    66             @Override public void onComplete() { }

    67         });

    68     }

    69 

    70     @Override

    71     public Subscriber<List<ByteBuffer>> createFlowSubscriber() {

    72         return BodySubscribers.ofPublisher();

    73     }

    74 

    75     @Override

    76     public List<ByteBuffer> createElement(int element) {

    77         return S.listOfBuffersFromBufferOfNBytes(element % 17);

    78     }

    79 }