feat(context): Add autoFlush.

cartant · Aug 21, 2017 · 5f9ff6a · 5f9ff6a
1 parent fa03d18
commit 5f9ff6a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/source/context.ts b/source/context.ts
@@ -15,6 +15,7 @@ import { Expect } from "./expect";
 
 export class Context {
 
+    public autoFlush = true;
     public configure = configure;
 
     constructor(public testScheduler: TestScheduler) {}

diff --git a/source/marbles.ts b/source/marbles.ts
@@ -20,6 +20,8 @@ export function marbles(test: (context: Context, ...args: any[]) => any): (...ar
         const testScheduler = new TestScheduler((a, b) => observableMatcher(a, b, get("assert"), get("assertDeepEqual")));
         const context = new Context(testScheduler);
         test.apply(null, [context].concat(args));
-        testScheduler.flush();
+        if (context.autoFlush) {
+            testScheduler.flush();
+        }
     };
 }