Merge pull request #277 from cirocosta/binary-ack

Fixes event type when emitting binary_ack

Author: nkzawa

src/main/java/io/socket/client/Socket.java

@@ -346,7 +346,13 @@ public void run() {
                         sent[0] = true;
                         logger.fine(String.format("sending ack %s", args.length != 0 ? args : null));
 
-                        int type = HasBinary.hasBinary(args) ? Parser.BINARY_ACK : Parser.ACK;
+                        JSONArray jsonArgs = new JSONArray();
+                        for (Object arg : args) {
+                            jsonArgs.put(arg);
+                        }
+
+                        int type = HasBinary.hasBinary(jsonArgs)
+                            ? Parser.BINARY_ACK : Parser.ACK;
                         Packet<JSONArray> packet = new Packet<JSONArray>(type, new JSONArray(Arrays.asList(args)));
                         packet.id = id;
                         self.packet(packet);

src/test/java/io/socket/client/ConnectionTest.java

@@ -3,6 +3,7 @@
 import io.socket.emitter.Emitter;
 import org.json.JSONException;
 import org.json.JSONObject;
+import org.junit.Assert;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
@@ -90,6 +91,7 @@ public void call(Object... args) {
     @Test(timeout = TIMEOUT)
     public void receiveDateWithAck() throws URISyntaxException, InterruptedException {
         final BlockingQueue<Object> values = new LinkedBlockingQueue<Object>();
+
         socket = client();
         socket.on(Socket.EVENT_CONNECT, new Emitter.Listener() {
             @Override
@@ -111,6 +113,61 @@ public void call(Object... args) {
         socket.close();
     }
 
+    @Test(timeout = TIMEOUT)
+    public void sendBinaryAck() throws URISyntaxException, InterruptedException {
+        final BlockingQueue<Object> values = new LinkedBlockingQueue<Object>();
+        final byte[] buf = "huehue".getBytes(Charset.forName("UTF-8"));
+
+        socket = client();
+        socket.on(Socket.EVENT_CONNECT, new Emitter.Listener() {
+            @Override
+            public void call(Object... objects) {
+                socket.emit("callAckBinary");
+                socket.on("ack", new Emitter.Listener() {
+                    @Override
+                    public void call(Object... args) {
+                        Ack fn = (Ack) args[0];
+                        fn.call(buf);
+                    }
+                });
+
+                socket.on("ackBack", new Emitter.Listener() {
+                    @Override
+                    public void call(Object... args) {
+                        byte[] data = (byte[])args[0];
+                        values.offer(data);
+                    }
+                });
+            }
+        });
+        socket.connect();
+        Assert.assertArrayEquals(buf, (byte[])values.take());
+        socket.close();
+    }
+
+    @Test(timeout = TIMEOUT)
+    public void receiveBinaryDataWithAck() throws URISyntaxException, InterruptedException {
+        final BlockingQueue<Object> values = new LinkedBlockingQueue<Object>();
+        final byte[] buf = "huehue".getBytes(Charset.forName("UTF-8"));
+
+        socket = client();
+        socket.on(Socket.EVENT_CONNECT, new Emitter.Listener() {
+            @Override
+            public void call(Object... objects) {
+                socket.emit("getAckBinary", "", new Ack() {
+
+                    @Override
+                    public void call(Object... args) {
+                       values.offer(args[0]);
+                    }
+                });
+            }
+        });
+        socket.connect();
+        Assert.assertArrayEquals(buf, (byte[])values.take());
+        socket.close();
+    }
+
     @Test(timeout = TIMEOUT)
     public void workWithFalse() throws URISyntaxException, InterruptedException {
         final BlockingQueue<Object> values = new LinkedBlockingQueue<Object>();
@@ -731,6 +788,7 @@ public void call(Object... args) {
         socket.close();
     }
 
+
     @Test(timeout = TIMEOUT)
     public void sendAndGetBinaryData() throws URISyntaxException, InterruptedException {
         final BlockingQueue<Object> values = new LinkedBlockingQueue<Object>();

src/test/resources/server.js

@@ -57,6 +57,17 @@ io.of(nsp).on('connection', function(socket) {
     });
   });
 
+  socket.on('callAckBinary', function() {
+    socket.emit('ack', function(buf) {
+      socket.emit('ackBack', buf);
+    });
+  });
+
+  socket.on('getAckBinary', function(data, callback) {
+    var buf = new Buffer('huehue', 'utf8');
+    callback(buf);
+  });
+
   socket.on('getAckDate', function(data, callback) {
     callback(new Date());
   });