/**
* 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.azurequeue;
import static com.google.common.base.Preconditions.checkNotNull;
import static org.jclouds.azure.storage.options.ListOptions.Builder.prefix;
import static org.jclouds.azurequeue.options.GetOptions.Builder.maxMessages;
import static org.jclouds.azurequeue.options.PutMessageOptions.Builder.withTTL;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertTrue;
import java.security.SecureRandom;
import java.util.Properties;
import java.util.Set;
import org.jclouds.Constants;
import org.jclouds.azure.storage.domain.BoundedSet;
import org.jclouds.azure.storage.options.CreateOptions;
import org.jclouds.azurequeue.domain.QueueMessage;
import org.jclouds.azurequeue.domain.QueueMetadata;
import org.jclouds.http.HttpResponseException;
import org.jclouds.logging.log4j.config.Log4JLoggingModule;
import org.jclouds.rest.RestContextFactory;
import org.testng.annotations.BeforeGroups;
import org.testng.annotations.Test;
import com.google.common.collect.ImmutableMultimap;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Iterables;
import com.google.inject.Module;
/**
* Tests behavior of {@code AzureQueueClient}
*
* @author Adrian Cole
*/
@Test(groups = "live", sequential = true)
public class AzureQueueClientLiveTest {
protected AzureQueueClient connection;
private String queuePrefix = System.getProperty("user.name") + "-azurequeue";
protected String provider = "azurequeue";
protected String identity;
protected String credential;
protected String endpoint;
protected String apiVersion;
protected void setupCredentials() {
identity = checkNotNull(System.getProperty("test." + provider + ".identity"), "test." + provider + ".identity");
credential = checkNotNull(System.getProperty("test." + provider + ".credential"), "test." + provider
+ ".credential");
endpoint = checkNotNull(System.getProperty("test." + provider + ".endpoint"), "test." + provider + ".endpoint");
apiVersion = checkNotNull(System.getProperty("test." + provider + ".api-version"), "test." + provider
+ ".api-version");
}
protected Properties setupProperties() {
Properties overrides = new Properties();
overrides.setProperty(Constants.PROPERTY_TRUST_ALL_CERTS, "true");
overrides.setProperty(Constants.PROPERTY_RELAX_HOSTNAME, "true");
overrides.setProperty(provider + ".identity", identity);
overrides.setProperty(provider + ".credential", credential);
overrides.setProperty(provider + ".endpoint", endpoint);
overrides.setProperty(provider + ".api-version", apiVersion);
return overrides;
}
@BeforeGroups(groups = { "live" })
public void setupClient() {
setupCredentials();
Properties overrides = setupProperties();
connection = (AzureQueueClient) new RestContextFactory().createContext(provider,
ImmutableSet.<Module> of(new Log4JLoggingModule()), overrides).getApi();
}
@Test
public void testListQueues() throws Exception {
BoundedSet<QueueMetadata> response = connection.listQueues();
assert null != response;
long initialQueueCount = response.size();
assertTrue(initialQueueCount >= 0);
}
String privateQueue;
@Test(timeOut = 5 * 60 * 1000)
public void testCreateQueue() throws Exception {
boolean created = false;
while (!created) {
privateQueue = queuePrefix + new SecureRandom().nextInt();
try {
created = connection.createQueue(privateQueue, CreateOptions.Builder.withMetadata(ImmutableMultimap.of(
"foo", "bar")));
} catch (HttpResponseException htpe) {
if (htpe.getResponse().getStatusCode() == 409) {
continue;
} else {
throw htpe;
}
}
}
BoundedSet<QueueMetadata> response = connection.listQueues();
assert null != response;
long queueCount = response.size();
assertTrue(queueCount >= 1);
// TODO ... check to see the queue actually exists
}
@Test(timeOut = 5 * 60 * 1000, dependsOnMethods = { "testCreateQueue" })
public void testListQueuesWithOptions() throws Exception {
BoundedSet<QueueMetadata> response = connection.listQueues(prefix(privateQueue).maxResults(1));
assert null != response;
long initialQueueCount = response.size();
assertTrue(initialQueueCount >= 0);
assertEquals(privateQueue, response.getPrefix());
assertEquals(1, response.getMaxResults());
}
@Test(timeOut = 5 * 60 * 1000, dependsOnMethods = { "testCreateQueue" })
public void testPutMessage() throws Exception {
connection.putMessage(privateQueue, "holycow", withTTL(4));
connection.putMessage(privateQueue, "holymoo", withTTL(4));
}
@Test(timeOut = 5 * 60 * 1000, dependsOnMethods = { "testPutMessage" })
public void testGetMessages() throws Exception {
Set<QueueMessage> messages = connection.getMessages(privateQueue, maxMessages(2));
QueueMessage m1 = Iterables.get(messages, 0);
assertEquals(m1.getMessageText(), "holycow");
QueueMessage m2 = Iterables.get(messages, 1);
assertEquals(m2.getMessageText(), "holymoo");
assertEquals(connection.getMessages(privateQueue).size(), 0);
}
@Test(timeOut = 5 * 60 * 1000, dependsOnMethods = { "testGetMessages" })
public void testDeleteQueue() throws Exception {
connection.clearMessages(privateQueue);
connection.deleteQueue(privateQueue);
// TODO loop for up to 30 seconds checking if they are really gone
}
}