Assert the full execution of extend() tests

Author: ZacSweers

src/test/java/io/reactivex/flowable/FlowableTests.java

@@ -1114,7 +1114,7 @@ public void testForEachWithNull() {
     public void testExtend() {
         final TestSubscriber<Object> subscriber = new TestSubscriber<Object>();
         final Object value = new Object();
-        Flowable.just(value).to(new Function<Flowable<Object>, Object>() {
+        Object returned = Flowable.just(value).to(new Function<Flowable<Object>, Object>() {
             @Override
             public Object apply(Flowable<Object> onSubscribe) {
                     onSubscribe.subscribe(subscriber);
@@ -1124,13 +1124,14 @@ public Object apply(Flowable<Object> onSubscribe) {
                     return subscriber.values().get(0);
                 }
         });
+        assertSame(returned, value);
     }
 
     @Test
     public void testAsExtend() {
         final TestSubscriber<Object> subscriber = new TestSubscriber<Object>();
         final Object value = new Object();
-        Flowable.just(value).as(new FlowableConverter<Object, Object>() {
+        Object returned = Flowable.just(value).as(new FlowableConverter<Object, Object>() {
             @Override
             public Object apply(Flowable<Object> onSubscribe) {
                     onSubscribe.subscribe(subscriber);
@@ -1140,6 +1141,7 @@ public Object apply(Flowable<Object> onSubscribe) {
                     return subscriber.values().get(0);
                 }
         });
+        assertSame(returned, value);
     }
 
     @Test

src/test/java/io/reactivex/observable/ObservableTest.java

@@ -1153,7 +1153,7 @@ public void testForEachWithNull() {
     public void testExtend() {
         final TestObserver<Object> subscriber = new TestObserver<Object>();
         final Object value = new Object();
-        Observable.just(value).to(new Function<Observable<Object>, Object>() {
+        Object returned = Observable.just(value).to(new Function<Observable<Object>, Object>() {
             @Override
             public Object apply(Observable<Object> onSubscribe) {
                     onSubscribe.subscribe(subscriber);
@@ -1163,13 +1163,14 @@ public Object apply(Observable<Object> onSubscribe) {
                     return subscriber.values().get(0);
                 }
         });
+        assertSame(returned, value);
     }
 
     @Test
     public void testAsExtend() {
         final TestObserver<Object> subscriber = new TestObserver<Object>();
         final Object value = new Object();
-        Observable.just(value).as(new ObservableConverter<Object, Object>() {
+        Object returned = Observable.just(value).as(new ObservableConverter<Object, Object>() {
             @Override
             public Object apply(Observable<Object> onSubscribe) {
                 onSubscribe.subscribe(subscriber);
@@ -1179,6 +1180,7 @@ public Object apply(Observable<Object> onSubscribe) {
                 return subscriber.values().get(0);
             }
         });
+        assertSame(returned, value);
     }
 
     @Test