JsonTelemetryDataSerializerTest.java

/*
 * ApplicationInsights-Java
 * Copyright (c) Microsoft Corporation
 * All rights reserved.
 *
 * MIT License
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this
 * software and associated documentation files (the ""Software""), to deal in the Software
 * without restriction, including without limitation the rights to use, copy, modify, merge,
 * publish, distribute, sublicense, and/or sell copies of the Software, and to permit
 * persons to whom the Software is furnished to do so, subject to the following conditions:
 * The above copyright notice and this permission notice shall be included in all copies or
 * substantial portions of the Software.
 * THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
 * INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
 * PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE
 * FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */

package com.microsoft.applicationinsights.telemetry;

import java.io.BufferedWriter;
import java.io.IOException;
import java.io.Serializable;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.google.gson.Gson;
import org.junit.Test;

import static org.junit.Assert.assertEquals;

public class JsonTelemetryDataSerializerTest {
    private final static class TestClassWithStrings implements JsonSerializable, Serializable {
        private String s1;
        private String s2;

        @Override
        public void serialize(JsonTelemetryDataSerializer serializer) throws IOException {
            serializer.write("s1", s1);
            serializer.write("s2", s2);
        }

        public String getS1() {
            return s1;
        }

        public void setS1(String s1) {
            this.s1 = s1;
        }

        public String getS2() {
            return s2;
        }

        public void setS2(String s2) {
            this.s2 = s2;
        }

        @Override
        public boolean equals(Object other) {
            if (this == other) {
                return true;
            }

            if (!(other instanceof TestClassWithStrings)) {
                return false;
            }

            TestClassWithStrings that = (TestClassWithStrings)other;
            return this.s1.equals(that.getS1()) && this.s2.equals(that.getS2());
        }
    }

    private final static class StubClass implements JsonSerializable, Serializable {
        private int i1;
        private int i2;
        private String s1;
        private Long l1;
        private long l2;
        private String s2;
        private Map<String, Integer> m1 = new HashMap<String, Integer>();
        private List<String> list1 = new ArrayList<String>();

        public int getI1() {
            return i1;
        }

        public void setI1(int i1) {
            this.i1 = i1;
        }

        public int getI2() {
            return i2;
        }

        public void setI2(int i2) {
            this.i2 = i2;
        }

        public String getS1() {
            return s1;
        }

        public void setS1(String s1) {
            this.s1 = s1;
        }

        public Long getL1() {
            return l1;
        }

        public void setL1(Long l1) {
            this.l1 = l1;
        }

        public long getL2() {
            return l2;
        }

        public void setL2(long l2) {
            this.l2 = l2;
        }

        public String getS2() {
            return s2;
        }

        public void setS2(String s2) {
            this.s2 = s2;
        }

        public Map<String, Integer> getM1() {
            return m1;
        }

        public void setM1(Map<String, Integer> m1) {
            this.m1 = m1;
        }

        public List<String> getList1() {
            return list1;
        }

        public void setList1(List<String> list1) {
            this.list1 = list1;
        }

        @Override
        public void serialize(JsonTelemetryDataSerializer serializer) throws IOException {
            serializer.write("i1", i1);
            serializer.write("i2", i2);
            serializer.write("s1", s1);
            serializer.write("l1", l1);
            serializer.write("l2", l2);
            serializer.write("s2", s2);
            serializer.write("m1", m1);
            serializer.write("list1", list1);
        }

        @Override
        public boolean equals(Object other) {
            if (this == other) {
                return true;
            }

            if (!(other instanceof TestClassWithStrings)) {
                return false;
            }

            StubClass that = (StubClass)other;
            return
                    this.s1.equals(that.getS1()) && this.s2.equals(that.getS2()) &&
                    this.i1 == that.getI1() && this.i2 == that.getI2() &&
                            (this.l1 == null ? that.getL1() == null : this.l1.equals(getL1())) && this.l2 == that.getL2() &&
                    this.list1.equals(that.getList1()) && this.m1.equals(that.getM1());
        }
    }

    @Test
    public void testStrings() throws IOException {
        TestClassWithStrings testClassWithStrings = new TestClassWithStrings();
        testClassWithStrings.setS1("s1");
        testClassWithStrings.setS2("s2");

        StringWriter stringWriter = new StringWriter();
        BufferedWriter bufferedWriter = new BufferedWriter(stringWriter);
        JsonTelemetryDataSerializer tested = new JsonTelemetryDataSerializer(stringWriter);
        testClassWithStrings.serialize(tested);
        tested.close();
        String str = stringWriter.toString();
        TestClassWithStrings bac = new Gson().fromJson(str, TestClassWithStrings.class);
        assertEquals(bac, testClassWithStrings);
    }

    @Test
    public void test() throws IOException {
        StubClass stubClass = new StubClass();
        stubClass.setI1(1);
        stubClass.setI2(2);
        stubClass.setL1(3L);
        stubClass.setL2(4L);
        stubClass.getList1().add("str1");
        stubClass.getList1().add("str2");
        stubClass.getM1().put("key1", 5);
        stubClass.getM1().put("key2", 6);
        stubClass.getM1().put("key3", 7);

        StringWriter stringWriter = new StringWriter();
        BufferedWriter bufferedWriter = new BufferedWriter(stringWriter);
        JsonTelemetryDataSerializer tested = new JsonTelemetryDataSerializer(stringWriter);
        stubClass.serialize(tested);
        tested.close();
        String str = stringWriter.toString();
        StubClass bac = new Gson().fromJson(str, StubClass.class);
//        System.out.println(str);
    }

}