Add more transaction related tests for Artemis

ballerina-platform · May 22, 2019 · c0ed565 · c0ed565
1 parent ee1e797
commit c0ed565
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 5 deletions.
diff --git a/...ion-test/src/test/java/org/ballerinalang/test/messaging/artemis/LocalTransactionTest.java b/...ion-test/src/test/java/org/ballerinalang/test/messaging/artemis/LocalTransactionTest.java
@@ -23,6 +23,7 @@
 import org.ballerinalang.launcher.util.CompileResult;
 import org.ballerinalang.model.values.BValue;
 import org.ballerinalang.test.util.TestUtils;
+import org.ballerinalang.util.exceptions.BLangRuntimeException;
 import org.testng.Assert;
 import org.testng.annotations.BeforeClass;
 import org.testng.annotations.Test;
@@ -66,4 +67,17 @@ public void testSend() {
         String returnVal = BRunUtil.invoke(consumerResult, "transactionConsumerReceive", consumerVal)[0].stringValue();
         Assert.assertEquals(returnVal, "Example Example ", "Invalid message received");
     }
+
+    @Test(description = "Tests transaction erring")
+    public void testErringSend() {
+        //Invoking this function is needed to make sure the queue is created before the sending
+        BValue[] consumerVal = BRunUtil.invoke(consumerResult, "createErringConsumer");
+        try {
+            BRunUtil.invoke(producerResult, "testErringSend");
+        } catch (BLangRuntimeException ex) {
+            // Ignore
+        }
+        String returnVal = BRunUtil.invoke(consumerResult, "receiveAndGetText", consumerVal)[0].stringValue();
+        Assert.assertEquals(returnVal, "Example ", "Invalid message received");
+    }
 }
diff --git a/...-integration-test/src/test/resources/messaging/artemis/consumers/transaction_consumer.bal b/...-integration-test/src/test/resources/messaging/artemis/consumers/transaction_consumer.bal
@@ -17,7 +17,7 @@
 import ballerina/artemis;
 
 public function createSimpleConsumer() returns artemis:Consumer|error {
-    artemis:Listener lis = new artemis:Listener({host:"localhost", port:61616});
+    artemis:Listener lis = new artemis:Listener({host: "localhost", port: 61616});
     return lis.createAndGetConsumer({queueName: "example"});
 }
 
@@ -46,7 +46,12 @@ public function transactionConsumerReceive(artemis:Consumer consumer) returns st
     return msgTxt;
 }
 
-function receiveAndGetText(artemis:Consumer consumer) returns string {
+public function createErringConsumer() returns artemis:Consumer|error {
+    artemis:Listener lis = new artemis:Listener({host: "localhost", port: 61616});
+    return lis.createAndGetConsumer({queueName: "example3"});
+}
+
+public function receiveAndGetText(artemis:Consumer consumer) returns string {
     string msgTxt = "";
     var msg = consumer->receive();
     if(msg is artemis:Message) {

diff --git a/...-integration-test/src/test/resources/messaging/artemis/producers/transaction_producer.bal b/...-integration-test/src/test/resources/messaging/artemis/producers/transaction_producer.bal
@@ -18,26 +18,36 @@ import ballerina/artemis;
 import ballerina/io;
 
 public function testSimpleTransactionSend() {
-    artemis:Producer prod = new({host:"localhost", port:61616}, "example");
+    artemis:Producer prod = new({host: "localhost", port: 61616}, "example");
     send(prod);
     transaction {
         send(prod);
     }
 }
 
+public function testErringSend() {
+    artemis:Producer prod = new({host: "localhost", port: 61616}, "example3");
+    send(prod);
+    transaction {
+        send(prod);
+        error err = error("Failed during send");
+        panic err;
+    }
+}
+
 public function testTransactionSend() {
     artemis:Connection con = new("tcp://localhost:61616");
     artemis:Session session = new(con);
     artemis:Producer prod = new(session, "example2");
-    send(prod);
     transaction {
         send(prod);
+        send(prod);
     }
 }
 
 function send(artemis:Producer prod) {
     var err = prod->send("Example ");
-    if(err is error) {
+    if (err is error) {
         io:println("Error occurred sending message");
     }
 }