Fix DoubleObjectWithKey type; add respective test; Rename generics

WaleedTayeb · May 11, 2017 · e86769c · e86769c
1 parent 3e895af
commit e86769c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/index.d.ts b/index.d.ts
@@ -2,11 +2,11 @@
 // types and interfaces
 // ----------------------------------------------------------------------------
 
-export type DoubledObject<Subject> = Subject;
+export type DoubledObject<T> = T;
 
-export type DoubledObjectWithKey<Key extends string> = {};
+export type DoubledObjectWithKey<T extends string> = { [K in T] };
 
-export type TestDouble<Function> = Function;
+export type TestDouble<T> = T;
 
 export type TestDoubleConstructor<T> = Constructor<T>;
 
@@ -156,7 +156,7 @@ export function object<T>(constructor: Constructor<T> ): DoubledObject<T>;
  * @param {Key[]} props Array of properties.
  * @returns {DoubledObjectWithKey<Key>}
  */
-export function object<Key extends string>(props: Key[]): DoubledObjectWithKey<Key>;
+export function object<T extends string>(props: T[]): DoubledObjectWithKey<T>;
 
 /**
  * Create a fake empty object that is cast as the generic using a Proxy object.

diff --git a/test/src/typescript/test.ts b/test/src/typescript/test.ts
@@ -43,6 +43,10 @@ td.when(dog.bark()).thenReturn("bark!");
 
 // td.object()
 
+const cat = td.object(["scratch","meow"]);
+td.when(cat.scratch()).thenReturn("scratch!");
+td.when(cat.meow()).thenReturn("meow!");
+
 const bird = td.object({ fly: function(){} });
 td.when(bird.fly()).thenReturn("fly!");