From 9fb39b1cff12bb2ec80ac9cd798f8a50069f4afc Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 22 Apr 2016 22:16:11 +0100
Subject: Handle try-with-resources

---
 .../tests/frontends/java/JavaReaderTests.java      | 29 +++++++++++++++++++++-
 .../couscous/tests/frontends/java/JavaReading.java |  2 +-
 .../tests/frontends/java/NodeMatchers.java         |  5 ++++
 3 files changed, 34 insertions(+), 2 deletions(-)

(limited to 'src/test')

diff --git a/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java b/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
index 8225baa..35756f5 100644
--- a/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
@@ -10,6 +10,7 @@ import org.zwobble.couscous.types.Types;
 import java.util.List;
 
 import static org.hamcrest.Matchers.contains;
+import static org.hamcrest.Matchers.equalTo;
 import static org.hamcrest.Matchers.hasSize;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertThat;
@@ -23,12 +24,14 @@ import static org.zwobble.couscous.ast.FormalTypeParameterNode.formalTypeParamet
 import static org.zwobble.couscous.ast.IfStatementNode.ifStatement;
 import static org.zwobble.couscous.ast.LiteralNode.literal;
 import static org.zwobble.couscous.ast.LocalVariableDeclarationNode.localVariableDeclaration;
+import static org.zwobble.couscous.ast.MethodCallNode.methodCall;
 import static org.zwobble.couscous.ast.MethodCallNode.staticMethodCall;
 import static org.zwobble.couscous.ast.Operations.boxInt;
 import static org.zwobble.couscous.ast.Operations.integerAdd;
 import static org.zwobble.couscous.ast.ReturnNode.returns;
 import static org.zwobble.couscous.ast.ThisReferenceNode.thisReference;
 import static org.zwobble.couscous.ast.ThrowNode.throwNode;
+import static org.zwobble.couscous.ast.TryNode.tryStatement;
 import static org.zwobble.couscous.ast.TypeCoercionNode.typeCoercion;
 import static org.zwobble.couscous.ast.VariableReferenceNode.reference;
 import static org.zwobble.couscous.ast.WhileNode.whileLoop;
@@ -209,7 +212,31 @@ public class JavaReaderTests {
                 hasTryBody(list(returns(literal(1)))),
                 hasFinally(returns(literal(2)))));
     }
-    
+
+    @Test
+    public void canReadTryWithResourceWithoutOtherClauses() {
+        ScalarType type = ScalarType.of("java.io.ByteArrayOutputStream");
+        Identifier identifier = Identifiers.variable(
+            Identifiers.method(
+                Identifiers.type(Identifiers.TOP, "com.example.Example"),
+                "main"
+            ),
+            "closeable"
+        );
+        LocalVariableDeclarationNode expectedResource = localVariableDeclaration(identifier, "closeable", type, constructorCall(type, list()));
+        assertThat(
+            readStatements("int", "try(java.io.ByteArrayOutputStream closeable = new java.io.ByteArrayOutputStream()) { return 1; }"),
+            equalTo(list(
+                expectedResource,
+                tryStatement(
+                    list(returns(literal(1))),
+                    list(),
+                    list(expressionStatement(methodCall(reference(expectedResource), "close", list(), Types.VOID)))
+                )
+            ))
+        );
+    }
+
     @Test
     public void canDeclareConstructor() {
         ClassNode classNode = readClass(
diff --git a/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReading.java b/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReading.java
index a7dc180..e64de11 100644
--- a/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReading.java
+++ b/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReading.java
@@ -76,7 +76,7 @@ public class JavaReading {
     }
 
     static String generateMethodSource(String returnType, String statementsSource) {
-        return "public static " + returnType + " main() {" +
+        return "public static " + returnType + " main() throws Exception {" +
             statementsSource +
             "}";
     }
diff --git a/src/test/java/org/zwobble/couscous/tests/frontends/java/NodeMatchers.java b/src/test/java/org/zwobble/couscous/tests/frontends/java/NodeMatchers.java
index 5a39e82..e9c77f0 100644
--- a/src/test/java/org/zwobble/couscous/tests/frontends/java/NodeMatchers.java
+++ b/src/test/java/org/zwobble/couscous/tests/frontends/java/NodeMatchers.java
@@ -57,6 +57,11 @@ public class NodeMatchers {
         return hasFeature("type", ExpressionNode::getType, equalTo(type));
     }
 
+    @SafeVarargs
+    public static Matcher<StatementNode> isLocalVariableDeclaration(Matcher<? super LocalVariableDeclarationNode>... matchers) {
+        return isInstance(LocalVariableDeclarationNode.class, allOf(matchers));
+    }
+
     @SafeVarargs
     public static Matcher<StatementNode> isTryStatement(Matcher<? super TryNode>... matchers) {
         return isInstance(TryNode.class, allOf(matchers));
-- 
cgit v1.2.3