feat: retain error information

Author: KnisterPeter

Diff for: src/types.ts

@@ -143,8 +143,18 @@ function getEnumValues(ast: AstQuery, oneOfTypes: any): any[] {
       ]
       /:init *
     `);
+    if (!res[0]) {
+      const error = new Error('Failed to lookup enum values');
+      (error as any).loc = oneOfTypes.loc;
+      throw error;
+    }
     oneOfTypes = res[0];
   }
+  if (!oneOfTypes.elements) {
+    const error = new Error('Failed to lookup enum values');
+    (error as any).loc = oneOfTypes.loc;
+    throw error;
+  }
   return (oneOfTypes.elements as any[]).map((element: any) => {
     // fixme: This are not named references!
     if (element.type === 'StringLiteral') {
@@ -172,5 +182,10 @@ function getShapeProperties(ast: AstQuery, input: any): any[] {
     (error as any).loc = input.loc;
     throw error;
   }
+  if (!input.properties) {
+    const error = new Error('Failed to lookup shape properties');
+    (error as any).loc = input.loc;
+    throw error;
+  }
   return input.properties;
 }

Diff for: tests/error-reporting-test.ts

@@ -18,7 +18,7 @@ test.afterEach(() => {
   console.error = orignalConsoleError;
 });
 
-test('In case of error during type inference the error information should be retained', t => {
+test('In case of error during shape type inference the error information should be retained', t => {
   react2dts.generateFromSource(null, `
     import React from 'react';
 
@@ -32,3 +32,48 @@ test('In case of error during type inference the error information should be ret
   const idx = args.indexOf('Line 6:     someShape: React.PropTypes.shape(shape)');
   t.is(idx, 1);
 });
+
+test('In case of error during enum type inference the error information should be retained', t => {
+  react2dts.generateFromSource(null, `
+    import React from 'react';
+
+    export class Component extends React.Component {
+      static propTypes = {
+        list: React.PropTypes.oneOf(list)
+      };
+    }
+  `);
+  const args = t.context.args.reduce((akku: any[], args: any[]) => [...akku, ...args], []);
+  const idx = args.indexOf('Line 6:     list: React.PropTypes.oneOf(list)');
+  t.is(idx, 1);
+});
+
+test('In case of error during enum value creation inference the error information should be retained', t => {
+  react2dts.generateFromSource(null, `
+    import React from 'react';
+
+    export class Component extends React.Component {
+      static propTypes = {
+        list: React.PropTypes.oneOf(Object.keys(object))
+      };
+    }
+  `);
+  const args = t.context.args.reduce((akku: any[], args: any[]) => [...akku, ...args], []);
+  const idx = args.indexOf('Line 6:     list: React.PropTypes.oneOf(Object.keys(object))');
+  t.is(idx, 1);
+});
+
+test('In case of error during shape type inference the error information should be retained', t => {
+  react2dts.generateFromSource(null, `
+    import React from 'react';
+
+    export class Component extends React.Component {
+      static propTypes = {
+        shape: React.PropTypes.shape(some.shape)
+      };
+    }
+  `);
+  const args = t.context.args.reduce((akku: any[], args: any[]) => [...akku, ...args], []);
+  const idx = args.indexOf('Line 6:     shape: React.PropTypes.shape(some.shape)');
+  t.is(idx, 1);
+});