/** * @param spi SPI. * @param cacheName Cache name. */ private void blockSupplySend(TestRecordingCommunicationSpi spi, final String cacheName) { final int grpId = groupIdForCache(spi.ignite(), cacheName); spi.blockMessages(new IgniteBiPredicate<ClusterNode, Message>() { @Override public boolean apply(ClusterNode node, Message msg) { if (!msg.getClass().equals(GridDhtPartitionSupplyMessage.class)) return false; return ((GridDhtPartitionSupplyMessage)msg).groupId() == grpId; } }); }