/*
 * Copyright 2014 Red Hat, Inc.
 *
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License v1.0
 *  and Apache License v2.0 which accompanies this distribution.
 *
 *  The Eclipse Public License is available at
 *  http://www.eclipse.org/legal/epl-v10.html
 *
 *  The Apache License v2.0 is available at
 *  http://www.opensource.org/licenses/apache2.0.php
 *
 *  You may elect to redistribute this code under either of these licenses.
 */
package io.vertx.ext.web.handler.sockjs;

import io.vertx.core.buffer.Buffer;
import io.vertx.test.core.TestUtils;
import org.junit.Test;

/**
 * @author <a href="mailto:julien@julienviet.com">Julien Viet</a>
 */
public class SockJSRawTransportTest extends SockJSTestBase {

  @Test
  public void testWriteText() throws Exception {
    testWrite(true);
  }

  @Test
  public void testWriteBinary() throws Exception {
    testWrite(false);
  }

  private void testWrite(boolean text) throws Exception {
    String expected = TestUtils.randomAlphaString(64);
    socketHandler = () -> socket -> {
      if (text) {
        socket.write(expected);
      } else {
        socket.write(Buffer.buffer(expected));
      }
      socket.endHandler(v -> {
        testComplete();
      });
    };
    startServers();
    client.webSocket("/test/websocket", onSuccess(ws -> {
      ws.frameHandler(frame -> {
        if (frame.isClose()) {
          //
        } else {
          if (text) {
            assertTrue(frame.isText());
            assertEquals(expected, frame.textData());
          } else {
            assertTrue(frame.isBinary());
            assertEquals(Buffer.buffer(expected), frame.binaryData());
          }
          ws.end();
        }
      });
    }));
    await();
  }
}
