/**
 * @license
 * Copyright 2019 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

suite('ASTNode', function() {
  setup(function() {
    Blockly.defineBlocksWithJsonArray([{
      "type": "input_statement",
      "message0": "%1 %2 %3 %4",
      "args0": [
        {
          "type": "field_input",
          "name": "NAME",
          "text": "default"
        },
        {
          "type": "field_input",
          "name": "NAME",
          "text": "default"
        },
        {
          "type": "input_value",
          "name": "NAME"
        },
        {
          "type": "input_statement",
          "name": "NAME"
        }
      ],
      "previousStatement": null,
      "nextStatement": null,
      "colour": 230,
      "tooltip": "",
      "helpUrl": ""
    },
    {
      "type": "value_input",
      "message0": "%1",
      "args0": [
        {
          "type": "input_value",
          "name": "NAME"
        }
      ],
      "colour": 230,
      "tooltip": "",
      "helpUrl": ""
    },
    {
      "type": "field_input",
      "message0": "%1",
      "args0": [
        {
          "type": "field_input",
          "name": "NAME",
          "text": "default"
        }
      ],
      "output": null,
      "colour": 230,
      "tooltip": "",
      "helpUrl": ""
    }
    ]);
    this.workspace = new Blockly.Workspace();
    this.cursor = this.workspace.cursor;
    var statementInput1 = this.workspace.newBlock('input_statement');
    var statementInput2 = this.workspace.newBlock('input_statement');
    var statementInput3 = this.workspace.newBlock('input_statement');
    var statementInput4 = this.workspace.newBlock('input_statement');
    var fieldWithOutput = this.workspace.newBlock('field_input');
    var valueInput = this.workspace.newBlock('value_input');

    statementInput1.nextConnection.connect(statementInput2.previousConnection);
    statementInput1.inputList[0].connection
        .connect(fieldWithOutput.outputConnection);
    statementInput2.inputList[1].connection
        .connect(statementInput3.previousConnection);

    this.blocks = {
      statementInput1: statementInput1,
      statementInput2: statementInput2,
      statementInput3: statementInput3,
      statementInput4: statementInput4,
      fieldWithOutput: fieldWithOutput,
      valueInput: valueInput
    };
    sinon.stub(Blockly, "getMainWorkspace").returns(new Blockly.Workspace());
  });
  teardown(function() {
    delete Blockly.Blocks['input_statement'];
    delete Blockly.Blocks['field_input'];
    delete Blockly.Blocks['value_input'];

    this.workspace.dispose();
    sinon.restore();
  });

  suite('HelperFunctions', function() {
    test('findNextForInput_', function() {
      var input = this.blocks.statementInput1.inputList[0];
      var input2 = this.blocks.statementInput1.inputList[1];
      var connection = input.connection;
      var node = Blockly.ASTNode.createConnectionNode(connection);
      var newASTNode = node.findNextForInput_(input);
      assertEquals(newASTNode.getLocation(), input2.connection);
    });

    test('findPrevForInput_', function() {
      var input = this.blocks.statementInput1.inputList[0];
      var input2 = this.blocks.statementInput1.inputList[1];
      var connection = input2.connection;
      var node = Blockly.ASTNode.createConnectionNode(connection);
      var newASTNode = node.findPrevForInput_(input2);
      assertEquals(newASTNode.getLocation(), input.connection);
    });

    test('findNextForField_', function() {
      var field = this.blocks.statementInput1.inputList[0].fieldRow[0];
      var field2 = this.blocks.statementInput1.inputList[0].fieldRow[1];
      var node = Blockly.ASTNode.createFieldNode(field);
      var newASTNode = node.findNextForField_(field);
      assertEquals(newASTNode.getLocation(), field2);
    });

    test('findPrevForField_', function() {
      var field = this.blocks.statementInput1.inputList[0].fieldRow[0];
      var field2 = this.blocks.statementInput1.inputList[0].fieldRow[1];
      var node = Blockly.ASTNode.createFieldNode(field2);
      var newASTNode = node.findPrevForField_(field2);
      assertEquals(newASTNode.getLocation(), field);
    });

    test('navigateBetweenStacks_Forward', function() {
      var node = new Blockly.ASTNode(
          Blockly.ASTNode.types.NEXT, this.blocks.statementInput1.nextConnection);
      var newASTNode = node.navigateBetweenStacks_(true);
      assertEquals(newASTNode.getLocation(), this.blocks.statementInput4);
    });

    test('navigateBetweenStacks_Backward', function() {
      var node = new Blockly.ASTNode(
          Blockly.ASTNode.types.BLOCK, this.blocks.statementInput4);
      var newASTNode = node.navigateBetweenStacks_(false);
      assertEquals(newASTNode.getLocation(), this.blocks.statementInput1);
    });
    test('getOutAstNodeForBlock_', function() {
      var node = new Blockly.ASTNode(
          Blockly.ASTNode.types.BLOCK, this.blocks.statementInput2);
      var newASTNode = node.getOutAstNodeForBlock_(this.blocks.statementInput2);
      assertEquals(newASTNode.getLocation(), this.blocks.statementInput1);
    });
    test('getOutAstNodeForBlock_OneBlock', function() {
      var node = new Blockly.ASTNode(
          Blockly.ASTNode.types.BLOCK, this.blocks.statementInput4);
      var newASTNode = node.getOutAstNodeForBlock_(this.blocks.statementInput4);
      assertEquals(newASTNode.getLocation(), this.blocks.statementInput4);
    });
    test('findFirstFieldOrInput_', function() {
      var node = new Blockly.ASTNode(
          Blockly.ASTNode.types.BLOCK, this.blocks.statementInput4);
      var field = this.blocks.statementInput4.inputList[0].fieldRow[0];
      var newASTNode = node.findFirstFieldOrInput_(this.blocks.statementInput4);
      assertEquals(newASTNode.getLocation(), field);
    });

  });

  suite('NavigationFunctions', function() {
    suite('Next', function() {
      setup(function() {
        Blockly.defineBlocksWithJsonArray([{
          "type": "top_connection",
          "message0": "",
          "previousStatement": null,
          "colour": 230,
          "tooltip": "",
          "helpUrl": ""
        },
        {
          "type": "fields_and_input",
          "message0": "%1 hi %2 %3 %4",
          "args0": [
            {
              "type": "field_input",
              "name": "NAME",
              "text": "default"
            },
            {
              "type": "input_dummy"
            },
            {
              "type": "field_input",
              "name": "NAME",
              "text": "default"
            },
            {
              "type": "input_value",
              "name": "NAME"
            }
          ],
          "previousStatement": null,
          "nextStatement": null,
          "colour": 230,
          "tooltip": "",
          "helpUrl": ""
        },
        {
          "type": "two_fields",
          "message0": "%1 hi",
          "args0": [
            {
              "type": "field_input",
              "name": "NAME",
              "text": "default"
            }
          ],
          "colour": 230,
          "tooltip": "",
          "helpUrl": ""
        },
        {
          "type": "fields_and_input2",
          "message0": "%1 %2 %3 hi %4 bye",
          "args0": [
            {
              "type": "input_value",
              "name": "NAME"
            },
            {
              "type": "field_input",
              "name": "NAME",
              "text": "default"
            },
            {
              "type": "input_value",
              "name": "NAME"
            },
            {
              "type": "input_statement",
              "name": "NAME"
            }
          ],
          "colour": 230,
          "tooltip": "",
          "helpUrl": ""
        }]);

        this.singleBlockWorkspace = new Blockly.Workspace();

        var noNextConnection = this.workspace.newBlock('top_connection');
        var fieldAndInputs = this.workspace.newBlock('fields_and_input');
        var twoFields = this.workspace.newBlock('two_fields');
        var fieldAndInputs2 = this.workspace.newBlock('fields_and_input2');
        var singleBlock = this.singleBlockWorkspace.newBlock('two_fields');

        this.blocks.noNextConnection = noNextConnection;
        this.blocks.fieldAndInputs = fieldAndInputs;
        this.blocks.twoFields = twoFields;
        this.blocks.fieldAndInputs2 = fieldAndInputs2;
        this.blocks.singleBlock = singleBlock;


      });
      teardown(function() {
        delete this.blocks.noNextConnection;
        delete this.blocks.fieldAndInputs;
        delete this.blocks.twoFields;
        delete this.blocks.fieldAndInputs2;
        delete this.blocks.singleBlock;

        delete Blockly.Blocks['fields_and_input2'];
        delete Blockly.Blocks['two_fields'];
        delete Blockly.Blocks['fields_and_input'];
        delete Blockly.Blocks['top_connection'];
      });

      test('fromPreviousToBlock', function() {
        var prevConnection = this.blocks.statementInput1.previousConnection;
        var node = Blockly.ASTNode.createConnectionNode(prevConnection);
        var nextNode = node.next();
        assertEquals(nextNode.getLocation(), this.blocks.statementInput1);
      });
      test('fromBlockToNext', function() {
        var nextConnection = this.blocks.statementInput1.nextConnection;
        var node = Blockly.ASTNode.createBlockNode(this.blocks.statementInput1);
        var nextNode = node.next();
        assertEquals(nextNode.getLocation(), nextConnection);
      });
      test('fromBlockToNull', function() {
        var node = Blockly.ASTNode.createBlockNode(this.blocks.noNextConnection);
        var nextNode = node.next();
        assertEquals(nextNode, null);
      });
      test('fromNextToPrevious', function() {
        var nextConnection = this.blocks.statementInput1.nextConnection;
        var prevConnection = this.blocks.statementInput2.previousConnection;
        var node = Blockly.ASTNode.createConnectionNode(nextConnection);
        var nextNode = node.next();
        assertEquals(nextNode.getLocation(), prevConnection);
      });
      test('fromNextToNull', function() {
        var nextConnection = this.blocks.statementInput2.nextConnection;
        var node = Blockly.ASTNode.createConnectionNode(nextConnection);
        var nextNode = node.next();
        assertEquals(nextNode, null);
      });
      test('fromInputToInput', function() {
        var input = this.blocks.statementInput1.inputList[0];
        var inputConnection = this.blocks.statementInput1.inputList[1].connection;
        var node = Blockly.ASTNode.createInputNode(input);
        var nextNode = node.next();
        assertEquals(nextNode.getLocation(), inputConnection);
      });
      test('fromInputToStatementInput', function() {
        var input = this.blocks.fieldAndInputs2.inputList[1];
        var inputConnection = this.blocks.fieldAndInputs2.inputList[2].connection;
        var node = Blockly.ASTNode.createInputNode(input);
        var nextNode = node.next();
        assertEquals(nextNode.getLocation(), inputConnection);
      });
      test('fromInputToField', function() {
        var input = this.blocks.fieldAndInputs2.inputList[0];
        var field = this.blocks.fieldAndInputs2.inputList[1].fieldRow[0];
        var node = Blockly.ASTNode.createInputNode(input);
        var nextNode = node.next();
        assertEquals(nextNode.getLocation(), field);
      });
      test('fromInputToNull', function() {
        var input = this.blocks.fieldAndInputs2.inputList[2];
        var node = Blockly.ASTNode.createInputNode(input);
        var nextNode = node.next();
        assertEquals(nextNode, null);
      });
      test('fromOutputToBlock', function() {
        var output = this.blocks.fieldWithOutput.outputConnection;
        var node = Blockly.ASTNode.createConnectionNode(output);
        var nextNode = node.next();
        assertEquals(nextNode.getLocation(), this.blocks.fieldWithOutput);
      });
      test('fromFieldToInput', function() {
        var field = this.blocks.statementInput1.inputList[0].fieldRow[1];
        var inputConnection = this.blocks.statementInput1.inputList[0].connection;
        var node = Blockly.ASTNode.createFieldNode(field);
        var nextNode = node.next();
        assertEquals(nextNode.getLocation(), inputConnection);
      });
      test('fromFieldToField', function() {
        var field = this.blocks.fieldAndInputs.inputList[0].fieldRow[0];
        var node = Blockly.ASTNode.createFieldNode(field);
        var field2 = this.blocks.fieldAndInputs.inputList[1].fieldRow[0];
        var nextNode = node.next();
        assertEquals(nextNode.getLocation(), field2);
      });
      test('fromFieldToNull', function() {
        var field = this.blocks.twoFields.inputList[0].fieldRow[0];
        var node = Blockly.ASTNode.createFieldNode(field);
        var nextNode = node.next();
        assertEquals(nextNode, null);
      });
      test('fromStackToStack', function() {
        var node = Blockly.ASTNode.createStackNode(this.blocks.statementInput1);
        var nextNode = node.next();
        assertEquals(nextNode.getLocation(), this.blocks.statementInput4);
        assertEquals(nextNode.getType(), Blockly.ASTNode.types.STACK);
      });
      test('fromStackToNull', function() {
        var node = Blockly.ASTNode.createStackNode(this.blocks.singleBlock);
        var nextNode = node.next();
        assertEquals(nextNode, null);
      });
    });

    suite('Previous', function() {
      setup(function() {
        Blockly.defineBlocksWithJsonArray([{
          "type": "start_block",
          "message0": "",
          "nextStatement": null,
          "colour": 230,
          "tooltip": "",
          "helpUrl": ""
        },
        {
          "type": "fields_and_input",
          "message0": "%1 hi %2 %3 %4",
          "args0": [
            {
              "type": "field_input",
              "name": "NAME",
              "text": "default"
            },
            {
              "type": "input_dummy"
            },
            {
              "type": "field_input",
              "name": "NAME",
              "text": "default"
            },
            {
              "type": "input_value",
              "name": "NAME"
            }
          ],
          "previousStatement": null,
          "nextStatement": null,
          "colour": 230,
          "tooltip": "",
          "helpUrl": ""
        },
        {
          "type": "fields_and_input2",
          "message0": "%1 %2 %3 hi %4 bye",
          "args0": [
            {
              "type": "input_value",
              "name": "NAME"
            },
            {
              "type": "field_input",
              "name": "NAME",
              "text": "default"
            },
            {
              "type": "input_value",
              "name": "NAME"
            },
            {
              "type": "input_statement",
              "name": "NAME"
            }
          ],
          "colour": 230,
          "tooltip": "",
          "helpUrl": ""
        }]);

        var noPrevConnection = this.workspace.newBlock('start_block');
        var fieldsAndInputs = this.workspace.newBlock('fields_and_input');
        var fieldsAndInputs2 = this.workspace.newBlock('fields_and_input2');

        this.blocks.noPrevConnection = noPrevConnection;
        this.blocks.fieldsAndInputs = fieldsAndInputs;
        this.blocks.fieldsAndInputs2 = fieldsAndInputs2;
      });
      teardown(function() {
        delete this.blocks.noPrevConnection;
        delete this.blocks.fieldAndInputs;
        delete this.blocks.fieldsAndInputs2;

        delete Blockly.Blocks['fields_and_input2'];
        delete Blockly.Blocks['start_block'];
        delete Blockly.Blocks['fields_and_input'];
      });

      test('fromPreviousToNull', function() {
        var prevConnection = this.blocks.statementInput1.previousConnection;
        var node = Blockly.ASTNode.createConnectionNode(prevConnection);
        var prevNode = node.prev();
        assertEquals(prevNode, null);
      });
      test('fromPreviousToNext', function() {
        var prevConnection = this.blocks.statementInput2.previousConnection;
        var node = Blockly.ASTNode.createConnectionNode(prevConnection);
        var prevNode = node.prev();
        var nextConnection = this.blocks.statementInput1.nextConnection;
        assertEquals(prevNode.getLocation(), nextConnection);
      });
      test('fromPreviousToInput', function() {
        var prevConnection = this.blocks.statementInput3.previousConnection;
        var node = Blockly.ASTNode.createConnectionNode(prevConnection);
        var prevNode = node.prev();
        assertEquals(prevNode, null);
      });
      test('fromBlockToPrevious', function() {
        var node = Blockly.ASTNode.createBlockNode(this.blocks.statementInput1);
        var prevNode = node.prev();
        var prevConnection = this.blocks.statementInput1.previousConnection;
        assertEquals(prevNode.getLocation(), prevConnection);
      });
      test('fromBlockToNull', function() {
        var node = Blockly.ASTNode.createBlockNode(this.blocks.noPrevConnection);
        var prevNode = node.prev();
        assertEquals(prevNode, null);
      });
      test('fromBlockToOutput', function() {
        var node = Blockly.ASTNode.createBlockNode(this.blocks.fieldWithOutput);
        var prevNode = node.prev();
        var outputConnection = this.blocks.fieldWithOutput.outputConnection;
        assertEquals(prevNode.getLocation(), outputConnection);
      });
      test('fromNextToBlock', function() {
        var nextConnection = this.blocks.statementInput1.nextConnection;
        var node = Blockly.ASTNode.createConnectionNode(nextConnection);
        var prevNode = node.prev();
        assertEquals(prevNode.getLocation(), this.blocks.statementInput1);
      });
      test('fromInputToField', function() {
        var input = this.blocks.statementInput1.inputList[0];
        var node = Blockly.ASTNode.createInputNode(input);
        var prevNode = node.prev();
        assertEquals(prevNode.getLocation(), input.fieldRow[1]);
      });
      test('fromInputToNull', function() {
        var input = this.blocks.fieldsAndInputs2.inputList[0];
        var node = Blockly.ASTNode.createInputNode(input);
        var prevNode = node.prev();
        assertEquals(prevNode, null);
      });
      test('fromInputToInput', function() {
        var input = this.blocks.fieldsAndInputs2.inputList[2];
        var inputConnection = this.blocks.fieldsAndInputs2.inputList[1].connection;
        var node = Blockly.ASTNode.createInputNode(input);
        var prevNode = node.prev();
        assertEquals(prevNode.getLocation(), inputConnection);
      });
      test('fromOutputToNull', function() {
        var output = this.blocks.fieldWithOutput.outputConnection;
        var node = Blockly.ASTNode.createConnectionNode(output);
        var prevNode = node.prev();
        assertEquals(prevNode, null);
      });
      test('fromFieldToNull', function() {
        var field = this.blocks.statementInput1.inputList[0].fieldRow[0];
        var node = Blockly.ASTNode.createFieldNode(field);
        var prevNode = node.prev();
        assertEquals(prevNode, null);
      });
      test('fromFieldToInput', function() {
        var field = this.blocks.fieldsAndInputs2.inputList[1].fieldRow[0];
        var inputConnection = this.blocks.fieldsAndInputs2.inputList[0].connection;
        var node = Blockly.ASTNode.createFieldNode(field);
        var prevNode = node.prev();
        assertEquals(prevNode.getLocation(), inputConnection);
      });
      test('fromFieldToField', function() {
        var field = this.blocks.fieldsAndInputs.inputList[1].fieldRow[0];
        var field2 = this.blocks.fieldsAndInputs.inputList[0].fieldRow[0];
        var node = Blockly.ASTNode.createFieldNode(field);
        var prevNode = node.prev();
        assertEquals(prevNode.getLocation(), field2);
      });
      test('fromStackToStack', function() {
        var node = Blockly.ASTNode.createStackNode(this.blocks.statementInput4);
        var prevNode = node.prev();
        assertEquals(prevNode.getLocation(), this.blocks.statementInput1);
        assertEquals(prevNode.getType(), Blockly.ASTNode.types.STACK);
      });
    });

    suite('In', function() {
      setup(function() {
        Blockly.defineBlocksWithJsonArray([{
          "type": "dummy_input",
          "message0": "Hello",
          "colour": 230,
          "tooltip": "",
          "helpUrl": ""
        },
        {
          "type": "dummy_inputValue",
          "message0": "Hello %1  %2",
          "args0": [
            {
              "type": "input_dummy"
            },
            {
              "type": "input_value",
              "name": "NAME"
            }
          ],
          "colour": 230,
          "tooltip": "",
          "helpUrl": ""
        }]);
        var dummyInput = this.workspace.newBlock('dummy_input');
        var dummyInputValue = this.workspace.newBlock('dummy_inputValue');
        var fieldWithOutput2 = this.workspace.newBlock('field_input');

        this.blocks.dummyInput = dummyInput;
        this.blocks.dummyInputValue = dummyInputValue;
        this.blocks.fieldWithOutput2 = fieldWithOutput2;
        this.emptyWorkspace = new Blockly.Workspace();
      });
      teardown(function() {
        delete this.blocks.dummyInput;
        delete this.blocks.dummyInputValue;
        delete this.blocks.fieldWithOutput2;
        delete this.emptyWorkspace;

        delete Blockly.Blocks['dummy_input'];
        delete Blockly.Blocks['dummy_inputValue'];
      });

      test('fromInputToOutput', function() {
        var input = this.blocks.statementInput1.inputList[0];
        var node = Blockly.ASTNode.createInputNode(input);
        var inNode = node.in();
        var outputConnection = this.blocks.fieldWithOutput.outputConnection;
        assertEquals(inNode.getLocation(), outputConnection);
      });
      test('fromInputToNull', function() {
        var input = this.blocks.statementInput2.inputList[0];
        var node = Blockly.ASTNode.createInputNode(input);
        var inNode = node.in();
        assertEquals(inNode, null);
      });
      test('fromInputToPrevious', function() {
        var input = this.blocks.statementInput2.inputList[1];
        var previousConnection = this.blocks.statementInput3.previousConnection;
        var node = Blockly.ASTNode.createInputNode(input);
        var inNode = node.in();
        assertEquals(inNode.getLocation(), previousConnection);
      });
      test('fromBlockToInput', function() {
        var input = this.blocks.valueInput.inputList[0];
        var node = Blockly.ASTNode.createBlockNode(this.blocks.valueInput);
        var inNode = node.in();
        assertEquals(inNode.getLocation(), input.connection);
      });
      test('fromBlockToField', function() {
        var node = Blockly.ASTNode.createBlockNode(this.blocks.statementInput1);
        var inNode = node.in();
        var field = this.blocks.statementInput1.inputList[0].fieldRow[0];
        assertEquals(inNode.getLocation(), field);
      });
      test('fromBlockToPrevious', function() {
        var prevConnection = this.blocks.statementInput4.previousConnection;
        var node = Blockly.ASTNode.createStackNode(this.blocks.statementInput4);
        var inNode = node.in();
        assertEquals(inNode.getLocation(), prevConnection);
        assertEquals(inNode.getType(), Blockly.ASTNode.types.PREVIOUS);
      });
      test('fromBlockToNull_DummyInput', function() {
        var node = Blockly.ASTNode.createBlockNode(this.blocks.dummyInput);
        var inNode = node.in();
        assertEquals(inNode, null);
      });
      test('fromBlockToInput_DummyInputValue', function() {
        var node = Blockly.ASTNode.createBlockNode(this.blocks.dummyInputValue);
        var inputConnection = this.blocks.dummyInputValue.inputList[1].connection;
        var inNode = node.in();
        assertEquals(inNode.getLocation(), inputConnection);
      });
      test('fromOuputToNull', function() {
        var output = this.blocks.fieldWithOutput.outputConnection;
        var node = Blockly.ASTNode.createConnectionNode(output);
        var inNode = node.in();
        assertEquals(inNode, null);
      });
      test('fromFieldToNull', function() {
        var field = this.blocks.statementInput1.inputList[0].fieldRow[0];
        var node = Blockly.ASTNode.createFieldNode(field);
        var inNode = node.in();
        assertEquals(inNode, null);
      });
      test('fromWorkspaceToStack', function() {
        var coordinate = new Blockly.utils.Coordinate(100, 100);
        var node = Blockly.ASTNode.createWorkspaceNode(this.workspace, coordinate);
        var inNode = node.in();
        assertEquals(inNode.getLocation(), this.workspace.getTopBlocks()[0]);
        assertEquals(inNode.getType(), Blockly.ASTNode.types.STACK);
      });
      test('fromWorkspaceToNull', function() {
        var coordinate = new Blockly.utils.Coordinate(100, 100);
        var node = Blockly.ASTNode.createWorkspaceNode(
            this.emptyWorkspace, coordinate);
        var inNode = node.in();
        assertEquals(inNode, null);
      });
      test('fromStackToPrevious', function() {
        var node = Blockly.ASTNode.createStackNode(this.blocks.statementInput1);
        var previous = this.blocks.statementInput1.previousConnection;
        var inNode = node.in();
        assertEquals(inNode.getLocation(), previous);
        assertEquals(inNode.getType(), Blockly.ASTNode.types.PREVIOUS);
      });
      test('fromStackToOutput', function() {
        var node = Blockly.ASTNode.createStackNode(this.blocks.fieldWithOutput2);
        var output = this.blocks.fieldWithOutput2.outputConnection;
        var inNode = node.in();
        assertEquals(inNode.getLocation(), output);
        assertEquals(inNode.getType(), Blockly.ASTNode.types.OUTPUT);
      });
      test('fromStackToBlock', function() {
        var node = Blockly.ASTNode.createStackNode(this.blocks.dummyInput);
        var inNode = node.in();
        assertEquals(inNode.getLocation(), this.blocks.dummyInput);
        assertEquals(inNode.getType(), Blockly.ASTNode.types.BLOCK);
      });
    });

    suite('Out', function() {
      setup(function() {
        Blockly.defineBlocksWithJsonArray([{
          "type": "start_block",
          "message0": "",
          "nextStatement": null,
          "colour": 230,
          "tooltip": "",
          "helpUrl": ""
        },
        {
          "type": "output_next",
          "message0": "",
          "output": null,
          "colour": 230,
          "tooltip": "",
          "helpUrl": "",
          "nextStatement": null
        }]);
        var noPrevConnection = this.workspace.newBlock('start_block');
        var secondBlock = this.workspace.newBlock('input_statement');
        var outputNextBlock = this.workspace.newBlock('output_next');
        var fieldWithOutput2 = this.workspace.newBlock('field_input');

        noPrevConnection.nextConnection.connect(secondBlock.previousConnection);
        secondBlock.inputList[0].connection
            .connect(outputNextBlock.outputConnection);
        this.blocks.noPrevConnection = noPrevConnection;
        this.blocks.secondBlock = secondBlock;
        this.blocks.outputNextBlock = outputNextBlock;
        this.blocks.fieldWithOutput2 = fieldWithOutput2;
      });
      teardown(function() {
        delete this.blocks.noPrevConnection;
        delete this.blocks.secondBlock;
        delete this.blocks.outputNextBlock;
        delete this.blocks.fieldWithOutput2;

        delete Blockly.Blocks['start_block'];
        delete Blockly.Blocks['output_next'];
      });

      test('fromInputToBlock', function() {
        var input = this.blocks.statementInput1.inputList[0];
        var node = Blockly.ASTNode.createInputNode(input);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.BLOCK);
        chai.assert.equal(outNode.getLocation(), this.blocks.statementInput1);
      });
      test('fromOutputToInput', function() {
        var output = this.blocks.fieldWithOutput.outputConnection;
        var node = Blockly.ASTNode.createConnectionNode(output);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.INPUT);
        chai.assert.equal(outNode.getLocation(),
            this.blocks.statementInput1.inputList[0].connection);
      });
      test('fromOutputToStack', function() {
        var output = this.blocks.fieldWithOutput2.outputConnection;
        var node = Blockly.ASTNode.createConnectionNode(output);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.STACK);
        chai.assert.equal(outNode.getLocation(), this.blocks.fieldWithOutput2);
      });
      test('fromFieldToBlock', function() {
        var field = this.blocks.statementInput1.inputList[0].fieldRow[0];
        var node = Blockly.ASTNode.createFieldNode(field);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.BLOCK);
        chai.assert.equal(outNode.getLocation(), this.blocks.statementInput1);
      });
      test('fromStackToWorkspace', function() {
        var stub = sinon.stub(this.blocks.statementInput4,
            "getRelativeToSurfaceXY").returns({x: 10, y:10});
        var node = Blockly.ASTNode.createStackNode(this.blocks.statementInput4);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.WORKSPACE);
        chai.assert.equal(outNode.wsCoordinate_.x, 10);
        chai.assert.equal(outNode.wsCoordinate_.y, -10);
        stub.restore();
      });
      test('fromPreviousToInput', function() {
        var previous = this.blocks.statementInput3.previousConnection;
        var inputConnection = this.blocks.statementInput2.inputList[1].connection;
        var node = Blockly.ASTNode.createConnectionNode(previous);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.INPUT);
        chai.assert.equal(outNode.getLocation(), inputConnection);
      });
      test('fromPreviousToStack', function() {
        var previous = this.blocks.statementInput2.previousConnection;
        var node = Blockly.ASTNode.createConnectionNode(previous);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.STACK);
        chai.assert.equal(outNode.getLocation(), this.blocks.statementInput1);
      });
      test('fromNextToInput', function() {
        var next = this.blocks.statementInput3.nextConnection;
        var inputConnection = this.blocks.statementInput2.inputList[1].connection;
        var node = Blockly.ASTNode.createConnectionNode(next);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.INPUT);
        chai.assert.equal(outNode.getLocation(), inputConnection);
      });
      test('fromNextToStack', function() {
        var next = this.blocks.statementInput2.nextConnection;
        var node = Blockly.ASTNode.createConnectionNode(next);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.STACK);
        chai.assert.equal(outNode.getLocation(), this.blocks.statementInput1);
      });
      test('fromNextToStack_NoPreviousConnection', function() {
        var next = this.blocks.secondBlock.nextConnection;
        var node = Blockly.ASTNode.createConnectionNode(next);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.STACK);
        chai.assert.equal(outNode.getLocation(), this.blocks.noPrevConnection);
      });
      /**
       * This is where there is a block with both an output connection and a
       * next connection attached to an input.
       */
      test('fromNextToInput_OutputAndPreviousConnection', function() {
        var next = this.blocks.outputNextBlock.nextConnection;
        var node = Blockly.ASTNode.createConnectionNode(next);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.INPUT);
        chai.assert.equal(outNode.getLocation(),
            this.blocks.secondBlock.inputList[0].connection);
      });
      test('fromBlockToStack', function() {
        var node = Blockly.ASTNode.createBlockNode(this.blocks.statementInput2);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.STACK);
        chai.assert.equal(outNode.getLocation(), this.blocks.statementInput1);
      });
      test('fromBlockToInput', function() {
        var input = this.blocks.statementInput2.inputList[1].connection;
        var node = Blockly.ASTNode.createBlockNode(this.blocks.statementInput3);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.INPUT);
        chai.assert.equal(outNode.getLocation(), input);
      });
      test('fromTopBlockToStack', function() {
        var node = Blockly.ASTNode.createBlockNode(this.blocks.statementInput1);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.STACK);
        chai.assert.equal(outNode.getLocation(), this.blocks.statementInput1);
      });
      test('fromBlockToStack_OutputConnection', function() {
        var node = Blockly.ASTNode.createBlockNode(this.blocks.fieldWithOutput2);
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.STACK);
        chai.assert.equal(outNode.getLocation(), this.blocks.fieldWithOutput2);
      });
      test('fromBlockToInput_OutputConnection', function() {
        var node = Blockly.ASTNode.createBlockNode(this.blocks.outputNextBlock);
        var inputConnection = this.blocks.secondBlock.inputList[0].connection;
        var outNode = node.out();
        chai.assert.equal(outNode.getType(), Blockly.ASTNode.types.INPUT);
        chai.assert.equal(outNode.getLocation(), inputConnection);
      });
    });

    suite('createFunctions', function() {
      test('createFieldNode', function() {
        var field = this.blocks.statementInput1.inputList[0].fieldRow[0];
        var node = Blockly.ASTNode.createFieldNode(field);
        assertEquals(node.getLocation(), field);
        assertEquals(node.getType(), Blockly.ASTNode.types.FIELD);
        assertFalse(node.isConnection());
      });
      test('createConnectionNode', function() {
        var prevConnection = this.blocks.statementInput4.previousConnection;
        var node = Blockly.ASTNode.createConnectionNode(prevConnection);
        assertEquals(node.getLocation(), prevConnection);
        assertEquals(node.getType(), Blockly.ASTNode.types.PREVIOUS);
        assertTrue(node.isConnection());
      });
      test('createInputNode', function() {
        var input = this.blocks.statementInput1.inputList[0];
        var node = Blockly.ASTNode.createInputNode(input);
        assertEquals(node.getLocation(), input.connection);
        assertEquals(node.getType(), Blockly.ASTNode.types.INPUT);
        assertTrue(node.isConnection());
      });
      test('createWorkspaceNode', function() {
        var coordinate = new Blockly.utils.Coordinate(100, 100);
        var node = Blockly.ASTNode
            .createWorkspaceNode(this.workspace, coordinate);
        assertEquals(node.getLocation(), this.workspace);
        assertEquals(node.getType(), Blockly.ASTNode.types.WORKSPACE);
        assertEquals(node.getWsCoordinate(), coordinate);
        assertFalse(node.isConnection());
      });
      test('createStatementConnectionNode', function() {
        var nextConnection = this.blocks.statementInput1.inputList[1].connection;
        var inputConnection = this.blocks.statementInput1.inputList[1].connection;
        var node = Blockly.ASTNode.createConnectionNode(nextConnection);
        assertEquals(node.getLocation(), inputConnection);
        assertEquals(node.getType(), Blockly.ASTNode.types.INPUT);
        assertTrue(node.isConnection());
      });
    });
  });
});