Implement deletePrimitive method for mock primitive registry in Raft …

…test.
atomix · Aug 15, 2018 · 0884b1c · 0884b1c
1 parent 3a35842
commit 0884b1c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/protocols/raft/src/test/java/io/atomix/protocols/raft/RaftTest.java b/protocols/raft/src/test/java/io/atomix/protocols/raft/RaftTest.java
@@ -1332,6 +1332,7 @@ private TestPrimitive createPrimitive(RaftClient client, ReadConsistency consist
         (key, partitions) -> partitions.get(0));
     PrimitiveRegistry registry = mock(PrimitiveRegistry.class);
     when(registry.createPrimitive(any(String.class), any(PrimitiveType.class))).thenReturn(CompletableFuture.completedFuture(new PrimitiveInfo("raft-test", TestPrimitiveType.INSTANCE)));
+    when(registry.deletePrimitive(any(String.class))).thenReturn(CompletableFuture.completedFuture(null));
     return new TestPrimitiveImpl(proxy, registry);
   }