/*
 * Copyright 2012-2017 Red Hat, Inc.
 *
 * This file is part of Thermostat.
 *
 * Thermostat is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published
 * by the Free Software Foundation; either version 2, or (at your
 * option) any later version.
 *
 * Thermostat is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Thermostat; see the file COPYING.  If not see
 * <http://www.gnu.org/licenses/>.
 *
 * Linking this code with other modules is making a combined work
 * based on this code.  Thus, the terms and conditions of the GNU
 * General Public License cover the whole combination.
 *
 * As a special exception, the copyright holders of this code give
 * you permission to link this code with independent modules to
 * produce an executable, regardless of the license terms of these
 * independent modules, and to copy and distribute the resulting
 * executable under terms of your choice, provided that you also
 * meet, for each linked independent module, the terms and conditions
 * of the license of that module.  An independent module is a module
 * which is not derived from or based on this code.  If you modify
 * this code, you may extend this exception to your version of the
 * library, but you are not obligated to do so.  If you do not wish
 * to do so, delete this exception statement from your version.
 */

package com.redhat.thermostat.vm.byteman.agent.internal;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.nio.ByteBuffer;
import java.nio.charset.Charset;
import java.util.List;
import java.util.Map;

import org.junit.Test;
import org.mockito.ArgumentCaptor;

import com.redhat.thermostat.agent.ipc.server.IPCMessage;
import com.redhat.thermostat.vm.byteman.agent.BytemanMetric;
import com.redhat.thermostat.vm.byteman.agent.VmBytemanDAO;
import com.redhat.thermostat.vm.byteman.agent.internal.typeadapter.JsonHelper;

public class BytemanMetricsReceiverTest {

    private static final int SOME_PID = 23;

    @Test
    public void canSendDataToStorage() {
        VmBytemanDAO dao = mock(VmBytemanDAO.class);
        ArgumentCaptor<BytemanMetric> metricsCaptor = ArgumentCaptor.forClass(BytemanMetric.class);
        VmSocketIdentifier sockId = new VmSocketIdentifier("vm-id", SOME_PID, "agent-id");
        BytemanMetricsReceiver receiver = new BytemanMetricsReceiver(dao, sockId);
        String jsonString = JsonHelper.buildJsonArray(3);
        ByteBuffer data = Charset.forName("UTF-8").encode(jsonString);
        IPCMessage message = mock(IPCMessage.class);
        when(message.get()).thenReturn(data);
        receiver.messageReceived(message);
        verify(dao, times(3)).addMetric(metricsCaptor.capture());
        List<BytemanMetric> metrics = metricsCaptor.getAllValues();
        assertEquals("vm-id", metrics.get(0).getJvmId());
        assertEquals("agent-id", metrics.get(2).getAgentId());
        assertTrue(metrics.get(1).getTimeStamp() > 0);
        assertEquals("baz0", metrics.get(0).getMarker());
        assertNotNull(metrics.get(2).getDataAsJson());
        Map<String, Object> payloadAsMap = metrics.get(2).getDataAsMap();
        assertEquals("Expected 7 keys, including one with null value", 7, payloadAsMap.keySet().size());
    }
    
    /*
     * There was an issue where the metric data json became HTML escaped:
     * 
     * before JSON convert: 'count = foo'
     * after JSON convert:  'count \u003d foo'
     * 
     * This test is supposed to catch that data won't get HTML escaped.
     */
    @Test
    public void canSendDataToStorageUtf8() {
        VmBytemanDAO dao = mock(VmBytemanDAO.class);
        String jsonMetric = 
                "[ {\n" +
                        "    \"marker\": \"marker\",\n" +
                        "    \"timestamp\":\"30\",\n" +
                        "    \"data\": {\n" +
                        "        \"key\": \"value = foo\"\n" +
                        "    }\n" +
                        "} ]";
        ByteBuffer data = Charset.forName("UTF-8").encode(jsonMetric);
        IPCMessage message = mock(IPCMessage.class);
        when(message.get()).thenReturn(data);
        ArgumentCaptor<BytemanMetric> metricsCaptor = ArgumentCaptor.forClass(BytemanMetric.class);
        
        
        BytemanMetricsReceiver receiver = new BytemanMetricsReceiver(dao, mock(VmSocketIdentifier.class));
        receiver.messageReceived(message);
        
        verify(dao, times(1)).addMetric(metricsCaptor.capture());
        
        List<BytemanMetric> metrics = metricsCaptor.getAllValues();
        BytemanMetric metric = metrics.get(0);
        // make sure HTML chars are not escaped
        assertEquals("{\"key\":\"value = foo\"}", metric.getData());
        Map<String, Object> dataVals = metric.getDataAsMap();
        assertEquals("value = foo", dataVals.get("key"));
    }
    
}