/**
 * @author David Yu
 * @created Sep 30, 2009
 */
public class ProtobufJSONTest extends TestCase {
  static final ReflectionJSON REFLECTION_LITE = new ReflectionJSON(new Class[] {V2Lite.class});
  static final ReflectionNumericJSON REFLECTION_LITE_NUM =
      new ReflectionNumericJSON(new Class[] {V2Lite.class});
  static final V2LiteHandcodedJSON HANDCODED_LITE = new V2LiteHandcodedJSON();

  static final Charset utf8 = Charset.forName("UTF-8");

  static final Task task =
      Task.newBuilder()
          .setId(1)
          .setName("task_name")
          .setDescription("task_description")
          .setStatus(Task.Status.COMPLETED)
          .setAttachment(ByteString.copyFrom(new byte[] {0x34}))
          .setAboolean(true)
          .setAfloat(10.101f)
          .setAdouble(10.1010101d)
          .setAlong(1010101l)
          .build();

  static final Person person =
      Person.newBuilder()
          .setId(1)
          .setFirstName("john")
          .setLastName("doe")
          .setAGe(2)
          .setEmail("*****@*****.**")
          .setCurrentTask(task)
          // .addDelegatedTask(task)
          .addPriorityTask(task)
          .addPriorityTask(task)
          .addRepeatedLong(3)
          .addRepeatedLong(4)
          .addRepeatedLong(5)
          .addImage(ByteString.copyFrom(new byte[] {0x35}))
          .addImage(ByteString.copyFrom(new byte[] {0x36}))
          .addImage(ByteString.copyFrom(new byte[] {0x37}))
          .addImage(ByteString.copyFrom(new byte[] {0x38}))
          .build();

  static void generateAndParse(PrintStream out) throws Exception {
    String generated = generateAndParse(HANDCODED_LITE);
    String lite = generateAndParse(REFLECTION_LITE);
    String numLite = generateAndParse(REFLECTION_LITE_NUM);

    assertEquals(generated, lite);

    out.println(generated);
    out.println(lite);
    out.println(numLite);
  }

  static String generateAndParse(ProtobufJSON json) throws Exception {
    StringWriter sw = new StringWriter();
    json.writeTo(sw, person);
    String generated = sw.toString();

    JsonParser parser = json.getJsonFactory().createJsonParser(generated);
    Person parsedPerson = json.parseFrom(parser, Person.class);
    parser.close();

    assertEquals(person, parsedPerson);

    return generated;
  }

  static void assertEquals(Person p, Person p2) {
    assertTrue(p.getId() == p2.getId());

    assertEquals(p.getFirstName(), p2.getFirstName());

    assertEquals(p.getLastName(), p2.getLastName());

    assertTrue(p.getAGe() == p2.getAGe());

    assertEquals(p.getEmail(), p2.getEmail());

    assertEquals(p.getCurrentTask(), p2.getCurrentTask());

    assertTrue(p.getDelegatedTaskCount() == p2.getDelegatedTaskCount());

    for (int i = 0, len = p.getDelegatedTaskCount(); i < len; i++) {
      assertEquals(p.getDelegatedTask(i), p2.getDelegatedTask(i));
    }

    assertTrue(p.getImageCount() == p2.getImageCount());
    for (int i = 0, len = p.getImageCount(); i < len; i++) {
      assertEquals(p.getImage(i), p2.getImage(i));
    }
  }

  static void assertEquals(Task t, Task t2) {
    if (t == null && t2 == null) return;

    assertTrue(t.getId() == t2.getId());

    assertEquals(t.getName(), t2.getName());

    assertEquals(t.getDescription(), t2.getDescription());

    assertTrue(t.getStatus() == t2.getStatus());

    assertEquals(t.getAttachment(), t2.getAttachment());

    assertTrue(t.getAboolean() == t2.getAboolean());

    assertTrue(t.getAfloat() == t2.getAfloat());

    assertTrue(t.getAdouble() == t2.getAdouble());

    assertTrue(t.getAlong() == t2.getAlong());
  }

  public void jacksonTest() throws Exception {
    JsonFactory factory = new JsonFactory();
    StringWriter writer = new StringWriter();
    JsonGenerator generator = factory.createJsonGenerator(writer);
    generator.writeStartObject();
    generator.writeFieldName("bool");
    generator.writeBoolean(true);
    generator.writeFieldName("firstName");
    generator.writeString("john");
    generator.writeFieldName("age");
    generator.writeNumber(1);
    generator.writeFieldName("gg");
    generator.writeStartObject();
    generator.writeFieldName("firstName");
    generator.writeString("john");
    generator.writeEndObject();

    generator.writeEndObject();
    generator.close();
    String generated = writer.toString();
    System.out.print(generated);

    JsonParser parser = factory.createJsonParser(generated);
    assertTrue(parser.nextToken() == JsonToken.START_OBJECT);
    parser.nextToken();

    assertEquals("bool", parser.getCurrentName());
    assertTrue(parser.nextToken() == JsonToken.VALUE_TRUE);
    parser.nextToken();

    assertEquals("firstName", parser.getCurrentName());
    parser.nextToken();
    assertEquals("john", parser.getText());
    parser.nextToken();

    assertEquals("age", parser.getCurrentName());
    parser.nextToken();
    assertTrue(1 == parser.getIntValue());
    parser.nextToken();

    assertEquals("gg", parser.getCurrentName());
    assertTrue(parser.nextToken() == JsonToken.START_OBJECT);
    parser.nextToken();
    assertEquals("firstName", parser.getCurrentName());
    parser.nextToken();
    assertEquals("john", parser.getText());
    assertTrue(parser.nextToken() == JsonToken.END_OBJECT);

    assertTrue(parser.nextToken() == JsonToken.END_OBJECT);

    parser.close();
  }

  static String generatedAndParse(
      ProtobufJSON json, List<Person> personList, List<Person> parsedPersonList) throws Exception {
    StringWriter sw = new StringWriter();
    json.writeTo(sw, personList, Person.class);
    String generated = sw.toString();

    JsonParser parser = json.getJsonFactory().createJsonParser(generated);
    json.appendMessageFrom(parser, parsedPersonList, Person.class);
    parser.close();
    assertTrue(personList.size() == parsedPersonList.size());
    for (int i = 0, len = personList.size(); i < len; i++)
      assertEquals(personList.get(i), parsedPersonList.get(i));

    return generated;
  }

  public void testCollection() throws Exception {
    ArrayList<Person> personList = new ArrayList<Person>();
    ArrayList<Person> parsedPersonList = new ArrayList<Person>();
    personList.add(person);
    personList.add(person);

    // System.err.println(generatedAndParse(HANDCODED_LITE, personList, parsedPersonList));
    parsedPersonList.clear();

    // System.err.println(generatedAndParse(REFLECTION_LITE, personList, parsedPersonList));
    parsedPersonList.clear();

    // System.err.println(generatedAndParse(REFLECTION_LITE_NUM, personList, parsedPersonList));
    parsedPersonList.clear();
  }

  public void testGenerateAndParse() throws Exception {
    generateAndParse(HANDCODED_LITE);
    generateAndParse(REFLECTION_LITE);
    generateAndParse(REFLECTION_LITE_NUM);
    // System.err.println(generateAndParse(HANDCODED_LITE));
    // System.err.println(generateAndParse(REFLECTION_LITE));
    // System.err.println(generateAndParse(REFLECTION_LITE_NUM));
  }

  public void testBenchmark() throws Exception {
    if (!"false".equals(System.getProperty("benchmark.skip"))) return;

    String dir = System.getProperty("benchmark.output_dir");

    PrintStream out =
        dir == null
            ? System.out
            : new PrintStream(
                new FileOutputStream(
                    new File(
                        new File(dir),
                        "protostuff-benchmark-" + System.currentTimeMillis() + ".html"),
                    true));

    int warmups = Integer.getInteger("benchmark.warmups", 100000);
    int loops = Integer.getInteger("benchmark.loops", 1000000);
    String title = "protostuff-json ser/deser benchmark for " + loops + " runs";
    out.println("<html><head><title>");
    out.println(title);
    out.println("</title></head><body><p>");
    out.println(title);
    out.println("</p><pre>\n\n");

    generateAndParse(out);
    out.println("\n\n</pre><hr/><pre>");
    start(out, warmups, loops);

    if (System.out != out) out.close();
  }

  public static void main(String[] args) throws Exception {
    String dir = System.getProperty("benchmark.output_dir");

    PrintStream out =
        dir == null
            ? System.out
            : new PrintStream(
                new FileOutputStream(
                    new File(
                        new File(dir),
                        "protostuff-benchmark-" + System.currentTimeMillis() + ".html"),
                    true));
    int runs = Integer.getInteger("benchmark.runs", 1);

    int warmups = Integer.getInteger("benchmark.warmups", 100000);
    int loops = Integer.getInteger("benchmark.loops", 1000000);
    String title = "protostuff-json ser/deser benchmark for " + loops + " runs";
    out.println("<html><head><title>");
    out.println(title);
    out.println("</title></head><body><p>");
    out.println(title);
    out.println("</p><pre>\n\n");

    generateAndParse(out);
    out.println("\n\n</pre><hr/><pre>");
    for (int i = 0; i < runs; i++) start(out, warmups, loops);

    if (System.out != out) out.close();
  }

  public static void start(PrintStream out, int warmups, int loops) throws Exception {
    serDeser(out, HANDCODED_LITE, "HANDCODED_LITE", warmups, loops);
    serDeser(out, REFLECTION_LITE, "REFLECTION_LITE", warmups, loops);
    serDeser(out, REFLECTION_LITE_NUM, "REFLECTION_LITE_NUM", warmups, loops);
  }

  static void serDeser(PrintStream out, ProtobufJSON json, String name, int warmups, int loops)
      throws Exception {
    int len = generateAndParse(json).getBytes(utf8).length;
    for (int i = 0; i < warmups; i++) generateAndParse(json);
    long start = System.currentTimeMillis();
    for (int i = 0; i < loops; i++) generateAndParse(json);
    long finish = System.currentTimeMillis();
    long elapsed = finish - start;
    out.println(elapsed + " ms elapsed with " + len + " bytes for " + name);
  }
}
  static void assertEquals(Person p, Person p2) {
    assertTrue(p.getId() == p2.getId());

    assertEquals(p.getFirstName(), p2.getFirstName());

    assertEquals(p.getLastName(), p2.getLastName());

    assertTrue(p.getAGe() == p2.getAGe());

    assertEquals(p.getEmail(), p2.getEmail());

    assertEquals(p.getCurrentTask(), p2.getCurrentTask());

    assertTrue(p.getDelegatedTaskCount() == p2.getDelegatedTaskCount());

    for (int i = 0, len = p.getDelegatedTaskCount(); i < len; i++) {
      assertEquals(p.getDelegatedTask(i), p2.getDelegatedTask(i));
    }

    assertTrue(p.getImageCount() == p2.getImageCount());
    for (int i = 0, len = p.getImageCount(); i < len; i++) {
      assertEquals(p.getImage(i), p2.getImage(i));
    }
  }