COMPASS-2965: Proper state reset on disconnect

Author: durran

package-lock.json

@@ -226,7 +226,7 @@
     "url": "git://github.com/10gen/compass.git"
   },
   "dependencies": {
-    "@mongodb-js/compass-aggregations": "^1.6.2",
+    "@mongodb-js/compass-aggregations": "^1.6.3",
     "@mongodb-js/compass-auto-updates": "^0.0.2",
     "@mongodb-js/compass-auth-kerberos": "^1.0.1",
     "@mongodb-js/compass-auth-ldap": "^1.0.1",

package.json

@@ -20,6 +20,7 @@ const CollectionStore = Reflux.createStore({
    */
   onActivated(appRegistry) {
     this.appRegistry = appRegistry;
+    appRegistry.on('data-service-disconnected', this.onDisconnected.bind(this));
     appRegistry.on('show-agg-pipeline-out-results', (ns) => {
       this.setCollection({
         _id: ns,
@@ -31,6 +32,11 @@ const CollectionStore = Reflux.createStore({
     });
   },
 
+  onDisconnected() {
+    this.collection = {};
+    this.activeTabIndex = 0;
+  },
+
   /**
    * Set the collection information.
    *

src/internal-plugins/app/lib/stores/collection-store.js

@@ -36,6 +36,11 @@ const FieldStore = Reflux.createStore({
 
     appRegistry.on('collection-changed', this.onCollectionChanged.bind(this));
     appRegistry.on('database-changed', this.onDatabaseChanged.bind(this));
+    appRegistry.on('data-service-disconnected', this.onDisconnected.bind(this));
+  },
+
+  onDisconnected() {
+    this.setState(this.getInitialState());
   },
 
   /**

src/internal-plugins/app/lib/stores/field-store.js

@@ -35,6 +35,7 @@ const InstanceStore = Reflux.createStore({
   onDisconnected() {
     const MongoDBInstance = require('../../../../app/models/mongodb-instance');
     app.state.instance = new MongoDBInstance();
+    this.setState(this.getInitialState());
   },
 
   /**

src/internal-plugins/app/lib/stores/instance-store.js

@@ -20,6 +20,14 @@ const NamespaceStore = Reflux.createStore({
     this._ns = DEFAULT_NAMESPACE;
   },
 
+  onActivated(appRegistry) {
+    appRegistry.on('data-service-disconnected', this.onDisconnected.bind(this));
+  },
+
+  onDisconnected() {
+    this.init();
+  },
+
   /**
    * Gets the current namespace being worked with in the application.
    */

src/internal-plugins/app/lib/stores/namespace-store.js

@@ -40,16 +40,7 @@ const HomeStore = Reflux.createStore({
   },
 
   onDisconnected() {
-    this.setState({
-      errorMessage: '',
-      namespace: '',
-      isConnected: false,
-      isAtlas: false,
-      authentication: 'NONE',
-      ssl: 'NONE',
-      sshTunnel: 'NONE',
-      uiStatus: UI_STATES.INITIAL
-    });
+    this.setState(this.getInitialState());
   },
 
   onConnected(appRegistry, err, ds) {