Add finally block to TryNode equals/hashCode/toString

1 files changed, 4 insertions, 2 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/TryNode.java b/src/main/java/org/zwobble/couscous/ast/TryNode.java
index bb276ee..800931f 100644
--- a/src/main/java/org/zwobble/couscous/ast/TryNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/TryNode.java
@@ -61,6 +61,7 @@ public class TryNode implements StatementNode {
         return "TryNode(" +
             "body=" + body +
             ", exceptionHandlers=" + exceptionHandlers +
+            ", finallyBody=" + finallyBody +
             ')';
     }
 
@@ -72,14 +73,15 @@ public class TryNode implements StatementNode {
         TryNode tryNode = (TryNode) o;
 
         if (!body.equals(tryNode.body)) return false;
-        return exceptionHandlers.equals(tryNode.exceptionHandlers);
-
+        if (!exceptionHandlers.equals(tryNode.exceptionHandlers)) return false;
+        return finallyBody.equals(tryNode.finallyBody);
     }
 
     @Override
     public int hashCode() {
         int result = body.hashCode();
         result = 31 * result + exceptionHandlers.hashCode();
+        result = 31 * result + finallyBody.hashCode();
         return result;
     }
 }