cluster fixes, would not update list on new cluster creation

Kitware · Jun 20, 2016 · 6aebfe5 · 6aebfe5
1 parent a8197f0
commit 6aebfe5
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 9 deletions.
diff --git a/src/pages/Preferences/Cluster/index.js b/src/pages/Preferences/Cluster/index.js
@@ -93,7 +93,7 @@ const ClusterPrefs = React.createClass({
 
   clusterHasSimulation(id) {
     for (let i = 0; i < this.props.taskflows.length; i++) {
-      if (this.props.taskflows[i].flow && this.props.taskflows[i].flow.meta.cluster._id === id) {
+      if (this.props.taskflows[i].flow && get(this.props.taskflows[i].flow, 'meta.cluster._id') === id) {
         return this.props.taskflows[i].simulation;
       }
     }

diff --git a/src/redux/actions/taskflows.js b/src/redux/actions/taskflows.js
@@ -351,7 +351,12 @@ client.onEvent((resp) => {
         break;
       }
       case 'cluster': {
-        dispatch(clusterActions.updateClusterStatus(id, status));
+        if (status === 'created') {
+          // we need to fetch some new cluster props when this happens
+          dispatch(clusterActions.fetchCluster(id));
+        } else {
+          dispatch(clusterActions.updateClusterStatus(id, status));
+        }
         break;
       }
       case 'profile': {

diff --git a/src/redux/reducers/clusters.js b/src/redux/reducers/clusters.js
@@ -159,11 +159,20 @@ export default function clustersReducer(state = initialState, action) {
     }
 
     case Actions.UPDATE_CLUSTER_STATUS: {
+      const list = [].concat(state.list);
       const mapById = Object.assign({}, state.mapById);
       const cluster = Object.assign({}, state.mapById[action.id]);
       cluster.status = action.status;
       mapById[action.id] = cluster;
-      return Object.assign({}, state, { mapById });
+      if (cluster.type === 'trad') {
+        for (let i = 0; i < list.length; i++) {
+          if (list[i]._id === action.id) {
+            list[i].status = action.status;
+            break;
+          }
+        }
+      }
+      return Object.assign({}, state, { list, mapById });
     }
 
     case Actions.CLUSTER_APPLY_PRESET: {

diff --git a/test/redux/clusters.js b/test/redux/clusters.js
@@ -22,7 +22,10 @@ function setSpy(target, method, data) {
 Object.freeze(initialState);
 
 describe('cluster actions', () => {
-  const cluster = { _id: 'a1', name: 'myCluster',
+  const cluster = { _id: 'a1',
+    type: 'trad',
+    name: 'myCluster',
+    status: 'unknown',
     log: [{ entry: 'created...' }, { entry: 'running...' }],
   };
   describe('simple actions', () => {
@@ -159,11 +162,19 @@ describe('cluster actions', () => {
 
     // we only test the reducer here
     it('should update a cluster\'s status', () => {
-      const thisState = Object.assign({}, initialState);
-      thisState.mapById = { a1: { status: 'created' } };
-      const action = { type: Actions.UPDATE_CLUSTER_STATUS, id: 'a1', status: 'terminated' };
-      expect(clustersReducer(thisState, action).mapById.a1.status)
-        .toEqual('terminated');
+      const newStatus = 'terminated';
+      const myCluster = deepClone(cluster);
+      const givenState = deepClone(initialState);
+      givenState.mapById[myCluster._id] = myCluster;
+      givenState.list.push(myCluster);
+
+      const expectedState = deepClone(givenState);
+      expectedState.mapById[myCluster._id].status = newStatus;
+      expectedState.list[0].status = newStatus;
+
+      const action = { type: Actions.UPDATE_CLUSTER_STATUS, id: cluster._id, status: newStatus };
+      expect(clustersReducer(givenState, action))
+        .toEqual(expectedState);
     });
   });