MapMBeanTest.java

/*
 * Copyright (c) 2008-2017, Hazelcast, Inc. All Rights Reserved.
 *
 * 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 com.hazelcast.internal.jmx;

import com.hazelcast.core.IMap;
import com.hazelcast.test.HazelcastParallelClassRunner;
import com.hazelcast.test.HazelcastTestSupport;
import com.hazelcast.test.TestHazelcastInstanceFactory;
import com.hazelcast.test.annotation.ParallelTest;
import com.hazelcast.test.annotation.QuickTest;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

@RunWith(HazelcastParallelClassRunner.class)
@Category({QuickTest.class, ParallelTest.class})
public class MapMBeanTest extends HazelcastTestSupport {

    private static final String TYPE_NAME = "IMap";
    private static final Object[] EMPTY_STRING_PARAMETER = {""};

    private TestHazelcastInstanceFactory hazelcastInstanceFactory = createHazelcastInstanceFactory(1);
    private MBeanDataHolder holder = new MBeanDataHolder(hazelcastInstanceFactory);

    private long started;
    private IMap<String, String> map;
    private String objectName;

    @Before
    public void setUp() {
        started = System.currentTimeMillis();
        map = holder.getHz().getMap("distributedMap");
        objectName = map.getName();

        holder.assertMBeanExistEventually(TYPE_NAME, map.getName());
    }

    @After
    public void tearDown() {
        hazelcastInstanceFactory.shutdownAll();
    }

    @Test
    public void testName() throws Exception {
        String name = getStringAttribute("name");

        assertEquals("distributedMap", name);
    }

    @Test
    public void testConfig() throws Exception {
        String config = getStringAttribute("config");

        assertTrue("configuration string should start with 'MapConfig{'", config.startsWith("MapConfig{"));
    }

    @Test
    public void testAttributesAndOperations() throws Exception {
        map.put("firstKey", "firstValue");
        map.put("secondKey", "secondValue");
        map.remove("secondKey");
        String value = map.get("firstKey");
        String values = invokeMethod("values", EMPTY_STRING_PARAMETER);
        String entries = invokeMethod("entrySet", EMPTY_STRING_PARAMETER);

        long localEntryCount = getLongAttribute("localOwnedEntryCount");
        long localBackupEntryCount = getLongAttribute("localBackupEntryCount");
        int localBackupCount = getIntegerAttribute("localBackupCount");
        long localOwnedEntryMemoryCost = getLongAttribute("localOwnedEntryMemoryCost");
        long localBackupEntryMemoryCost = getLongAttribute("localBackupEntryMemoryCost");

        long localCreationTime = getLongAttribute("localCreationTime");
        long localLastAccessTime = getLongAttribute("localLastAccessTime");
        long localLastUpdateTime = getLongAttribute("localLastUpdateTime");
        long localHits = getLongAttribute("localHits");
        long localLockedEntryCount = getLongAttribute("localLockedEntryCount");
        long localDirtyEntryCount = getLongAttribute("localDirtyEntryCount");

        long localPutOperationCount = getLongAttribute("localPutOperationCount");
        long localGetOperationCount = getLongAttribute("localGetOperationCount");
        long localRemoveOperationCount = getLongAttribute("localRemoveOperationCount");

        long localTotalPutLatency = getLongAttribute("localTotalPutLatency");
        long localTotalGetLatency = getLongAttribute("localTotalGetLatency");
        long localTotalRemoveLatency = getLongAttribute("localTotalRemoveLatency");
        long localMaxPutLatency = getLongAttribute("localMaxPutLatency");
        long localMaxGetLatency = getLongAttribute("localMaxGetLatency");
        long localMaxRemoveLatency = getLongAttribute("localMaxRemoveLatency");

        long localEventOperationCount = getLongAttribute("localEventOperationCount");
        long localOtherOperationCount = getLongAttribute("localOtherOperationCount");
        long localTotal = getLongAttribute("localTotal");
        long localTotalHeapCost = getLongAttribute("localHeapCost");
        int size = getIntegerAttribute("size");

        assertEquals("firstValue", value);
        assertEquals("[firstValue,]", values);
        assertEquals("[{key:firstKey, value:firstValue},]", entries);

        assertEquals(1, localEntryCount);
        assertEquals(0, localBackupEntryCount);
        assertEquals(1, localBackupCount);
        assertTrue(localOwnedEntryMemoryCost > 0);
        assertEquals(0, localBackupEntryMemoryCost);

        assertTrue(localCreationTime >= started);
        assertTrue(localLastAccessTime >= started);
        assertTrue(localLastUpdateTime >= started);
        assertEquals(1, localHits);
        assertEquals(0, localLockedEntryCount);
        assertEquals(0, localDirtyEntryCount);

        assertEquals(2, localPutOperationCount);
        assertEquals(1, localGetOperationCount);
        assertEquals(1, localRemoveOperationCount);

        assertTrue("localTotalPutLatency should be >= 0", localTotalPutLatency >= 0);
        assertTrue("localTotalGetLatency should be >= 0", localTotalGetLatency >= 0);
        assertTrue("localTotalRemoveLatency should be >= 0", localTotalRemoveLatency >= 0);
        assertTrue("localMaxPutLatency should be >= 0", localMaxPutLatency >= 0);
        assertTrue("localMaxGetLatency should be >= 0", localMaxGetLatency >= 0);
        assertTrue("localMaxRemoveLatency should be >= 0", localMaxRemoveLatency >= 0);

        assertEquals(0, localEventOperationCount);
        assertTrue("localOtherOperationCount should be > 0", localOtherOperationCount > 0);
        assertTrue("localTotal should be > 0", localTotal > 0);
        assertTrue("localTotalHeapCost should be > 0", localTotalHeapCost > 0);
        assertEquals(1, size);

        holder.invokeMBeanOperation(TYPE_NAME, objectName, "clear", null, null);
        values = invokeMethod("values", EMPTY_STRING_PARAMETER);
        entries = invokeMethod("entrySet", EMPTY_STRING_PARAMETER);
        size = getIntegerAttribute("size");

        assertEquals("Empty", values);
        assertEquals("Empty", entries);
        assertEquals(0, size);
    }

    private String getStringAttribute(String name) throws Exception {
        return (String) holder.getMBeanAttribute(TYPE_NAME, objectName, name);
    }

    private Long getLongAttribute(String name) throws Exception {
        return (Long) holder.getMBeanAttribute(TYPE_NAME, objectName, name);
    }

    private Integer getIntegerAttribute(String name) throws Exception {
        return (Integer) holder.getMBeanAttribute(TYPE_NAME, objectName, name);
    }

    private String invokeMethod(String methodName, Object[] params) throws Exception {
        return (String) holder.invokeMBeanOperation(TYPE_NAME, objectName, methodName, params, null);
    }
}