2 files changed, 19 insertions, 9 deletions

diff --git a/web/react/stores/socket_store.jsx b/web/react/stores/socket_store.jsx
index 9c3270f68..bc2bdbe64 100644
--- a/web/react/stores/socket_store.jsx
+++ b/web/react/stores/socket_store.jsx
@@ -28,10 +28,13 @@ class SocketStoreClass extends EventEmitter {
         this.addChangeListener = this.addChangeListener.bind(this);
         this.removeChangeListener = this.removeChangeListener.bind(this);
         this.sendMessage = this.sendMessage.bind(this);
+        this.close = this.close.bind(this);
+
         this.failCount = 0;
 
         this.initialize();
     }
+
     initialize() {
         if (!UserStore.getCurrentId()) {
             return;
@@ -106,15 +109,19 @@ class SocketStoreClass extends EventEmitter {
             };
         }
     }
+
     emitChange(msg) {
         this.emit(CHANGE_EVENT, msg);
     }
+
     addChangeListener(callback) {
         this.on(CHANGE_EVENT, callback);
     }
+
     removeChangeListener(callback) {
         this.removeListener(CHANGE_EVENT, callback);
     }
+
     handleMessage(msg) {
         switch (msg.action) {
         case SocketEvents.POSTED:
@@ -153,6 +160,7 @@ class SocketStoreClass extends EventEmitter {
         default:
         }
     }
+
     sendMessage(msg) {
         if (conn && conn.readyState === WebSocket.OPEN) {
             conn.send(JSON.stringify(msg));
@@ -161,9 +169,16 @@ class SocketStoreClass extends EventEmitter {
             this.initialize();
         }
     }
+
     setTranslations(messages) {
         this.translations = messages;
     }
+
+    close() {
+        if (conn && conn.readyState === WebSocket.OPEN) {
+            conn.close();
+        }
+    }
 }
 
 function handleNewPostEvent(msg, translations) {
@@ -305,12 +320,5 @@ function handlePreferenceChangedEvent(msg) {
 
 var SocketStore = new SocketStoreClass();
 
-/*SocketStore.dispatchToken = AppDispatcher.register((payload) => {
-    var action = payload.action;
-
-    switch (action.type) {
-    default:
-    }
-    });*/
-
 export default SocketStore;
+window.SocketStore = SocketStore;
diff --git a/web/react/stores/suggestion_store.jsx b/web/react/stores/suggestion_store.jsx
index 9cd566c22..dd5c107e0 100644
--- a/web/react/stores/suggestion_store.jsx
+++ b/web/react/stores/suggestion_store.jsx
@@ -223,7 +223,9 @@ class SuggestionStore extends EventEmitter {
             this.emitSuggestionsChanged(id);
             break;
         case ActionTypes.SUGGESTION_RECEIVED_SUGGESTIONS:
-            if (other.matchedPretext === this.getMatchedPretext(id)) {
+            if (this.getMatchedPretext(id) === '') {
+                this.setMatchedPretext(id, other.matchedPretext);
+
                 // ensure the matched pretext hasn't changed so that we don't receive suggestions for outdated pretext
                 this.addSuggestions(id, other.terms, other.items, other.component);