fix(lab): clear references to a task key after deleting a task

app/pages/lab-fem/workflow.jsx

@@ -20,6 +20,7 @@ import TaskOptions from '../lab/workflow-components/task-options.jsx';
 import TaskEditor from '../lab/workflow-components/task-editor.jsx';
 import TaskPicker from '../lab/workflow-components/task-picker.jsx';
 import { isThisProjectUsingFEMLab, isWorkflowUsingJSONSubjects, FEM_LAB_PREVIEW_HOST } from './fem-lab-utilities.js';
+import removeTaskKeyFromWorkflow from '../lab/helpers/removeTaskKeyFromWorkflow.js';
 
 const DEMO_SUBJECT_SET_ID = process.env.NODE_ENV === 'production'
 ? '6' // Cats
@@ -876,8 +877,9 @@ class EditWorkflowPage extends Component {
         changes[`tasks.${taskKey}`] = undefined;
         this.props.workflow.update(changes);
       }
+      removeTaskKeyFromWorkflow(this.props.workflow, taskKey);
 
-      return this.updateFirstTask();
+      return this.updateFirstTask().save();
     }
   }
 };

app/pages/lab/helpers/removeTaskKeyFromWorkflow.js

@@ -0,0 +1,14 @@
+export default function removeTaskKeyFromWorkflow(workflow, taskKey) {
+    const changes = {};
+    Object.entries(workflow.tasks).forEach(([key, task]) => {
+      if (task.next === taskKey) {
+        changes[`tasks.${key}.next`] = '';
+      }
+      task.answers?.forEach((answer, index) => {
+        if (answer.next === taskKey) {
+          changes[`tasks.${key}.answers.${index}.next`] = '';
+        }
+      });
+    });
+    return workflow.update(changes);
+}

app/pages/lab/helpers/removeTaskKeyFromWorkflow.spec.js

@@ -0,0 +1,43 @@
+import { expect } from 'chai';
+import sinon from 'sinon';
+import removeTaskKeyFromWorkflow from './removeTaskKeyFromWorkflow.js';
+
+describe('removeTaskKeyFromWorkflow', function () {
+  let workflow;
+
+  beforeEach(function () {
+    workflow = {
+      tasks: {
+        'T1': {
+          type: 'single',
+          answers: [
+            { label: 'Yes', next: 'T3' },
+            { label: 'No', next: 'T2' }
+          ]
+        },
+        'T2': {
+          type: 'multiple',
+          next: 'T3',
+          answers: [
+            { label: 'Blue' },
+            { label: 'Green' }
+          ]
+        },
+        'T3': {
+          type: 'text',
+          next: ''
+        }
+      },
+      update: sinon.stub(),
+      save: sinon.stub()
+    };
+  });
+
+  it('should delete the selected key from each task', function () {
+    removeTaskKeyFromWorkflow(workflow, 'T3');
+    expect(workflow.update).to.have.been.calledWith({
+      'tasks.T1.answers.0.next': '',
+      'tasks.T2.next': ''
+    });
+  });
+});

app/pages/lab/workflow.jsx

@@ -1,16 +1,13 @@
 import { Component, useEffect, useState } from 'react';
 import PropTypes from 'prop-types';
 import handleInputChange from '../../lib/handle-input-change.js';
-import TriggeredModalForm from 'modal-form/triggered';
 import ModalFormDialog from 'modal-form/dialog';
 import apiClient from 'panoptes-client/lib/api-client';
-import ChangeListener from '../../components/change-listener.cjsx';
 import RetirementRulesEditor from '../../components/retirement-rules-editor.cjsx';
 import {Link} from 'react-router';
 import MultiImageSubjectOptionsEditor from '../../components/multi-image-subject-options-editor.jsx';
 import taskComponents from '../../classifier/tasks/index.js';
 import AutoSave from '../../components/auto-save.coffee';
-import FileButton from '../../components/file-button.cjsx';
 import WorkflowCreateForm from './workflow-create-form.cjsx';
 import workflowActions from './actions/workflow.js';
 import classnames from 'classnames';
@@ -21,6 +18,7 @@ import Tutorials from './workflow-components/tutorials.jsx';
 import TaskOptions from './workflow-components/task-options.jsx';
 import TaskPicker from './workflow-components/task-picker.jsx';
 import TaskEditor from './workflow-components/task-editor.jsx';
+import removeTaskKeyFromWorkflow from './helpers/removeTaskKeyFromWorkflow.js';
 
 const DEMO_SUBJECT_SET_ID = process.env.NODE_ENV === 'production'
 ? '6' // Cats
@@ -807,8 +805,9 @@ class EditWorkflowPage extends Component {
         changes[`tasks.${taskKey}`] = undefined;
         this.props.workflow.update(changes);
       }
+      removeTaskKeyFromWorkflow(this.props.workflow, taskKey);
 
-      return this.updateFirstTask();
+      return this.updateFirstTask().save();
     }
   }
 }