Wrap <Predict/> and add ids needed for unit tests

maddiedierker · maddiedierker · commit 8dfb6028b745 · 2019-12-16T11:17:59.000-08:00
diff --git a/src/oceans/ui.jsx b/src/oceans/ui.jsx
@@ -968,7 +968,7 @@ const MediaControl = Object.freeze({
   FastForward: 'fast-forward'
 });
 
-let Predict = class Predict extends React.Component {
+let UnwrappedPredict = class Predict extends React.Component {
   state = {
     displayControls: false,
     timeScale: defaultTimeScale
@@ -1040,14 +1040,12 @@ let Predict = class Predict extends React.Component {
       this.state.timeScale !== defaultTimeScale
     ) {
       selectedControl = MediaControl.FastForward;
-    } else {
-      selectedControl = MediaControl.Play;
     }
 
     return (
       <Body>
         {this.state.displayControls && (
-          <div style={styles.mediaControls}>
+          <div style={styles.mediaControls} id="uitest-media-ctrl">
             <span
               onClick={() => this.onScaleTime(true)}
               style={[
@@ -1090,21 +1088,29 @@ let Predict = class Predict extends React.Component {
           </div>
         )}
         {!state.isRunning && !state.isPaused && (
-          <Button style={styles.continueButton} onClick={this.onRun}>
+          <Button
+            style={styles.continueButton}
+            onClick={this.onRun}
+            id="uitest-run-btn"
+          >
             <FontAwesomeIcon icon={faPlay} />
             &nbsp; &nbsp; Run
           </Button>
         )}
         {(state.isRunning || state.isPaused) && state.canSkipPredict && (
-          <Button style={styles.continueButton} onClick={this.onContinue}>
+          <Button
+            style={styles.continueButton}
+            onClick={this.onContinue}
+            id="uitest-continue-btn"
+          >
             Continue
           </Button>
         )}
       </Body>
     );
   }
 };
-Predict = Radium(Predict);
+export const Predict = Radium(UnwrappedPredict); // Exported for unit tests.
 
 class PondPanel extends React.Component {
   onPondPanelClick(e) {
