hazelcast-master/hazelcast-spring-tests/src/test/java/com/hazelcast/spring/cache/TestCacheManager.java
/*
 * Copyright (c) 2008-2023, 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.spring.cache;

import com.hazelcast.config.Config;
import com.hazelcast.core.DistributedObject;
import com.hazelcast.core.DistributedObjectEvent;
import com.hazelcast.core.DistributedObjectListener;
import com.hazelcast.core.Hazelcast;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.map.IMap;
import com.hazelcast.spring.CustomSpringJUnit4ClassRunner;
import com.hazelcast.test.HazelcastTestSupport;
import com.hazelcast.test.annotation.QuickTest;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cache.CacheManager;
import org.springframework.test.context.ContextConfiguration;

import javax.annotation.Resource;
import java.util.Collection;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicBoolean;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.fail;

@RunWith(CustomSpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"cacheManager-applicationContext-hazelcast.xml"})
@Category(QuickTest.class)
public class TestCacheManager extends HazelcastTestSupport {

    @Resource(name = "instance")
    private HazelcastInstance instance;

    @Autowired
    private IDummyBean bean;

    @Autowired
    private CacheManager cacheManager;

    @BeforeClass
    @AfterClass
    public static void start() {
        Hazelcast.shutdownAll();
    }

    @Test
    public void testBean_withValue() {
        for (int i = 0; i < 100; i++) {
            assertEquals("name:" + i, bean.getName(i));
            assertEquals("city:" + i, bean.getCity(i));
        }
    }

    @Test
    public void testBean_withNull() {
        for (int i = 0; i < 100; i++) {
            assertNull(bean.getNull());
        }
    }

    @Test
    public void testBean_withTTL() {
        String name = bean.getNameWithTTL();
        assertEquals("ali", name);
        String nameFromCache = bean.getNameWithTTL();
        assertEquals("ali", nameFromCache);

        sleepSeconds(3);

        String nameFromCacheAfterTTL = bean.getNameWithTTL();
        assertNull(nameFromCacheAfterTTL);
    }

    @Test
    public void testCacheNames() {
        // create a test instance, to reproduce the behavior described in the GitHub issue
        // https://github.com/hazelcast/hazelcast/issues/492
        final String testMap = "test-map";

        final CountDownLatch distributionSignal = new CountDownLatch(1);
        instance.addDistributedObjectListener(new DistributedObjectListener() {
            @Override
            public void distributedObjectCreated(DistributedObjectEvent event) {
                DistributedObject distributedObject = event.getDistributedObject();
                if (distributedObject instanceof IMap) {
                    IMap<?, ?> map = (IMap) distributedObject;
                    if (testMap.equals(map.getName())) {
                        distributionSignal.countDown();
                    }
                }
            }

            @Override
            public void distributedObjectDestroyed(DistributedObjectEvent event) {
            }
        });

        HazelcastInstance testInstance = Hazelcast.newHazelcastInstance(extractConfig());
        testInstance.getMap(testMap);
        // be sure that test-map is distributed
        HazelcastTestSupport.assertOpenEventually(distributionSignal);

        Collection<String> test = cacheManager.getCacheNames();
        assertContains(test, testMap);
        testInstance.shutdown();
        // Wait for the cluster to scale down, so it doesn't affect other tests
        assertClusterSizeEventually(1, instance);
    }

    public static class DummyBean implements IDummyBean {

        final AtomicBoolean nullCall = new AtomicBoolean(false);
        final AtomicBoolean firstCall = new AtomicBoolean(false);

        @Override
        public String getName(int k) {
            fail("should not call this method!");
            return null;
        }

        @Override
        public String getCity(int k) {
            fail("should not call this method!");
            return null;
        }

        @Override
        public Object getNull() {
            if (nullCall.compareAndSet(false, true)) {
                return null;
            }
            fail("should not call this method!");
            return null;
        }

        @Override
        public String getNameWithTTL() {
            if (firstCall.compareAndSet(false, true)) {
                return "ali";
            }
            return null;
        }
    }

    private Config extractConfig() {
        Config config = instance.getConfig();
        Config extractedConfig = new Config();
        extractedConfig
                .setProperties(config.getProperties())
                .setClusterName(config.getClusterName())
                .setNetworkConfig(config.getNetworkConfig())
                .setJetConfig(config.getJetConfig())
                .setSqlConfig(config.getSqlConfig());

        return extractedConfig;
    }
}
