Spring/spring-test-data-geode
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Implement additional Region data access operations.

Browse Source 
Implement Region.invalidate(key).

Implement Region.remove(key).

Re-implement Region.get(key), Region.getEntry(key) and Region.put(key) to handle invalidated keys (Region entries).
This commit is contained in:
John Blum
2019-06-14 14:37:11 -07:00
parent 6f4367e963
commit 0b01f6b240
2 changed files with 80 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/mock/GemFireMockObjectsSupport.java
View File

@@ -78,6 +78,7 @@ import org.apache.geode.cache.CustomExpiry;
import org.apache.geode.cache.DataPolicy;
import org.apache.geode.cache.DiskStore;
import org.apache.geode.cache.DiskStoreFactory;
import org.apache.geode.cache.EntryNotFoundException;
import org.apache.geode.cache.EvictionAttributes;
import org.apache.geode.cache.EvictionAttributesMutator;
import org.apache.geode.cache.ExpirationAction;
@@ -2171,6 +2172,7 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport {
RegionAttributes<K, V> mockRegionAttributes = mockRegionAttributes(mockRegion, regionAttributes);
Set<K> invalidatedKeys = new HashSet<>();
Set<Region<?, ?>> subRegions = new CopyOnWriteArraySet<>();
when(mockRegion.getAttributes()).thenReturn(mockRegionAttributes);
@@ -2186,16 +2188,80 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport {
return regions.get(subRegionFullPath);
});
when(mockRegion.get(ArgumentMatchers.<K>any())).thenAnswer(invocation ->
data.get(invocation.<K>getArgument(0)));
// Region.containsKey(key)
when(mockRegion.containsKey(any())).thenAnswer(invocation ->
data.containsKey(invocation.getArgument(0)));
when(mockRegion.getEntry(ArgumentMatchers.<K>any())).thenAnswer(invocation ->
data.entrySet().stream().filter(entry -> entry.getKey().equals(invocation.getArgument(0))).findFirst()
// Region.get(key)
when(mockRegion.get(ArgumentMatchers.<K>any())).thenAnswer(invocation -> {
K key = invocation.getArgument(0);
V value = data.get(key);
if (invalidatedKeys.contains(key)) {
value = null;
}
// TODO: CacheLoader
Optional.ofNullable(value);
return value;
});
// Region.getEntry(key)
when(mockRegion.getEntry(ArgumentMatchers.<K>any())).thenAnswer(regionGetEntryInvocation ->
data.entrySet().stream()
.filter(entry -> entry.getKey().equals(regionGetEntryInvocation.getArgument(0)))
.findFirst()
.map(entry -> {
Map.Entry<K, V> entrySpy = spy(entry);
doAnswer(entryGetValueInvocation ->
invalidatedKeys.contains(entry.getKey()) ? null : entry.getValue())
.when(entrySpy).getValue();
return entrySpy;
})
.orElse(null));
when(mockRegion.put(any(), any())).thenAnswer(invocation ->
data.put(invocation.getArgument(0), invocation.getArgument(1)));
// Region.invalidate(key)
doAnswer(invocation -> {
K key = invocation.getArgument(0);
if (data.containsKey(key)) {
invalidatedKeys.add(key);
}
else {
throw new EntryNotFoundException(String.format("Entry with key [%s] not found", key));
}
return null;
}).when(mockRegion).invalidate(any());
// Region.put(key, value)
when(mockRegion.put(any(), any())).thenAnswer(invocation -> {
K key = invocation.getArgument(0);
V newValue = invocation.getArgument(1);
V existingValue = data.put(key, newValue);
return invalidatedKeys.remove(key) ? null : existingValue;
});
// Region.remove(key)
when(mockRegion.remove(any())).thenAnswer(invocation -> {
K key = invocation.getArgument(0);
V value = data.remove(key);
return invalidatedKeys.remove(key) ? null : value;
});
// Region.size()
when(mockRegion.size()).thenAnswer(invocation -> data.size());
when(mockRegion.subregions(anyBoolean())).thenAnswer(invocation -> {

8
spring-data-geode-test/src/test/java/org/springframework/data/gemfire/MockClientCacheApplicationIntegrationTests.java
View File

@@ -89,6 +89,14 @@ public class MockClientCacheApplicationIntegrationTests {
assertThat(this.example.getName()).isEqualTo("Example");
assertThat(this.example.put(1, "test")).isNull();
assertThat(this.example.get(1)).isEqualTo("test");
assertThat(this.example.containsKey(1)).isTrue();
this.example.invalidate(1);
assertThat(this.example.containsKey(1)).isTrue();
assertThat(this.example.get(1)).isNull();
assertThat(this.example.remove(1)).isNull();
assertThat(this.example.containsKey(1)).isFalse();
}
@ClientCacheApplication