dschneider-pivotal commented on a change in pull request #6503:
URL: https://github.com/apache/geode/pull/6503#discussion_r637097514



##########
File path: 
geode-core/src/test/java/org/apache/geode/internal/cache/LatestLastAccessTimeMessageTest.java
##########
@@ -14,30 +14,142 @@
  */
 package org.apache.geode.internal.cache;
 
+import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.spy;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
 
 import java.util.Collections;
 import java.util.Set;
 
 import org.junit.Test;
 
 import org.apache.geode.distributed.internal.ClusterDistributionManager;
-import org.apache.geode.distributed.internal.DistributionManager;
 import 
org.apache.geode.distributed.internal.membership.InternalDistributedMember;
+import org.apache.geode.internal.InternalStatisticsDisabledException;
 import org.apache.geode.internal.inet.LocalHostUtil;
 
 public class LatestLastAccessTimeMessageTest {
 
   @Test
-  public void processMessageShouldLookForNullCache() throws Exception {
-    final DistributionManager distributionManager = 
mock(DistributionManager.class);
+  public void processWithNullCacheRepliesZero() throws Exception {
     final LatestLastAccessTimeReplyProcessor replyProcessor =
         mock(LatestLastAccessTimeReplyProcessor.class);
     final InternalDistributedRegion region = 
mock(InternalDistributedRegion.class);
     Set<InternalDistributedMember> recipients = Collections.singleton(new 
InternalDistributedMember(
         LocalHostUtil.getLocalHost(), 1234));
+    final LatestLastAccessTimeMessage<String> realLastAccessTimeMessage =
+        new LatestLastAccessTimeMessage<>(replyProcessor, recipients, region, 
"foo");
+    final LatestLastAccessTimeMessage<String> lastAccessTimeMessage =
+        spy(realLastAccessTimeMessage);
+    lastAccessTimeMessage.setSender(new InternalDistributedMember(
+        LocalHostUtil.getLocalHost(), 12345));
+    final ClusterDistributionManager dm = 
mock(ClusterDistributionManager.class);
+    when(dm.getCache()).thenReturn(null);
+
+    lastAccessTimeMessage.process(dm);
+
+    verify(lastAccessTimeMessage).sendReply(dm, 0);
+  }
+
+  @Test
+  public void processWithNullRegionRepliesZero() throws Exception {
+    final LatestLastAccessTimeReplyProcessor replyProcessor =
+        mock(LatestLastAccessTimeReplyProcessor.class);
+    final InternalDistributedRegion region = 
mock(InternalDistributedRegion.class);
+    Set<InternalDistributedMember> recipients = Collections.singleton(new 
InternalDistributedMember(
+        LocalHostUtil.getLocalHost(), 1234));
+    final LatestLastAccessTimeMessage<String> realLastAccessTimeMessage =
+        new LatestLastAccessTimeMessage<>(replyProcessor, recipients, region, 
"foo");
+    final LatestLastAccessTimeMessage<String> lastAccessTimeMessage =
+        spy(realLastAccessTimeMessage);
+    lastAccessTimeMessage.setSender(new InternalDistributedMember(
+        LocalHostUtil.getLocalHost(), 12345));
+    final ClusterDistributionManager dm = 
mock(ClusterDistributionManager.class);
+    final InternalCache cache = mock(InternalCache.class);

Review comment:
       Thanks for the feedback Donal. I have updates the PR with your suggested 
test cleanup.




-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
[email protected]


Reply via email to