fix(time): Support explicit durations.

cartant · May 18, 2018 · 4a7ac1d · 4a7ac1d
1 parent 604db04
commit 4a7ac1d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/fixtures/mocha/passing-spec.ts b/fixtures/mocha/passing-spec.ts
@@ -474,5 +474,11 @@ describe("rxjs-marbles", () => {
             const expected = m.cold("----(a|)");
             m.expect(source.pipe(delay(duration, m.scheduler))).toBeObservable(expected);
         }));
+
+        it("should support explicit durations in time", marbles((m) => {
+
+            const duration = m.time(" 2ms |");
+            expect(duration).to.equal(2);
+        }));
     });
 });
diff --git a/source/context-run.ts b/source/context-run.ts
@@ -119,6 +119,11 @@ export class RunContext implements Context {
 
     time(marbles: string): number {
 
+        const messages = TestScheduler.parseMarbles(marbles, undefined, undefined, undefined, true);
+        const complete = messages.find(({ notification }) => notification.kind === "C");
+        if (complete) {
+            return complete.frame;
+        }
         return this.scheduler.createTime(marbles);
     }
 }