Fix take while method (#94)

Author: emil-bar

flows/src/main/java/com/softwaremill/jox/flows/Flow.java

@@ -1078,6 +1078,11 @@ public Flow<T> takeWhile(Predicate<T> f, boolean includeFirstFailing) {
                         throw new BreakException();
                     }
                 });
+            } catch (JoxScopeExecutionException e) {
+                if (!(e.getCause() instanceof BreakException)) {
+                    throw e;
+                }
+                // done
             } catch (BreakException e) {
                 // done
             }

flows/src/test/java/com/softwaremill/jox/flows/FlowTest.java

@@ -265,6 +265,15 @@ void shouldNotTakeIfPredicateFailsForFirstOrMoreElements() throws Exception {
         assertEquals(List.of(), flow.runToList());
     }
 
+    @Test
+    void shouldTakeWhileFromAsyncFlow() throws Exception {
+        // given
+        Flow<Integer> flow = Flows.fromValues(3, 2, 1).buffer().takeWhile(x -> x > 2, false);
+
+        // when & then
+        assertEquals(List.of(3), flow.runToList());
+    }
+
     @Test
     void shouldNotThrottleEmptySource() throws Exception {
         // given