PAYARA-1939 : SSE sample test with arquillian

Author: DanielContrerasAladro

sse/producer/pom.xml

@@ -11,12 +11,19 @@
     <packaging>war</packaging>
     <name>Java EE 8 Samples: Server Sent Events [Producer]</name>
     <dependencies>
-        <dependency>
-            <groupId>javax.ejb</groupId>
-            <artifactId>javax.ejb-api</artifactId>
-            <version>3.2</version>
-            <scope>test</scope>
-            <type>jar</type>
-        </dependency>
+    <!-- JSON-B API -->
+    <dependency>
+        <groupId>javax.json.bind</groupId>
+        <artifactId>javax.json.bind-api</artifactId>
+        <version>1.0.0-RC2</version>
+    </dependency>
+
+    <!-- Yasson (JSON-B RI) -->
+    <dependency>
+        <groupId>org.eclipse</groupId>
+        <artifactId>yasson</artifactId>
+        <version>1.0.0-RC1</version>
+        <scope>runtime</scope>
+    </dependency>
     </dependencies>
 </project>

sse/producer/src/main/java/org/javaee8/sse/data/EventData.java

@@ -0,0 +1,61 @@
+/*
+ * To change this license header, choose License Headers in Project Properties.
+ * To change this template file, choose Tools | Templates
+ * and open the template in the editor.
+ */
+package org.javaee8.sse.data;
+
+import java.util.Date;
+import java.util.UUID;
+import javax.json.bind.JsonbBuilder;
+
+/**
+ *
+ * @author daniel
+ */
+public class EventData {
+
+    private Date time;
+    private String id;
+    private String comment;
+
+    public EventData() {
+        super();
+    }
+    
+    public EventData(String comment) {
+        super();
+        this.setTime(new Date());
+        this.setId(UUID.randomUUID().toString());
+        this.setComment(comment);
+    }
+
+    public Date getTime() {
+        return time;
+    }
+
+    public void setTime(Date time) {
+        this.time = time;
+    }
+
+    public String getId() {
+        return id;
+    }
+
+    public void setId(String id) {
+        this.id = id;
+    }
+
+    public String getComment() {
+        return comment;
+    }
+
+    public void setComment(String comment) {
+        this.comment = comment;
+    }
+
+    @Override
+    public String toString() {
+        return JsonbBuilder.create().toJson(this);
+    }
+}

sse/producer/src/main/java/org/javaee8/sse/producer/SseResource.java

@@ -2,6 +2,8 @@
 
 import javax.annotation.PostConstruct;
 import javax.inject.Singleton;
+import javax.json.bind.Jsonb;
+import javax.json.bind.JsonbBuilder;
 import javax.ws.rs.GET;
 import javax.ws.rs.Path;
 import javax.ws.rs.Produces;
@@ -10,49 +12,48 @@
 import javax.ws.rs.sse.Sse;
 import javax.ws.rs.sse.SseBroadcaster;
 import javax.ws.rs.sse.SseEventSink;
+import org.javaee8.sse.data.EventData;
 
 /**
  * @author Daniel Contreras
  */
-@Path("/")
+@Path("sse")
 @Singleton
 public class SseResource {
-
+    
     @Context
     private Sse sse;
 
     private volatile SseBroadcaster sseBroadcaster;
 
     @PostConstruct
     public void init() {
-        System.out.println("1");
         this.sseBroadcaster = sse.newBroadcaster();
-        System.out.println("2");
     }
 
     @GET
     @Path("register")
     @Produces(MediaType.SERVER_SENT_EVENTS)
     public void register(@Context SseEventSink eventSink) {
-        System.out.println("3");
-        eventSink.send(sse.newEvent("welcome!"));
-        System.out.println("4");
+        
+        Jsonb jsonb = JsonbBuilder.create();
+
+        eventSink.send(sse.newEvent("INIT",new EventData("event:intialized").toString()));
+
         sseBroadcaster.register(eventSink);
-        System.out.println("5");
-        eventSink.send(sse.newEvent("event1"));
-        eventSink.send(sse.newEvent("event2"));
-        eventSink.send(sse.newEvent("event3"));
 
-        for (int i = 0; i < 20; i++) {
+        for (int i = 0; i < 5; i++) {
 
-            sseBroadcaster.broadcast(sse.newEvent("Repeated" + i));
+            sseBroadcaster.broadcast(sse.newEvent("EVENT",new EventData("event:"+i).toString()));
 
             try {
-                Thread.sleep(1000);
+                Thread.sleep(100);
             } catch (InterruptedException e) {
                 e.printStackTrace();
             }
         }
+        
+        eventSink.send(sse.newEvent("FINISH",new EventData("event:finished").toString()));
     }
-
+    
 }

sse/producer/src/main/java/org/javaee8/sse/rest/RestApplication.java

@@ -12,8 +12,7 @@
  *
  * @author daniel
  */
-@ApplicationPath(value = "rest")
-public class RestApplication extends Application{
+@ApplicationPath("rest")
+public class RestApplication extends Application {
 
-    
 }

sse/producer/src/main/webapp/index.html

@@ -3,6 +3,8 @@
 To change this license header, choose License Headers in Project Properties.
 To change this template file, choose Tools | Templates
 and open the template in the editor.
+
+@author Daniel Contreras
 -->
 <html>
     <head>
@@ -21,7 +23,7 @@
 
             function start() {
 
-                var eventSource = new EventSource("rest/register");
+                var eventSource = new EventSource("rest/sse/register");
                 console.log(eventSource);
 
                 eventSource.onmessage = function (event) {

sse/producer/src/test/java/org/javaee8/sse/producer/SseResourceTest.java

@@ -4,6 +4,10 @@
 
 import java.io.IOException;
 import java.net.URL;
+import java.util.Arrays;
+import java.util.Date;
+import javax.json.bind.Jsonb;
+import javax.json.bind.JsonbBuilder;
 
 import org.jboss.arquillian.container.test.api.Deployment;
 import org.jboss.arquillian.container.test.api.RunAsClient;
@@ -18,75 +22,81 @@
 import javax.ws.rs.client.Client;
 import javax.ws.rs.client.ClientBuilder;
 import javax.ws.rs.client.WebTarget;
-import javax.ws.rs.core.Context;
 import javax.ws.rs.sse.SseEventSource;
+import static org.hamcrest.CoreMatchers.instanceOf;
+import org.javaee8.sse.data.EventData;
+import org.javaee8.sse.rest.RestApplication;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
 
 /**
  * @author Daniel Contreras
  */
-//@RunWith(Arquillian.class)
+@RunWith(Arquillian.class)
 public class SseResourceTest {
-/*
+
     @ArquillianResource
     private URL base;
 
     private Client sseClient;
-    WebTarget target;
+    private WebTarget target;
+
+    SseEventSource eventSource;
 
-    @Deployment
+    @Deployment(testable = false)
     public static WebArchive createDeployment() {
         return create(WebArchive.class)
-                .addClass(SseResource.class);
+                .addClasses(RestApplication.class, SseResource.class, EventData.class, JsonbBuilder.class, Jsonb.class);
     }
 
     @Before
     public void setup() {
-        System.out.println("01");
         this.sseClient = ClientBuilder.newClient();
-        System.out.println("02");
+        this.target = this.sseClient.target(base + "rest/sse/register");
+        eventSource = SseEventSource.target(target).build();
+        System.out.println("SSE Event source created........");
     }
 
     @After
     public void teardown() {
-
+        eventSource.close();
+        System.out.println("Closed SSE Event source..");
+        sseClient.close();
+        System.out.println("Closed JAX-RS client..");
     }
 
+    String[] types = {"INIT", "EVENT", "FINISH"};
+
     @Test
     @RunAsClient
-    public void testClient() throws IOException {
-        System.out.println("Client");
-        System.out.println(base + "producer/register");
-       System.out.println(this.sseClient.getConfiguration().toString());
-    }
-    
-    @Test
-    public void testServer() throws IOException {
-        System.out.println("Server");
-        System.out.println(base + "producer/register");
-        //this.sseClient.target(base + "producer/register");
-        System.out.println(this.sseClient.getConfiguration().getProperties().toString());
-        System.out.println(this.sseClient.getConfiguration().getRuntimeType().toString());
-        System.out.println(this.sseClient.getConfiguration().getInstances().toArray().toString());
-        //this.sseClient.getHostnameVerifier();
-
-        System.out.println(base + "producer/register");
-        WebTarget target = sseClient.target(base + "producer/register");
-        System.out.println("1");
-        try (SseEventSource source = SseEventSource.target(target).build()) {
-            System.out.println("2");
-            source.register(System.out::println);
-            System.out.println("3");
-            source.open();
-            System.out.println("5");
-            Thread.sleep(500);      // Consume events for just 500 ms
-
+    public void testSSE() throws IOException {
+
+        Jsonb jsonb = JsonbBuilder.create();
+
+        System.out.println("SSE Client triggered in thread " + Thread.currentThread().getName());
+        try {
+            eventSource.register(
+                    (sseEvent)
+                    -> {
+                assertTrue(Arrays.asList(types).contains(sseEvent.getName()));
+                assertNotNull(sseEvent.readData());
+                EventData event = jsonb.fromJson(sseEvent.readData(), EventData.class);
+                assertThat(event.getTime(), instanceOf(Date.class));
+                assertNotNull(event.getId());
+                assertTrue(event.getComment().contains("event:"));
+                System.out.println("\nSSE Event received :: " + event.toString() +"\n");
+                
+            },
+                    (e) -> e.printStackTrace());
+
+            eventSource.open();
+            Thread.sleep(1500);
         } catch (Exception e) {
-
-            System.out.println("error!!!");
+            System.out.println("Error on SSE Test");
             System.out.println(e.getMessage());
-
         }
 
     }
-*/
+
 }