better handling of selection when removing tests

Author: corevo

packages/selenium-ide/src/neo/components/Suite/index.jsx

@@ -103,7 +103,10 @@ class Suite extends React.Component {
             <ListMenuItem onClick={this.props.remove}>Delete</ListMenuItem>
           </ListMenu>
         </div>
-        <TestList collapsed={!this.store.isOpen} suite={this.props.suite} tests={this.store.filteredTests.get()} removeTest={this.props.suite.removeTestCase} />
+        <TestList collapsed={!this.store.isOpen} suite={this.props.suite} tests={this.store.filteredTests.get()} removeTest={(test) => {
+          this.props.suite.removeTestCase(test);
+          UiState.selectTest();
+        }} />
       </div>
     );
   }

packages/selenium-ide/src/neo/containers/Navigation/index.jsx

@@ -35,7 +35,8 @@ import "./style.css";
   constructor(props) {
     super(props);
     this.state = {
-      showTests: true
+      showTests: true,
+      lastSelection: {}
     };
     this.handleChangedTab = this.handleChangedTab.bind(this);
   }
@@ -45,9 +46,12 @@ import "./style.css";
     moveTest: PropTypes.func.isRequired
   };
   handleChangedTab(tab) {
+    const lastSelection = this.state.lastSelection;
     this.setState({
-      showTests: tab === "Tests"
+      showTests: tab === "Tests",
+      lastSelection: UiState.selectedTest
     });
+    UiState.selectTest(lastSelection.test, lastSelection.suite);
   }
   handleKeyDown(event) {
     const e = event.nativeEvent;

packages/selenium-ide/src/neo/stores/view/ModalState.js

@@ -81,6 +81,7 @@ class ModalState {
     }, (choseDelete) => {
       if (choseDelete) {
         this._project.deleteSuite(suite);
+        UiState.selectTest();
       }
     });
   }
@@ -94,6 +95,7 @@ class ModalState {
     }, (choseDelete) => {
       if (choseDelete) {
         this._project.deleteTestCase(testCase);
+        UiState.selectTest();
       }
     });
   }

packages/selenium-ide/src/neo/stores/view/UiState.js

@@ -15,7 +15,7 @@
 // specific language governing permissions and limitations
 // under the License.
 
-import { action, computed, observable, observe, extendObservable } from "mobx";
+import { action, computed, observable, observe, extendObservable, reaction } from "mobx";
 import storage from "../../IO/storage";
 import SuiteState from "./SuiteState";
 import TestState from "./TestState";