mirror of https://github.com/apache/jclouds.git
Unit test for VolumeDetached predicate
This commit is contained in:
parent
fd01759c8a
commit
8489cc4e05
|
@ -18,20 +18,20 @@
|
||||||
*/
|
*/
|
||||||
package org.jclouds.ec2.domain;
|
package org.jclouds.ec2.domain;
|
||||||
|
|
||||||
import static com.google.common.base.Preconditions.checkNotNull;
|
|
||||||
|
|
||||||
import java.util.Date;
|
import java.util.Date;
|
||||||
|
|
||||||
|
import static com.google.common.base.Preconditions.checkNotNull;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
*
|
* @author Adrian Cole
|
||||||
* @see <a href=
|
* @see <a href=
|
||||||
* "http://docs.amazonwebservices.com/AWSEC2/latest/APIReference/ApiReference-query-CreateVolume.html"
|
* "http://docs.amazonwebservices.com/AWSEC2/latest/APIReference/ApiReference-query-CreateVolume.html"
|
||||||
* />
|
* />
|
||||||
* @author Adrian Cole
|
|
||||||
*/
|
*/
|
||||||
public class Attachment implements Comparable<Attachment> {
|
public class Attachment implements Comparable<Attachment> {
|
||||||
public static enum Status {
|
public static enum Status {
|
||||||
ATTACHING, ATTACHED, DETACHING, DETACHED, BUSY, UNRECOGNIZED;
|
ATTACHING, ATTACHED, DETACHING, DETACHED, BUSY, UNRECOGNIZED;
|
||||||
|
|
||||||
public String value() {
|
public String value() {
|
||||||
return name().toLowerCase();
|
return name().toLowerCase();
|
||||||
}
|
}
|
||||||
|
@ -50,6 +50,53 @@ public class Attachment implements Comparable<Attachment> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static Builder builder() {
|
||||||
|
return new Builder();
|
||||||
|
}
|
||||||
|
|
||||||
|
public static class Builder {
|
||||||
|
private String region;
|
||||||
|
private String volumeId;
|
||||||
|
private String instanceId;
|
||||||
|
private String device;
|
||||||
|
private Status status;
|
||||||
|
private Date attachTime;
|
||||||
|
|
||||||
|
public Builder region(String region) {
|
||||||
|
this.region = region;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder volumeId(String volumeId) {
|
||||||
|
this.volumeId = volumeId;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder instanceId(String instanceId) {
|
||||||
|
this.instanceId = instanceId;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder device(String device) {
|
||||||
|
this.device = device;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder status(Status status) {
|
||||||
|
this.status = status;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder attachTime(Date attachTime) {
|
||||||
|
this.attachTime = attachTime;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Attachment build() {
|
||||||
|
return new Attachment(region, volumeId, instanceId, device, status, attachTime);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
private final String region;
|
private final String region;
|
||||||
private final String volumeId;
|
private final String volumeId;
|
||||||
private final String instanceId;
|
private final String instanceId;
|
||||||
|
@ -68,7 +115,6 @@ public class Attachment implements Comparable<Attachment> {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Snapshots are tied to Regions and can only be used for volumes within the same Region.
|
* Snapshots are tied to Regions and can only be used for volumes within the same Region.
|
||||||
*
|
|
||||||
*/
|
*/
|
||||||
public String getRegion() {
|
public String getRegion() {
|
||||||
return region;
|
return region;
|
||||||
|
@ -167,7 +213,7 @@ public class Attachment implements Comparable<Attachment> {
|
||||||
@Override
|
@Override
|
||||||
public String toString() {
|
public String toString() {
|
||||||
return "Attachment [region=" + region + ", volumeId=" + volumeId + ", instanceId=" + instanceId + ", device="
|
return "Attachment [region=" + region + ", volumeId=" + volumeId + ", instanceId=" + instanceId + ", device="
|
||||||
+ device + ", attachTime=" + attachTime + ", status=" + status + "]";
|
+ device + ", attachTime=" + attachTime + ", status=" + status + "]";
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
@ -18,23 +18,22 @@
|
||||||
*/
|
*/
|
||||||
package org.jclouds.ec2.domain;
|
package org.jclouds.ec2.domain;
|
||||||
|
|
||||||
import static com.google.common.base.Preconditions.checkNotNull;
|
import com.google.common.base.CaseFormat;
|
||||||
|
import com.google.common.collect.ImmutableSet;
|
||||||
|
import org.jclouds.javax.annotation.Nullable;
|
||||||
|
|
||||||
import java.util.Date;
|
import java.util.Date;
|
||||||
import java.util.Set;
|
import java.util.Set;
|
||||||
|
|
||||||
import org.jclouds.javax.annotation.Nullable;
|
import static com.google.common.base.Preconditions.checkNotNull;
|
||||||
|
import static com.google.common.collect.Sets.newHashSet;
|
||||||
import com.google.common.base.CaseFormat;
|
|
||||||
import com.google.common.collect.Iterables;
|
|
||||||
import com.google.common.collect.Sets;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
*
|
*
|
||||||
* @see <a href=
|
* @see <a href=
|
||||||
* "http://docs.amazonwebservices.com/AWSEC2/latest/APIReference/ApiReference-query-CreateVolume.html"
|
* "http://docs.amazonwebservices.com/AWSEC2/latest/APIReference/ApiReference-query-CreateVolume.html"
|
||||||
* />
|
* />
|
||||||
* @author Adrian Cole
|
* @author Adrian Cole, Andrei Savu
|
||||||
*/
|
*/
|
||||||
public class Volume implements Comparable<Volume> {
|
public class Volume implements Comparable<Volume> {
|
||||||
|
|
||||||
|
@ -83,6 +82,71 @@ public class Volume implements Comparable<Volume> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static Builder builder() {
|
||||||
|
return new Builder();
|
||||||
|
}
|
||||||
|
|
||||||
|
public static class Builder {
|
||||||
|
private String region;
|
||||||
|
private String id;
|
||||||
|
private int size;
|
||||||
|
@Nullable
|
||||||
|
private String snapshotId;
|
||||||
|
private String availabilityZone;
|
||||||
|
private Status status;
|
||||||
|
private Date createTime;
|
||||||
|
private Set<Attachment> attachments;
|
||||||
|
|
||||||
|
public Builder region(String region) {
|
||||||
|
this.region = region;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder id(String id) {
|
||||||
|
this.id = id;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder size(int size) {
|
||||||
|
this.size = size;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder snapshotId(String snapshotId) {
|
||||||
|
this.snapshotId = snapshotId;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder availabilityZone(String availabilityZone) {
|
||||||
|
this.availabilityZone = availabilityZone;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder status(Status status) {
|
||||||
|
this.status = status;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder createTime(Date createTime) {
|
||||||
|
this.createTime = createTime;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder attachments(Attachment... attachments) {
|
||||||
|
this.attachments = newHashSet(attachments);
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder attachments(Set<Attachment> attachments) {
|
||||||
|
this.attachments = ImmutableSet.copyOf(attachments);
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Volume build() {
|
||||||
|
return new Volume(region, id, size, snapshotId, availabilityZone, status, createTime, attachments);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
private final String region;
|
private final String region;
|
||||||
private final String id;
|
private final String id;
|
||||||
|
@ -92,7 +156,7 @@ public class Volume implements Comparable<Volume> {
|
||||||
private final String availabilityZone;
|
private final String availabilityZone;
|
||||||
private final Status status;
|
private final Status status;
|
||||||
private final Date createTime;
|
private final Date createTime;
|
||||||
private final Set<Attachment> attachments = Sets.newLinkedHashSet();
|
private final Set<Attachment> attachments;
|
||||||
|
|
||||||
public Volume(String region, String id, int size, String snapshotId, String availabilityZone, Volume.Status status,
|
public Volume(String region, String id, int size, String snapshotId, String availabilityZone, Volume.Status status,
|
||||||
Date createTime, Iterable<Attachment> attachments) {
|
Date createTime, Iterable<Attachment> attachments) {
|
||||||
|
@ -103,7 +167,7 @@ public class Volume implements Comparable<Volume> {
|
||||||
this.availabilityZone = availabilityZone;
|
this.availabilityZone = availabilityZone;
|
||||||
this.status = status;
|
this.status = status;
|
||||||
this.createTime = createTime;
|
this.createTime = createTime;
|
||||||
Iterables.addAll(this.attachments, attachments);
|
this.attachments = ImmutableSet.copyOf(attachments);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
|
@ -1,5 +1,3 @@
|
||||||
package com.gravitant.cloud.adapters.provision.providers;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Licensed to jclouds, Inc. (jclouds) under one or more
|
* Licensed to jclouds, Inc. (jclouds) under one or more
|
||||||
* contributor license agreements. See the NOTICE file
|
* contributor license agreements. See the NOTICE file
|
||||||
|
@ -19,54 +17,55 @@ package com.gravitant.cloud.adapters.provision.providers;
|
||||||
* under the License.
|
* under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
import javax.annotation.Resource;
|
package org.jclouds.ec2.predicates;
|
||||||
import javax.inject.Singleton;
|
|
||||||
|
|
||||||
|
import com.google.common.base.Predicate;
|
||||||
|
import com.google.common.collect.Iterables;
|
||||||
|
import com.google.inject.Inject;
|
||||||
|
import com.google.inject.Singleton;
|
||||||
import org.jclouds.ec2.domain.Attachment;
|
import org.jclouds.ec2.domain.Attachment;
|
||||||
import org.jclouds.ec2.domain.Volume;
|
import org.jclouds.ec2.domain.Volume;
|
||||||
import org.jclouds.ec2.services.ElasticBlockStoreClient;
|
import org.jclouds.ec2.services.ElasticBlockStoreClient;
|
||||||
import org.jclouds.logging.Logger;
|
import org.jclouds.logging.Logger;
|
||||||
|
|
||||||
import com.google.common.base.Predicate;
|
import javax.annotation.Resource;
|
||||||
import com.google.common.collect.Iterables;
|
|
||||||
import com.google.common.collect.Sets;
|
import static com.google.common.collect.Iterables.getLast;
|
||||||
import com.google.inject.Inject;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
*
|
|
||||||
* Tests to see if a volume is detached.
|
* Tests to see if a volume is detached.
|
||||||
*
|
*
|
||||||
* @author Karthik Arunachalam
|
* @author Karthik Arunachalam
|
||||||
*/
|
*/
|
||||||
@Singleton
|
@Singleton
|
||||||
public class VolumeDetached implements Predicate<Attachment> {
|
public class VolumeDetached implements Predicate<Attachment> {
|
||||||
|
|
||||||
private final ElasticBlockStoreClient client;
|
private final ElasticBlockStoreClient client;
|
||||||
@Resource
|
@Resource
|
||||||
protected Logger logger = Logger.NULL;
|
protected Logger logger = Logger.NULL;
|
||||||
|
|
||||||
@Inject
|
@Inject
|
||||||
public VolumeDetached(ElasticBlockStoreClient client) {
|
public VolumeDetached(ElasticBlockStoreClient client) {
|
||||||
this.client = client;
|
this.client = client;
|
||||||
}
|
}
|
||||||
|
|
||||||
public boolean apply(Attachment attachment) {
|
public boolean apply(Attachment attachment) {
|
||||||
logger.trace("looking for volume %s", attachment.getVolumeId());
|
logger.trace("looking for volume %s", attachment.getVolumeId());
|
||||||
Volume volume = Iterables.getOnlyElement(client.describeVolumesInRegion(attachment
|
Volume volume = Iterables.getOnlyElement(client.describeVolumesInRegion(attachment
|
||||||
.getRegion(), attachment.getVolumeId()));
|
.getRegion(), attachment.getVolumeId()));
|
||||||
|
|
||||||
/*If attachment size is 0 volume is detached for sure.*/
|
/*If attachment size is 0 volume is detached for sure.*/
|
||||||
if (volume.getAttachments().size() == 0) {
|
if (volume.getAttachments().size() == 0) {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* But if attachment size is > 0, then the attachment could be in any state.
|
/* But if attachment size is > 0, then the attachment could be in any state.
|
||||||
* So we need to check if the status is DETACHED (return true) or not (return false).
|
* So we need to check if the status is DETACHED (return true) or not (return false).
|
||||||
*/
|
*/
|
||||||
Attachment lastAttachment = Sets.newTreeSet(volume.getAttachments()).last();
|
Attachment lastAttachment = getLast(volume.getAttachments());
|
||||||
logger.trace("%s: looking for status %s: currently: %s", lastAttachment,
|
logger.trace("%s: looking for status %s: currently: %s", lastAttachment,
|
||||||
Attachment.Status.DETACHED, lastAttachment.getStatus());
|
Attachment.Status.DETACHED, lastAttachment.getStatus());
|
||||||
return lastAttachment.getStatus() == Attachment.Status.DETACHED;
|
return lastAttachment.getStatus() == Attachment.Status.DETACHED;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,115 @@
|
||||||
|
/**
|
||||||
|
* Licensed to jclouds, Inc. (jclouds) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file
|
||||||
|
* distributed with this work for additional information
|
||||||
|
* regarding copyright ownership. jclouds licenses this file
|
||||||
|
* to you under the Apache License, Version 2.0 (the
|
||||||
|
* "License"); you may not use this file except in compliance
|
||||||
|
* with the License. You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing,
|
||||||
|
* software distributed under the License is distributed on an
|
||||||
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||||
|
* KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations
|
||||||
|
* under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package org.jclouds.ec2.predicates;
|
||||||
|
|
||||||
|
import org.jclouds.ec2.domain.Attachment;
|
||||||
|
import org.jclouds.ec2.domain.Volume;
|
||||||
|
import org.jclouds.ec2.services.ElasticBlockStoreClient;
|
||||||
|
import org.testng.annotations.BeforeMethod;
|
||||||
|
import org.testng.annotations.DataProvider;
|
||||||
|
import org.testng.annotations.Test;
|
||||||
|
|
||||||
|
import java.util.Date;
|
||||||
|
import java.util.Set;
|
||||||
|
|
||||||
|
import static com.google.common.collect.Sets.newHashSet;
|
||||||
|
import static org.easymock.EasyMock.createMock;
|
||||||
|
import static org.easymock.EasyMock.expect;
|
||||||
|
import static org.easymock.EasyMock.replay;
|
||||||
|
import static org.easymock.EasyMock.verify;
|
||||||
|
import static org.jclouds.ec2.domain.Attachment.Status;
|
||||||
|
import static org.testng.Assert.assertFalse;
|
||||||
|
import static org.testng.Assert.assertTrue;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Andrei Savu
|
||||||
|
*/
|
||||||
|
@Test(groups = "unit", singleThreaded = true)
|
||||||
|
public class VolumeDetachedTest {
|
||||||
|
|
||||||
|
private ElasticBlockStoreClient client;
|
||||||
|
private VolumeDetached volumeDetached;
|
||||||
|
|
||||||
|
@BeforeMethod
|
||||||
|
public void setUp() {
|
||||||
|
client = createMock(ElasticBlockStoreClient.class);
|
||||||
|
volumeDetached = new VolumeDetached(client);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testVolumeWithEmptyListOfAttachments() {
|
||||||
|
Attachment attachment = newAttachmentWithStatus(Status.ATTACHED);
|
||||||
|
Set<Volume> volumes = newHashSet(newVolumeWithAttachments(/* empty */));
|
||||||
|
|
||||||
|
expect(client.describeVolumesInRegion(attachment.getRegion(),
|
||||||
|
attachment.getVolumeId())).andReturn(volumes);
|
||||||
|
replay(client);
|
||||||
|
|
||||||
|
assertTrue(volumeDetached.apply(attachment));
|
||||||
|
verify(client);
|
||||||
|
}
|
||||||
|
|
||||||
|
@DataProvider(name = "notDetachedStatuses")
|
||||||
|
public Object[][] provideNotDetachedStatuses() {
|
||||||
|
return new Object[][]{
|
||||||
|
{Status.ATTACHED},
|
||||||
|
{Status.ATTACHING},
|
||||||
|
{Status.BUSY},
|
||||||
|
{Status.DETACHING},
|
||||||
|
{Status.UNRECOGNIZED}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test(dataProvider = "notDetachedStatuses")
|
||||||
|
public void testWithDifferentStatus(Status attachmentStatus) {
|
||||||
|
Attachment attachment = newAttachmentWithStatus(attachmentStatus);
|
||||||
|
Set<Volume> volumes = newHashSet(newVolumeWithAttachments(attachment));
|
||||||
|
|
||||||
|
expect(client.describeVolumesInRegion(attachment.getRegion(),
|
||||||
|
attachment.getVolumeId())).andReturn(volumes);
|
||||||
|
replay(client);
|
||||||
|
|
||||||
|
assertFalse(volumeDetached.apply(attachment));
|
||||||
|
verify(client);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testWithStatusDetached() {
|
||||||
|
Attachment attachment = newAttachmentWithStatus(Status.DETACHED);
|
||||||
|
Set<Volume> volumes = newHashSet(newVolumeWithAttachments(attachment));
|
||||||
|
|
||||||
|
expect(client.describeVolumesInRegion(attachment.getRegion(),
|
||||||
|
attachment.getVolumeId())).andReturn(volumes);
|
||||||
|
replay(client);
|
||||||
|
|
||||||
|
assertTrue(volumeDetached.apply(attachment));
|
||||||
|
verify(client);
|
||||||
|
}
|
||||||
|
|
||||||
|
private Volume newVolumeWithAttachments(Attachment... attachments) {
|
||||||
|
return Volume.builder().region("us-east-1").attachments(attachments).build();
|
||||||
|
}
|
||||||
|
|
||||||
|
private Attachment newAttachmentWithStatus(Status status) {
|
||||||
|
return Attachment.builder()
|
||||||
|
.volumeId("1").status(status).region("us-east-1").attachTime(new Date())
|
||||||
|
.device("/dev/sda").instanceId("us-east-1/i-1234").build();
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue