fix(dispose): support drivers that don't return streams

Add check it output of a driver has .until()
motorcyclejs · Dec 19, 2015 · 786adb2 · 786adb2
1 parent 0a8f718
commit 786adb2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -11,9 +11,13 @@ const makeSinkProxies = drivers =>
 const callDrivers = (drivers, sinkProxies, disposableStream) =>
   Object.keys(drivers)
     .reduce((sources, driverName) => {
-      sources[driverName] =
+      const source =
         drivers[driverName](hold(sinkProxies[driverName].stream), driverName)
-          .until(disposableStream)
+
+      sources[driverName] = typeof source.until === `function` ?
+          source.until(disposableStream) :
+          source
+
       return sources
     }, {})