src/test/java/io/reactivex/rxjava4/flowable/FlowableCovarianceTest.java - 4.x - githubmirror/RxJava

src/test/java/io/reactivex/rxjava4/flowable/FlowableCovarianceTest.java
178 строк6 KB
David Karnok
4.x: Streamable + takeUntil + groupBy + refactor + helpers (#8215)
05 июл 2026, 01:45
05 июл 2026, 01:45580b39c
/*
 * Copyright (c) 2016-present, RxJava Contributors.
 *
 * 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 io.reactivex.rxjava4.flowable;

import static org.junit.jupiter.api.Assertions.assertEquals;

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

import org.junit.jupiter.api.Test;

import io.reactivex.rxjava4.core.*;
import io.reactivex.rxjava4.functions.Function;
import io.reactivex.rxjava4.testsupport.TestSubscriberEx;

/**
 * Test super/extends of generics.
 *
 * See https://github.com/Netflix/RxJava/pull/331
 */
public class FlowableCovarianceTest extends RxJavaTest {

    /**
     * This won't compile if super/extends isn't done correctly on generics.
     */
    @Test
    public void covarianceOfFrom() {
        Flowable.<Movie> just(new HorrorMovie());
        Flowable.<Movie> fromIterable(new ArrayList<HorrorMovie>());
        // Observable.<HorrorMovie>from(new Movie()); // may not compile
    }

    @Test
    public void sortedList() {
        Comparator<Media> sortFunction = (_, _) -> 1;

        // this one would work without the covariance generics
        Flowable<Media> f = Flowable.just(new Movie(), new TVSeason(), new Album());
        f.toSortedList(sortFunction);

        // this one would NOT work without the covariance generics
        Flowable<Movie> f2 = Flowable.just(new Movie(), new ActionMovie(), new HorrorMovie());
        f2.toSortedList(sortFunction);
    }

    @Test
    public void groupByCompose() {
        Flowable<Movie> movies = Flowable.just(new HorrorMovie(), new ActionMovie(), new Movie());
        TestSubscriberEx<String> ts = new TestSubscriberEx<>();

        movies
        .groupBy((Function<Movie, Object>) Movie::getClass)
        .doOnNext(g -> System.out.println(g.getKey()))
        .flatMap((Function<GroupedFlowable<Object, Movie>, Publisher<String>>) g -> g
                .doOnNext(System.out::println)
                .compose(m -> m.concatWith(Flowable.just(new ActionMovie())))
                .map((Function<Object, String>) Object::toString))
        .subscribe(ts);
        ts.assertTerminated();
        ts.assertNoErrors();
        //        System.out.println(ts.getOnNextEvents());
        assertEquals(6, ts.values().size());
    }

    @SuppressWarnings("unused")
    @Test
    public void covarianceOfCompose() {
        Flowable<HorrorMovie> movie = Flowable.just(new HorrorMovie());
        Flowable<Movie> movie2 = movie.compose(t -> Flowable.just(new Movie()));
    }

    @SuppressWarnings("unused")
    @Test
    public void covarianceOfCompose2() {
        Flowable<Movie> movie = Flowable.<Movie> just(new HorrorMovie());
        Flowable<HorrorMovie> movie2 = movie.compose(t -> Flowable.just(new HorrorMovie()));
    }

    @SuppressWarnings("unused")
    @Test
    public void covarianceOfCompose3() {
        Flowable<Movie> movie = Flowable.<Movie>just(new HorrorMovie());
        Flowable<HorrorMovie> movie2 = movie.compose(t -> Flowable.just(new HorrorMovie())
                .map(v -> v)
        );
    }

    @SuppressWarnings("unused")
    @Test
    public void covarianceOfCompose4() {
        Flowable<HorrorMovie> movie = Flowable.just(new HorrorMovie());
        Flowable<HorrorMovie> movie2 = movie.compose(t1 -> t1.map(v -> v));
    }

    @Test
    public void composeWithDeltaLogic() {
        List<Movie> list1 = Arrays.asList(new Movie(), new HorrorMovie(), new ActionMovie());
        List<Movie> list2 = Arrays.asList(new ActionMovie(), new Movie(), new HorrorMovie(), new ActionMovie());
        Flowable<List<Movie>> movies = Flowable.just(list1, list2);
        movies.compose(deltaTransformer);
    }

    static Function<List<List<Movie>>, Flowable<Movie>> calculateDelta = listOfLists -> {
        if (listOfLists.size() == 1) {
            return Flowable.fromIterable(listOfLists.getFirst());
        } else {
            // diff the two
            List<Movie> newList = listOfLists.get(1);
            List<Movie> oldList = new ArrayList<>(listOfLists.get(0));

            Set<Movie> delta = new LinkedHashSet<>();
            delta.addAll(newList);
            // remove all that match in old
            delta.removeAll(oldList);

            // filter oldList to those that aren't in the newList
            oldList.removeAll(newList);

            // for all left in the oldList we'll create DROP events
            for (@SuppressWarnings("unused") Movie old : oldList) {
                delta.add(new Movie());
            }

            return Flowable.fromIterable(delta);
        }
    };

    static FlowableTransformer<List<Movie>, Movie> deltaTransformer = movieList -> movieList
        .startWithItem(new ArrayList<>())
        .buffer(2, 1)
        .skip(1)
        .flatMap(calculateDelta);

    /*
     * Most tests are moved into their applicable classes such as [Operator]Tests.java
     */

    static class Media {
    }

    static class Movie extends Media {
    }

    static class HorrorMovie extends Movie {
    }

    static class ActionMovie extends Movie {
    }

    static class Album extends Media {
    }

    static class TVSeason extends Media {
    }

    static class Rating {
    }

    static class CoolRating extends Rating {
    }

    static class Result {
    }

    static class ExtendedResult extends Result {
    }
}