Tests for ProgramBlockEditor change loop iterations

Author: sbates-idrc

src/ProgramBlockEditor.test.js

@@ -145,6 +145,11 @@ function getProgramBlockLoopLabel(programBlockEditorWrapper, index: number) {
         .find('.command-block-loop-label-container').getDOMNode().textContent;
 }
 
+function getProgramBlockLoopIterationsInput(programBlockEditorWrapper, index: number) {
+    return getProgramBlocks(programBlockEditorWrapper).at(index)
+        .find('.command-block-loop-iterations');
+}
+
 function getProgramBlockLoopIterations(programBlockEditorWrapper, index: number) {
     return ((getProgramBlocks(programBlockEditorWrapper).at(index)
         .find('.command-block-loop-iterations')
@@ -473,6 +478,87 @@ describe('Active loop container highlight', () => {
     });
 });
 
+describe('Change loop iterations', () => {
+    test('The ProgramSequence should be updated when the loop iterations are changed when the program is stopped', () => {
+        const { wrapper, mockChangeProgramSequenceHandler } = createMountProgramBlockEditor({
+            runningState: 'stopped',
+            programSequence: new ProgramSequence(
+                [
+                    {block: 'startLoop', label: 'A', iterations: 2},
+                    {block: 'endLoop', label: 'A'}
+                ],
+                0,
+                0,
+                new Map([['A', 2]])
+            )
+        });
+
+        const input = getProgramBlockLoopIterationsInput(wrapper, 0);
+        ((input.getDOMNode(): any): HTMLInputElement).value = '4';
+        input.simulate('change');
+        input.getDOMNode().dispatchEvent(new KeyboardEvent('keydown', {key: 'Enter'}));
+
+        expect(mockChangeProgramSequenceHandler.mock.calls.length).toBe(1);
+        expect(mockChangeProgramSequenceHandler.mock.calls[0][0]).toStrictEqual(
+            new ProgramSequence(
+                [
+                    {
+                        block: 'startLoop',
+                        label: 'A',
+                        iterations: 4
+                    },
+                    {
+                        block: 'endLoop',
+                        label: 'A'
+                    }
+                ],
+                0,
+                0,
+                new Map([['A', 2]])
+            )
+        );
+    });
+    test('The ProgramSequence and loopIterationsLeft should be updated when the loop iterations are changed when the program is paused', () => {
+        const { wrapper, mockChangeProgramSequenceHandler } = createMountProgramBlockEditor({
+            runningState: 'paused',
+            programSequence: new ProgramSequence(
+                [
+                    {block: 'startLoop', label: 'A', iterations: 2},
+                    {block: 'endLoop', label: 'A'}
+                ],
+                0,
+                0,
+                new Map([['A', 1]])
+            )
+        });
+
+        const input = getProgramBlockLoopIterationsInput(wrapper, 0);
+        ((input.getDOMNode(): any): HTMLInputElement).value = '4';
+        input.simulate('change');
+        input.getDOMNode().dispatchEvent(new KeyboardEvent('keydown', {key: 'Enter'}));
+
+        expect(mockChangeProgramSequenceHandler.mock.calls.length).toBe(1);
+        expect(mockChangeProgramSequenceHandler.mock.calls[0][0]).toStrictEqual(
+            new ProgramSequence(
+                [
+                    {
+                        block: 'startLoop',
+                        label: 'A',
+                        iterations: 4
+                    },
+                    {
+                        block: 'endLoop',
+                        label: 'A'
+                    }
+                ],
+                0,
+                0,
+                new Map([['A', 4]])
+            )
+        );
+    });
+});
+
 describe('The expand add node toggle switch should be configurable via properties', () => {
     describe('Given that addNodeExpandedMode is false', () => {
         test('Then the toggle switch should be off, and the change handler should be wired up', () => {