ElastiCacheRaiderImpl implements ElastiCacheRaider { @Override public void rebootElastiCacheCluster(String clusterName, Integer numNodes) { RebootCacheClusterRequest rebootCacheClusterRequest = new RebootCacheClusterRequest() .withCacheClusterId(clusterName) .withCacheNodeIdsToReboot(this.getCachedNodeId(numNodes)); try { elastiCacheDelegator.getAmazonElastiCache().rebootCacheCluster(rebootCacheClusterRequest); } catch (Exception ex) { throw new RuntimeException(ex.getMessage()); } } ElastiCacheRaiderImpl(); @Override void rebootElastiCacheCluster(String clusterName, Integer numNodes); @Override void detachSubnet(String clusterName, String subnetId); @Override void attachSubnet(String clusterName, String subnetId); @Override String getElastiCacheClusterStatus(String clusterName); @Override void addNodesToCluster(String clusterName, Integer numNodes); Integer getNumNodes(String clusterName); @Override void removeNodesFromCluster(String clusterName, Integer numNodes); List<String> getSubnetIds(String clusterName); @Override void changeSecurityGroupId(String clusterName, String fromSecurityGroupId, String toSecurityGroupId); @Override List<String> getSecurityGroups(String clusterName); List<CacheNode> getAllElasticCacheNodes(); List<CacheNode> getAllElasticCacheNodesForGivenCluster(String clusterName); @Override List<String> getAllElastiCacheNodeNames(); @Override List<String> getAllElastiCacheNodeNames(String clusterName); @Override String getElastiCacheNodeStatus(String nodeName); @Override void rebootElasticCacheNode(String nodeName);  }
@Test public void testRebootElastiCacheCluster(){ PowerMockito.when(amazonElastiCache.rebootCacheCluster(Mockito.anyObject())).thenReturn(cacheCluster); elastiCacheRaider.rebootElastiCacheCluster("cacheCluster", 1); }