updated promise support.

Author: Mathias Lorenzen

dist/spec/index.test.d.ts

@@ -1,9 +1,12 @@
+declare class Dummy {
+}
 export declare class Example {
     a: string;
     c(arg1: string, arg2: string): string;
     readonly d: number;
     v: string;
     received(stuff: number | string): void;
-    returnPromise(): Promise<number>;
+    returnPromise(): Promise<Dummy>;
     foo(): string;
 }
+export {};

dist/spec/index.test.js

@@ -4,6 +4,10 @@ import { Substitute, Arg } from '../src/Index';
 import { areArgumentsEqual } from '../src/Utilities';
 import { OmitProxyMethods, ObjectSubstitute } from '../src/Transformations';
 
+class Dummy {
+
+}
+
 export class Example {
 	a = "1337";
 
@@ -24,7 +28,7 @@ export class Example {
 	}
 
 	returnPromise() {
-		return Promise.resolve(1337);
+		return Promise.resolve(new Dummy());
 	}
 
 	foo() {
@@ -44,6 +48,13 @@ test.beforeEach(() => {
 	substitute = Substitute.for<Example>();
 });
 
+test('returning other fake from promise works', async t => {
+	const otherSubstitute = Substitute.for<Dummy>();
+	substitute.returnPromise().returns(Promise.resolve(otherSubstitute));
+
+	t.is(otherSubstitute, await substitute.returnPromise());
+});
+
 test('returning resolved promises works', async t => {
 	substitute.returnPromise().returns(Promise.resolve(1338));
 

dist/spec/index.test.js.map

@@ -174,6 +174,9 @@ export class Substitute {
                 if (property === 'constructor')
                     return () => thisProxy;
 
+                if(property === 'then')
+                    return void 0;
+
                 const currentPropertyContext = objectContext.property;
                 const addPropertyToObjectContext = () => {
                     const expected = objectContext.calls.expected;