/*
 * Copyright 2012-2018 Chronicle Map Contributors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package net.openhft.lang.values;

import net.openhft.chronicle.core.OS;
import net.openhft.chronicle.core.pool.ClassAliasPool;
import net.openhft.chronicle.core.util.Time;
import net.openhft.chronicle.map.ChronicleMap;
import net.openhft.chronicle.map.ChronicleMapBuilder;
import org.json.JSONException;
import org.junit.Test;
import org.skyscreamer.jsonassert.JSONAssert;

import java.io.File;
import java.io.IOException;

import static junit.framework.TestCase.assertSame;
import static org.junit.Assert.assertEquals;

public class ArrayTest {

    @Test
    public void test0() throws IOException, JSONException {
        ClassAliasPool.CLASS_ALIASES.addAlias(MovingAverageArray.class);

        File file = new File(OS.getTarget() + "/pf-PosistionsAndClose-" + Time.uniqueId());

        ChronicleMap<Long, MovingAverageArray> mapWrite = ChronicleMap
                .of(Long.class, MovingAverageArray.class)
                .entries(100)
                .averageValue(createSampleWithSize(6, 0))
                .createPersistedTo(file);
        mapWrite.put(1L, createSampleWithSize(6, 1));
        mapWrite.put(2L, createSampleWithSize(4, 2));
        mapWrite.close();

        ChronicleMap<Long, MovingAverageArray> mapRead = ChronicleMapBuilder
                .of(Long.class, MovingAverageArray.class)
                .createPersistedTo(file);
        MovingAverageArray m = mapRead.get(1L);
        assertJSONEqualsAfterPrefix("!MovingAverageArray {\n" +
                "  values: [\n" +
                "    { movingAverage: 0.1, high: 0.1, low: 0.1, stdDev: 0.1 },\n" +
                "    { movingAverage: 1.1, high: 1.1, low: 1.1, stdDev: 1.1 },\n" +
                "    { movingAverage: 2.1, high: 2.1, low: 2.1, stdDev: 2.1 },\n" +
                "    { movingAverage: 3.1, high: 3.1, low: 3.1, stdDev: 3.1 },\n" +
                "    { movingAverage: 4.1, high: 4.1, low: 4.1, stdDev: 4.1 },\n" +
                "    { movingAverage: 5.1, high: 5.1, low: 5.1, stdDev: 5.1 }\n" +
                "  ]\n" +
                "}\n", m.toString());
        MovingAverageArray m2 = mapRead.getUsing(2L, m);
        assertSame(m, m2); // object is recycled, so no objects are created.
        assertJSONEqualsAfterPrefix("!MovingAverageArray {\n" +
                "  values: [\n" +
                "    { movingAverage: 0.2, high: 0.2, low: 0.2, stdDev: 0.2 },\n" +
                "    { movingAverage: 1.2, high: 1.2, low: 1.2, stdDev: 1.2 },\n" +
                "    { movingAverage: 2.2, high: 2.2, low: 2.2, stdDev: 2.2 },\n" +
                "    { movingAverage: 3.2, high: 3.2, low: 3.2, stdDev: 3.2 }\n" +
                "  ]\n" +
                "}\n", m.toString());
    }

    private MovingAverageArray createSampleWithSize(int size, int seed) {
        MovingAverageArray sample = new MovingAverageArray();
        for (int i = 0; i < size; i++) {
            double value = i + seed / 10.0;
            sample.add(new MovingAverageCompact(value, value, value, value));
        }
        return sample;
    }

    private void assertJSONEqualsAfterPrefix(String s1, String s2) throws JSONException {
        String[] a1 = s1.split(" ", 2);
        String[] a2 = s2.split(" ", 2);
        assertEquals(a1[0], a2[0]);
        JSONAssert.assertEquals(a1[1], a2[1], false);
    }
}
