package org.opendaylight.controller.datastore.internal; import org.junit.Before; import org.junit.Test; import org.mockito.Mockito; import org.opendaylight.controller.clustering.services.CacheConfigException; import org.opendaylight.controller.clustering.services.CacheExistException; import org.opendaylight.controller.clustering.services.IClusterGlobalServices; import org.opendaylight.controller.clustering.services.IClusterServices; import org.opendaylight.controller.md.sal.common.api.data.DataCommitHandler; import org.opendaylight.controller.md.sal.common.api.data.DataModification; import org.opendaylight.yangtools.yang.binding.InstanceIdentifier; import java.util.EnumSet; import java.util.Map; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ConcurrentMap; import static junit.framework.Assert.assertEquals; import static junit.framework.Assert.assertNotNull; import static junit.framework.Assert.assertNull; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.never; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; public class ClusteredDataStoreImplTest { @Before public void setUp(){ } @Test public void constructor_WhenPassedANullClusteringServices_ShouldThrowANullPointerException() throws CacheExistException, CacheConfigException { try { new ClusteredDataStoreImpl(null); } catch(NullPointerException npe){ assertEquals("clusterGlobalServices cannot be null", npe.getMessage()); } } @Test public void constructor_WhenClusteringServicesReturnsANullOperationalDataCache_ShouldThrowANullPointerException() throws CacheExistException, CacheConfigException { try { new ClusteredDataStoreImpl(mock(IClusterGlobalServices.class)); } catch(NullPointerException npe){ assertEquals("operationalDataCache cannot be null", npe.getMessage()); } } @Test public void constructor_WhenClusteringServicesReturnsANullOConfigurationDataCache_ShouldThrowANullPointerException() throws CacheExistException, CacheConfigException { IClusterGlobalServices mockClusterGlobalServices = mock(IClusterGlobalServices.class); // Confused about the following line? // See this http://stackoverflow.com/questions/10952629/a-strange-generics-edge-case-with-mockito-when-and-generic-type-inference Mockito.>when(mockClusterGlobalServices.createCache(ClusteredDataStoreImpl.OPERATIONAL_DATA_CACHE, EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL))).thenReturn(new ConcurrentHashMap()); try { new ClusteredDataStoreImpl(mockClusterGlobalServices); } catch(NullPointerException npe){ assertEquals("configurationDataCache cannot be null", npe.getMessage()); } } @Test public void constructor_WhenPassedAValidClusteringServices_ShouldNotThrowAnyExceptions() throws CacheExistException, CacheConfigException { IClusterGlobalServices mockClusterGlobalServices = createClusterGlobalServices(); new ClusteredDataStoreImpl(mockClusterGlobalServices); } @Test public void readOperationalData_WhenPassedANullPath_ShouldThrowANullPointerException() throws CacheExistException, CacheConfigException { IClusterGlobalServices mockClusterGlobalServices = createClusterGlobalServices(); ClusteredDataStoreImpl store = new ClusteredDataStoreImpl(mockClusterGlobalServices); try { store.readOperationalData(null); } catch(NullPointerException npe){ assertEquals("path cannot be null", npe.getMessage()); } } @Test public void readOperationalData_WhenPassedAKeyThatDoesNotExistInTheCache_ShouldReturnNull() throws CacheExistException, CacheConfigException { InstanceIdentifier path = InstanceIdentifier.builder().toInstance(); IClusterGlobalServices mockClusterGlobalServices = createClusterGlobalServices(); ClusteredDataStoreImpl store = new ClusteredDataStoreImpl(mockClusterGlobalServices); assertNull(store.readOperationalData(path)); } @Test public void readOperationalData_WhenPassedAKeyThatDoesExistInTheCache_ShouldReturnTheValueObject() throws CacheExistException, CacheConfigException { InstanceIdentifier path = InstanceIdentifier.builder().toInstance(); IClusterGlobalServices mockClusterGlobalServices = createClusterGlobalServices(); ConcurrentMap mockOperationalDataCache = mock(ConcurrentMap.class); Object valueObject = mock(Object.class); when(mockOperationalDataCache.get(path)).thenReturn(valueObject); Mockito.>when(mockClusterGlobalServices.createCache(ClusteredDataStoreImpl.OPERATIONAL_DATA_CACHE, EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL))).thenReturn(mockOperationalDataCache); Mockito.>when(mockClusterGlobalServices.createCache(ClusteredDataStoreImpl.CONFIGURATION_DATA_CACHE, EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL))).thenReturn(new ConcurrentHashMap()); ClusteredDataStoreImpl store = new ClusteredDataStoreImpl(mockClusterGlobalServices); assertEquals(valueObject, store.readOperationalData(path)); } @Test public void readConfigurationData_WhenPassedANullPath_ShouldThrowANullPointerException() throws CacheExistException, CacheConfigException { IClusterGlobalServices mockClusterGlobalServices = createClusterGlobalServices(); ClusteredDataStoreImpl store = new ClusteredDataStoreImpl(mockClusterGlobalServices); try { store.readConfigurationData(null); } catch(NullPointerException npe){ assertEquals("path cannot be null", npe.getMessage()); } } @Test public void readConfigurationData_WhenPassedAKeyThatDoesNotExistInTheCache_ShouldReturnNull() throws CacheExistException, CacheConfigException { InstanceIdentifier path = InstanceIdentifier.builder().toInstance(); IClusterGlobalServices mockClusterGlobalServices = createClusterGlobalServices(); ClusteredDataStoreImpl store = new ClusteredDataStoreImpl(mockClusterGlobalServices); assertNull(store.readConfigurationData(path)); } @Test public void readConfigurationData_WhenPassedAKeyThatDoesExistInTheCache_ShouldReturnTheValueObject() throws CacheExistException, CacheConfigException { InstanceIdentifier path = InstanceIdentifier.builder().toInstance(); IClusterGlobalServices mockClusterGlobalServices = createClusterGlobalServices(); ConcurrentMap mockConfigurationDataCache = mock(ConcurrentMap.class); Object valueObject = mock(Object.class); when(mockConfigurationDataCache.get(path)).thenReturn(valueObject); Mockito.>when(mockClusterGlobalServices.createCache(ClusteredDataStoreImpl.OPERATIONAL_DATA_CACHE, EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL))).thenReturn(mock(ConcurrentMap.class)); Mockito.>when(mockClusterGlobalServices.createCache(ClusteredDataStoreImpl.CONFIGURATION_DATA_CACHE, EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL))).thenReturn(mockConfigurationDataCache); ClusteredDataStoreImpl store = new ClusteredDataStoreImpl(mockClusterGlobalServices); assertEquals(valueObject, store.readConfigurationData(path)); } @Test public void requestCommit_ShouldReturnADataTransaction() throws CacheExistException, CacheConfigException { IClusterGlobalServices mockClusterGlobalServices = createClusterGlobalServices(); ClusteredDataStoreImpl store = new ClusteredDataStoreImpl(mockClusterGlobalServices); assertNotNull(store.requestCommit(mock(DataModification.class))); } @Test public void finishingADataTransaction_ShouldUpdateTheUnderlyingCache() throws CacheExistException, CacheConfigException { IClusterGlobalServices mockClusterGlobalServices = mock(IClusterGlobalServices.class); ConcurrentMap mockConfigurationDataCache = mock(ConcurrentMap.class); ConcurrentMap mockOperationalDataCache = mock(ConcurrentMap.class); Mockito.>when(mockClusterGlobalServices.createCache(ClusteredDataStoreImpl.OPERATIONAL_DATA_CACHE, EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL))).thenReturn(mockOperationalDataCache); Mockito.>when(mockClusterGlobalServices.createCache(ClusteredDataStoreImpl.CONFIGURATION_DATA_CACHE, EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL))).thenReturn(mockConfigurationDataCache); ClusteredDataStoreImpl store = new ClusteredDataStoreImpl(mockClusterGlobalServices); DataModification mockModification = mock(DataModification.class); Map configurationData = mock(Map.class); Map operationalData = mock(Map.class); when(mockModification.getUpdatedConfigurationData()).thenReturn(configurationData); when(mockModification.getUpdatedOperationalData()).thenReturn(operationalData); DataCommitHandler.DataCommitTransaction, Object> transaction = store.requestCommit(mockModification); transaction.finish(); verify(mockConfigurationDataCache).putAll(mockModification.getUpdatedConfigurationData()); verify(mockOperationalDataCache).putAll(mockModification.getUpdatedOperationalData()); } @Test public void rollingBackADataTransaction_ShouldDoNothing() throws CacheExistException, CacheConfigException { IClusterGlobalServices mockClusterGlobalServices = mock(IClusterGlobalServices.class); ConcurrentMap mockConfigurationDataCache = mock(ConcurrentMap.class); ConcurrentMap mockOperationalDataCache = mock(ConcurrentMap.class); Mockito.>when(mockClusterGlobalServices.createCache(ClusteredDataStoreImpl.OPERATIONAL_DATA_CACHE, EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL))).thenReturn(mockOperationalDataCache); Mockito.>when(mockClusterGlobalServices.createCache(ClusteredDataStoreImpl.CONFIGURATION_DATA_CACHE, EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL))).thenReturn(mockConfigurationDataCache); ClusteredDataStoreImpl store = new ClusteredDataStoreImpl(mockClusterGlobalServices); DataModification mockModification = mock(DataModification.class); Map configurationData = mock(Map.class); Map operationalData = mock(Map.class); when(mockModification.getUpdatedConfigurationData()).thenReturn(configurationData); when(mockModification.getUpdatedOperationalData()).thenReturn(operationalData); DataCommitHandler.DataCommitTransaction, Object> transaction = store.requestCommit(mockModification); transaction.rollback(); verify(mockConfigurationDataCache, never()).putAll(mockModification.getUpdatedConfigurationData()); verify(mockOperationalDataCache, never()).putAll(mockModification.getUpdatedOperationalData()); } private IClusterGlobalServices createClusterGlobalServices() throws CacheExistException, CacheConfigException { IClusterGlobalServices mockClusterGlobalServices = mock(IClusterGlobalServices.class); Mockito.>when(mockClusterGlobalServices.createCache(ClusteredDataStoreImpl.OPERATIONAL_DATA_CACHE, EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL))).thenReturn(mock(ConcurrentMap.class)); Mockito.>when(mockClusterGlobalServices.createCache(ClusteredDataStoreImpl.CONFIGURATION_DATA_CACHE, EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL))).thenReturn(mock(ConcurrentMap.class)); return mockClusterGlobalServices; } }