include("mjsunit");

var port = 12123;
var N = 1000;
var count = 0;

function Ponger (socket) {
  socket.setEncoding("utf8");
  socket.timeout = 0;

  puts("got socket.");

  socket.onReceive = function (data) {
    assertEquals("open", socket.readyState);
    //puts("server recved data: " + JSON.stringify(data));
    assertTrue(count <= N);
    stdout.print("-");
    if (/PING/.exec(data)) {
      socket.send("PONG");
    }
  };

  socket.onEOF = function () {
    assertEquals("writeOnly", socket.readyState);
    puts("ponger: onEOF");
    socket.close();
  };

  socket.onDisconnect = function (had_error) {
    assertFalse(had_error);
    assertEquals("closed", socket.readyState);
    puts("ponger: onDisconnect");
    socket.server.close();
  };
}

function onLoad() {
  var server = new node.tcp.Server(Ponger);
  server.listen(port);

  var client = new node.tcp.Connection();
  assertEquals("closed", client.readyState);

  client.setEncoding("utf8");

  client.onConnect = function () {
    assertEquals("open", client.readyState);
    puts("client is connected.");
    client.send("PING");
  };

  var sent_final_ping = false;

  client.onReceive = function (data) {
    //puts("client recved data: " + JSON.stringify(data));
    stdout.print(".");
    assertEquals("PONG", data);
    count += 1; 

    if (sent_final_ping) {
      assertEquals("readOnly", client.readyState);
      return;
    } else {
      assertEquals("open", client.readyState);
    }

    if (count < N) {
      client.send("PING");
    } else {
      puts("sending final ping");
      sent_final_ping = true;
      client.send("PING");
      client.close();
    }
  };
  
  client.onEOF = function () {
    puts("pinger: onEOF");
    assertEquals(N+1, count);
  };

  client.connect(port);
  assertEquals("closed", client.readyState);
}