/*
 * Copyright (C) 2013 EBay Software Foundation
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 *
 * Authors : Aswin Raveendran
 */
package org.opendaylight.ovsdb.lib.jsonrpc;

import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;
import junit.framework.TestCase;
import org.junit.Test;

import java.io.OutputStream;
import java.io.IOException;
import java.net.Socket;

public class TestClient extends TestCase {

        String serverurl = "127.0.0.1";
        int serverport = 12345;

        NettyBootStrapper bootstrapper = new NettyBootStrapper();
        JsonRpcDecoder jsonRpcDecoder = new JsonRpcDecoder(100000);

        public void setupServer() throws Exception {
            bootstrapper.startServer(serverport,
                    jsonRpcDecoder,
                    new LoggingHandler(LogLevel.DEBUG));
        }

        public void shutDownServer() throws InterruptedException {
            bootstrapper.stopServer();
        }

        /**
         * Testing appropriate ChannelHandler integration for
         * JsonRpcDecoder, so that Json strings written using an
         * OutputStream connected to a ServerSocket of a Netty
         * ServerBootstrap can be decoder properly.
         */

        @Test
        public void testBasicFlow() throws Exception {
            setupServer();
            Socket socket = new Socket(serverurl, serverport);
            OutputStream outputStream = socket.getOutputStream();

            int records = 20;

            for (int i = 0; i < records; i++) {
                writeJson(outputStream, 1);
                writePartialFirst(outputStream);
                outputStream.flush();
                Thread.sleep(10);
                writePartialLast(outputStream);
            }
            socket.close();
            shutDownServer();

            assertEquals("mismatch in records processed", records * 2, jsonRpcDecoder.getRecordsRead());
        }

    static int counter = 0;

    /*
       create and write a json string for specified number of times
     */
    private void writeJson(OutputStream outputStream, int times) throws IOException {
        outputStream.write("{".getBytes("UTF-8"));
        for (int i = 0 ; i < times; i ++) {
            counter++;
            String s = ",\"key1"+ counter +"\":\"planet of apes" + counter +
                    "\", \"key2"+ counter +"\":{\"k1\":\"ovs-db rocks the world\"}";
            outputStream.write(s.substring(i == 0 ? 1 : 0).getBytes("UTF-8"));
            System.out.println("data counter = " + counter);
        }
        outputStream.write("}".getBytes("UTF-8"));
    }

    /*
      writes a partial json and flush to simulate the case where netty gets half the message and
      has to frame it accordingly.
     */
    private void writePartialFirst(OutputStream outputStream) throws IOException {
        counter++;
        String s  = "                       {\"part"+ counter+"\":";
        outputStream.write(s.getBytes("UTF-8"));
        System.out.println("partial first half counter = " + counter);
    }

    /*
      finishes the json started by writePartialFirst
     */
    private void writePartialLast(OutputStream outputStream) throws IOException {
        String s  = "\"val"+ counter+"\"}";
        outputStream.write(s.getBytes("UTF-8"));
        System.out.println("partial second half counter = " + counter);
    }

}