Fixing the tests

Author: arun-gupta

websocket/endpoint-programmatic/src/main/java/org/javaee7/websocket/endpoint/programmatic/MyEndpointBinaryClient.java

@@ -13,7 +13,7 @@
  */
 @ClientEndpoint
 public class MyEndpointBinaryClient {
-    public static CountDownLatch latch= new CountDownLatch(2);
+    public static CountDownLatch latch;
     public static byte[] response;
 
     @OnOpen

websocket/endpoint-programmatic/src/main/java/org/javaee7/websocket/endpoint/programmatic/MyEndpointTextClient.java

@@ -12,7 +12,7 @@
  */
 @ClientEndpoint
 public class MyEndpointTextClient {
-    public static CountDownLatch latch= new CountDownLatch(2);
+    public static CountDownLatch latch;
     public static byte[] response;
 
     @OnOpen

websocket/endpoint-programmatic/src/test/java/org/javaee7/websocket/endpoint/programmatic/MyEndpointTest.java

@@ -4,6 +4,7 @@
 import java.net.URI;
 import java.net.URISyntaxException;
 import java.nio.ByteBuffer;
+import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeUnit;
 import javax.websocket.ContainerProvider;
 import javax.websocket.DeploymentException;
@@ -41,6 +42,7 @@ public static WebArchive createDeployment() {
 
     @Test
     public void testTextEndpoint() throws URISyntaxException, DeploymentException, IOException, InterruptedException {
+        MyEndpointTextClient.latch = new CountDownLatch(1);
         final String TEXT = "Hello World!";
         Session session = connectToServer(MyEndpointTextClient.class);
         assertNotNull(session);
@@ -55,6 +57,7 @@ public void onMessage(String text) {
 
     @Test
     public void testBinaryEndpoint() throws URISyntaxException, DeploymentException, IOException, InterruptedException {
+        MyEndpointBinaryClient.latch = new CountDownLatch(1);
         final String TEXT = "Hello World!";
         Session session = connectToServer(MyEndpointBinaryClient.class);
         assertNotNull(session);