Skip to content

Commit e943d6f

Browse files
committed
fix(testscheduler): fix return types for creation method
1 parent c42845c commit e943d6f

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

src/scheduler/TestScheduler.ts

+8-8
Original file line numberDiff line numberDiff line change
@@ -68,9 +68,9 @@ class TestScheduler extends VirtualTimeScheduler {
6868
throw new Error('not implemented');
6969
}
7070

71-
public createColdObservable<T = string>(marble: string, value?: { [key: string]: T }, error?: any): Observable<T>;
72-
public createColdObservable<T = string>(message: Array<TestMessage<T>>): Observable<T>;
73-
public createColdObservable<T = string>(...args: Array<any>): Observable<T> {
71+
public createColdObservable<T = string>(marble: string, value?: { [key: string]: T }, error?: any): ColdObservable<T>;
72+
public createColdObservable<T = string>(message: Array<TestMessage<T>>): ColdObservable<T>;
73+
public createColdObservable<T = string>(...args: Array<any>): ColdObservable<T> {
7474
const [marbleValue, value, error] = args;
7575

7676
if (typeof marbleValue === 'string' && marbleValue.indexOf(SubscriptionMarbleToken.SUBSCRIBE) !== -1) {
@@ -80,20 +80,20 @@ class TestScheduler extends VirtualTimeScheduler {
8080
const messages = Array.isArray(marbleValue)
8181
? marbleValue
8282
: parseObservableMarble(marbleValue, value, error, false, this.frameTimeFactor);
83-
const observable = new ColdObservable<T>(messages, this);
83+
const observable = new ColdObservable<T>(messages as Array<TestMessage<T | Array<TestMessage<T>>>>, this);
8484
this.coldObservables.push(observable);
8585
return observable;
8686
}
8787

88-
public createHotObservable<T = string>(marble: string, value?: { [key: string]: T }, error?: any): Observable<T>;
89-
public createHotObservable<T = string>(message: Array<TestMessage<T>>): Observable<T>;
90-
public createHotObservable<T = string>(...args: Array<any>): Observable<T> {
88+
public createHotObservable<T = string>(marble: string, value?: { [key: string]: T }, error?: any): HotObservable<T>;
89+
public createHotObservable<T = string>(message: Array<TestMessage<T>>): HotObservable<T>;
90+
public createHotObservable<T = string>(...args: Array<any>): HotObservable<T> {
9191
const [marbleValue, value, error] = args;
9292

9393
const messages = Array.isArray(marbleValue)
9494
? marbleValue
9595
: parseObservableMarble(marbleValue, value, error, false, this.frameTimeFactor);
96-
const subject = new HotObservable<T>(messages, this);
96+
const subject = new HotObservable<T>(messages as Array<TestMessage<T | Array<TestMessage<T>>>>, this);
9797
this.hotObservables.push(subject);
9898
return subject;
9999
}

0 commit comments

Comments
 (0)