org.flite.cach3.aop.ReadThroughMultiCacheTest.java Source code

Java tutorial

Introduction

Here is the source code for org.flite.cach3.aop.ReadThroughMultiCacheTest.java

Source

package org.flite.cach3.aop;

import org.apache.commons.lang.*;
import org.apache.commons.lang.math.*;
import org.flite.cach3.annotations.*;
import org.testng.annotations.*;

import java.security.*;
import java.util.*;

import static org.testng.AssertJUnit.*;

/**
Copyright (c) 2011-2012 Flite, Inc
    
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.
 */
public class ReadThroughMultiCacheTest {
    private ReadThroughMultiCacheAdvice cut;
    private ReadThroughMultiCacheAdvice.MultiCacheCoordinator coord;

    @BeforeClass
    public void beforeClass() {
        cut = new ReadThroughMultiCacheAdvice();
    }

    @BeforeMethod
    public void beforeMethod() {
        coord = new ReadThroughMultiCacheAdvice.MultiCacheCoordinator();
    }

    @Test
    public void testConvertIdObject() throws Exception {
        final String ns = RandomStringUtils.randomAlphanumeric(6);
        final Map<String, Object> expectedString2Object = new HashMap<String, Object>();
        final Map<Object, String> expectedObject2String = new HashMap<Object, String>();
        final List<Object> idObjects = new ArrayList<Object>();
        final int length = 10;
        for (int ix = 0; ix < length; ix++) {
            final String object = RandomStringUtils.randomAlphanumeric(2 + ix);
            final String key = cut.buildCacheKey(object, ns, null);
            idObjects.add(object);
            expectedObject2String.put(object, key);
            expectedString2Object.put(key, object);
        }

        final CacheKeyMethodStore store = new CacheKeyMethodStoreImpl();
        final List<Object> exceptionObjects = new ArrayList<Object>(idObjects);
        exceptionObjects.add(null);
        try {
            cut.convertIdObjectsToKeyMap(exceptionObjects, ns, null, null, null, store, null);
            fail("Expected Exception");
        } catch (InvalidParameterException ex) {
        }

        for (int ix = 0; ix < length; ix++) {
            if (ix % 2 == 0) {
                idObjects.add(idObjects.get(ix));
            }
        }
        assertTrue(idObjects.size() > length);

        final ReadThroughMultiCacheAdvice.MapHolder holder = cut.convertIdObjectsToKeyMap(idObjects, ns, null, null,
                null, store, null);

        assertEquals(length, holder.getKey2Obj().size());
        assertEquals(length, holder.getObj2Key().size());

        assertEquals(expectedObject2String, holder.getObj2Key());
        assertEquals(expectedString2Object, holder.getKey2Obj());

        coord.setHolder(holder);

        assertEquals(expectedObject2String, coord.getObj2Key());
        assertEquals(expectedString2Object, coord.getKey2Obj());

    }

    @Test
    public void testInitialKey2Result() {
        final String ns = RandomStringUtils.randomAlphanumeric(6);
        final Map<String, Object> expectedString2Object = new HashMap<String, Object>();
        final Map<String, Object> key2Result = new HashMap<String, Object>();
        final Set<Object> missObjects = new HashSet<Object>();
        final int length = 15;
        for (int ix = 0; ix < length; ix++) {

            final String object = RandomStringUtils.randomAlphanumeric(2 + ix);
            final String key = cut.buildCacheKey(object, ns, null);
            expectedString2Object.put(key, object);

            // There are 3 possible outcomes when fetching by key from memcached:
            // 0) You hit, and the key & result are in the map
            // 1) You hit, but the result is null, which counts as a miss.
            // 2) You miss, and the key doesn't even get into the result map.
            final int option = RandomUtils.nextInt(3);
            if (option == 0) {
                key2Result.put(key, key + RandomStringUtils.randomNumeric(5));
            }
            if (option == 1) {
                key2Result.put(key, null);
                missObjects.add(object);
            }
            if (option == 2) {
                missObjects.add(object);
            }
        }

        try {
            coord.setInitialKey2Result(null);
            fail("Expected Exception.");
        } catch (RuntimeException ex) {
        }

        coord.getKey2Obj().putAll(expectedString2Object);

        coord.setInitialKey2Result(key2Result);

        assertTrue(coord.getMissObjects().containsAll(missObjects));
        assertTrue(missObjects.containsAll(coord.getMissObjects()));
    }

    @Test
    public void testGenerateResultsException() {
        final List<Object> keyObjects = new ArrayList<Object>();
        final Map<Object, String> obj2key = new HashMap<Object, String>();
        final String keyObject = RandomStringUtils.randomAlphanumeric(8);
        final String key = keyObject + "-" + RandomStringUtils.randomAlphanumeric(4);
        keyObjects.add(keyObject);
        obj2key.put(keyObject, key);
        coord.setKeyObjects(keyObjects);
        coord.getObj2Key().putAll(obj2key);

        try {
            coord.generateResultList();
            fail("Expected Exception");
        } catch (RuntimeException ex) {
        }

    }

    @Test
    public void testGenerateResults() {
        final List<Object> keyObjects = new ArrayList<Object>();
        final Map<Object, String> obj2key = new HashMap<Object, String>();
        final Map<String, Object> key2result = new HashMap<String, Object>();
        final List<Object> expectedResults = new ArrayList<Object>();
        final int length = 10;
        for (int ix = 0; ix < length; ix++) {
            final String keyObject = RandomStringUtils.randomAlphanumeric(8);
            final String key = keyObject + "-" + RandomStringUtils.randomAlphanumeric(4);

            keyObjects.add(keyObject);
            obj2key.put(keyObject, key);

            if (RandomUtils.nextBoolean()) {
                final String result = RandomStringUtils.randomAlphanumeric(15);
                key2result.put(key, result);
                expectedResults.add(result);
            } else {
                key2result.put(key, new PertinentNegativeNull());
                expectedResults.add(null);
            }
        }

        coord.setKeyObjects(keyObjects);
        coord.getObj2Key().putAll(obj2key);
        coord.getKey2Result().putAll(key2result);

        final List<Object> results = coord.generateResultList();

        assertEquals(length, results.size());
        assertEquals(expectedResults, results);
    }

    private static class AnnotationValidator {
        @ReadThroughMultiCache(keyIndex = -1, namespace = "bubba")
        public String cacheMe1() {
            return null;
        }

        @ReadThroughMultiCache(keyIndex = 0, namespace = "")
        public String cacheMe2() {
            return null;
        }

        @ReadThroughMultiCache(keyIndex = 0)
        public String cacheMe3() {
            return null;
        }

        @ReadThroughMultiCache(keyIndex = 0, namespace = "bubba", expiration = -1)
        public String cacheMe4() {
            return null;
        }
    }
}