From eafc8028314805ded314648e02078cf596760197 Mon Sep 17 00:00:00 2001
From: Corey Hulen <corey@hulen.com>
Date: Wed, 17 Aug 2016 07:17:44 -0800
Subject: PLT-3881 Fix websocket errors after leaving the team (#3815)

---
 api/team.go                          | 2 ++
 webapp/actions/websocket_actions.jsx | 2 ++
 webapp/stores/team_store.jsx         | 4 ++++
 3 files changed, 8 insertions(+)

diff --git a/api/team.go b/api/team.go
index 702ea96d1..20a0671da 100644
--- a/api/team.go
+++ b/api/team.go
@@ -331,6 +331,8 @@ func LeaveTeam(team *model.Team, user *model.User) *model.AppError {
 			if result := <-Srv.Store.Channel().RemoveMember(channel.Id, user.Id); result.Err != nil {
 				return result.Err
 			}
+
+			InvalidateCacheForChannel(channel.Id)
 		}
 	}
 
diff --git a/webapp/actions/websocket_actions.jsx b/webapp/actions/websocket_actions.jsx
index 9150e4111..9548a58dd 100644
--- a/webapp/actions/websocket_actions.jsx
+++ b/webapp/actions/websocket_actions.jsx
@@ -194,6 +194,8 @@ function handleLeaveTeamEvent(msg) {
 
         // if the are on the team begin removed redirect them to the root
         if (TeamStore.getCurrentId() === msg.team_id) {
+            TeamStore.setCurrentId('');
+            Client.setTeamId('');
             browserHistory.push('/');
         }
     } else if (TeamStore.getCurrentId() === msg.team_id) {
diff --git a/webapp/stores/team_store.jsx b/webapp/stores/team_store.jsx
index 90cb66bb2..e59f0ca43 100644
--- a/webapp/stores/team_store.jsx
+++ b/webapp/stores/team_store.jsx
@@ -63,6 +63,10 @@ class TeamStoreClass extends EventEmitter {
         return this.currentTeamId;
     }
 
+    setCurrentId(id) {
+        this.currentTeamId = id;
+    }
+
     getCurrent() {
         const team = this.teams[this.currentTeamId];
 
-- 
cgit v1.2.3-1-g7c22