import { Component } from 'react';
import { AppStore } from '../services/';
import { loadAndBundleSpec } from '../utils';

interface SpecProps {
  specUrl?: string;
  spec?: object;
  store?: AppStore;

  children?: any;
}

interface SpecState {
  error?: Error;
  loading: boolean;
  store?: AppStore;
}

export class StoreProvider extends Component<SpecProps, SpecState> {
  store: AppStore;

  constructor(props: SpecProps) {
    super(props);

    this.state = {
      loading: true,
    };
  }

  componentDidMount() {
    this.load();
  }

  async load() {
    let { specUrl, spec } = this.props;

    this.setState({
      loading: true,
    });

    try {
      const resolvedSpec = await loadAndBundleSpec(spec || specUrl!);
      this.setState({
        loading: false,
        store: new AppStore(resolvedSpec, specUrl),
      });
    } catch (e) {
      this.setState({
        error: e,
      });
    }
  }

  setError(e?: Error) {
    this.setState({
      error: e,
    });
  }

  render() {
    if (this.state.error) throw this.state.error;
    return this.props.children(this.state);
  }
}

import { Component } from 'react';
import { AppStore } from '../services/';
import { loadAndBundleSpec } from '../utils';

interface SpecProps {
  specUrl?: string;
  spec?: object;
  store?: AppStore;

  children?: any;
}

interface SpecState {
  error?: Error;
  loading: boolean;
  store?: AppStore;
}

export class StoreProvider extends Component<SpecProps, SpecState> {
  store: AppStore;

  constructor(props: SpecProps) {
    super(props);

    this.state = {
      loading: true,
    };

    this.load();
  }

  async load() {
    let { specUrl, spec } = this.props;

    this.setState({
      loading: true,
    });

    try {
      const resolvedSpec = await loadAndBundleSpec(spec || specUrl!);
      this.setState({
        loading: false,
        store: new AppStore(resolvedSpec, specUrl),
      });
    } catch (e) {
      this.setState({
        error: e,
      });
    }
  }

  setError(e?: Error) {
    this.setState({
      error: e,
    });
  }

  render() {
    if (this.state.error) throw this.state.error;
    return this.props.children(this.state);
  }
}

--- 
+++ 
@@ -25,7 +25,9 @@
     this.state = {
       loading: true,
     };
+  }
 
+  componentDidMount() {
     this.load();
   }
 
Fix setState run before mount
