Merge pull request ballerina-platform#897 from nadeeshaan/master

Fixing the action invocation delete issue
Shan1024 · Jan 21, 2017 · 294495c · 294495c
2 parents a0e0035 + e91179f
commit 294495c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/modules/editor/web/js/ballerina/ast/assignment-statement.js b/modules/editor/web/js/ballerina/ast/assignment-statement.js
@@ -42,5 +42,13 @@ define(['lodash', './statement'], function (_, Statement) {
         });
     };
 
+    /**
+     * Override the removeChild function
+     * @param {ASTNode} child - child node
+     */
+    AssignmentStatement.prototype.removeChild = function (child) {
+        this.getParent().removeChild(this);
+    };
+
     return AssignmentStatement;
 });
diff --git a/modules/editor/web/js/ballerina/ast/right-operand-expression.js b/modules/editor/web/js/ballerina/ast/right-operand-expression.js
@@ -46,6 +46,14 @@ define(['lodash', './statement'], function(_, Statement){
         this._right_operand_expression_string = rightOperandExpStr;
     };
 
+    /**
+     * Override the removeChild function
+     * @param {ASTNode} child - child node
+     */
+    RightOperandExpression.prototype.removeChild = function (child) {
+        this.getParent().removeChild(this);
+    };
+
     /**
      * setting parameters from json
      * @param jsonNode