/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF 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.openstack.marconi.v1.features;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import org.jclouds.openstack.marconi.v1.domain.CreateMessage;
import org.jclouds.openstack.marconi.v1.domain.QueueStats;
import org.jclouds.openstack.marconi.v1.internal.BaseMarconiApiLiveTest;
import org.testng.annotations.Test;

import java.util.List;
import java.util.Map;
import java.util.UUID;

import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertFalse;
import static org.testng.Assert.assertNotNull;
import static org.testng.Assert.assertTrue;

@Test(groups = "live", testName = "QueueApiLiveTest", singleThreaded = true)
public class QueueApiLiveTest extends BaseMarconiApiLiveTest {

   public void create() throws Exception {
      for (String zoneId : api.getConfiguredZones()) {
         QueueApi queueApi = api.getQueueApiForZoneAndQueue(zoneId, "jclouds-test");
         boolean success = queueApi.create();

         assertTrue(success);
      }
   }

   @Test(dependsOnMethods = { "create" })
   public void exists() throws Exception {
      for (String zoneId : api.getConfiguredZones()) {
         QueueApi queueApi = api.getQueueApiForZoneAndQueue(zoneId, "jclouds-test");
         boolean success = queueApi.exists();

         assertTrue(success);
      }
   }

   @Test(dependsOnMethods = { "exists" })
   public void setMetadata() throws Exception {
      for (String zoneId : api.getConfiguredZones()) {
         QueueApi queueApi = api.getQueueApiForZoneAndQueue(zoneId, "jclouds-test");
         Map<String, String> metadata = ImmutableMap.of("key1", "value1");
         boolean success = queueApi.setMetadata(metadata);

         assertTrue(success);
      }
   }

   @Test(dependsOnMethods = { "setMetadata" })
   public void getMetadata() throws Exception {
      for (String zoneId : api.getConfiguredZones()) {
         QueueApi queueApi = api.getQueueApiForZoneAndQueue(zoneId, "jclouds-test");
         Map<String, String> metadata = queueApi.getMetadata();

         assertEquals(metadata.get("key1"), "value1");
      }
   }

   @Test(dependsOnMethods = { "getMetadata" })
   public void getStatsWithoutTotal() throws Exception {
      for (String zoneId : api.getConfiguredZones()) {
         QueueApi queueApi = api.getQueueApiForZoneAndQueue(zoneId, "jclouds-test");
         QueueStats stats = queueApi.getStats();

         assertEquals(stats.getMessagesStats().getClaimed(), 0);
         assertEquals(stats.getMessagesStats().getFree(), 0);
         assertEquals(stats.getMessagesStats().getTotal(), 0);
         assertFalse(stats.getMessagesStats().getOldest().isPresent());
         assertFalse(stats.getMessagesStats().getNewest().isPresent());
      }
   }

   @Test(dependsOnMethods = { "getStatsWithoutTotal" })
   public void getStatsWithTotal() throws Exception {
      for (String zoneId : api.getConfiguredZones()) {
         MessageApi messageApi = api.getMessageApiForZoneAndQueue(zoneId, "jclouds-test");

         UUID clientId = UUID.fromString("3381af92-2b9e-11e3-b191-71861300734c");
         String json1 = "{\"event\":{\"type\":\"hockey\",\"players\":[\"bob\",\"jim\",\"sally\"]}}";
         CreateMessage message1 = CreateMessage.builder().ttl(120).body(json1).build();
         List<CreateMessage> message = ImmutableList.of(message1);

         messageApi.create(clientId, message);

         QueueApi queueApi = api.getQueueApiForZoneAndQueue(zoneId, "jclouds-test");
         QueueStats stats = queueApi.getStats();

         assertEquals(stats.getMessagesStats().getClaimed(), 0);
         assertEquals(stats.getMessagesStats().getFree(), 1);
         assertEquals(stats.getMessagesStats().getTotal(), 1);
         assertTrue(stats.getMessagesStats().getOldest().isPresent());
         assertNotNull(stats.getMessagesStats().getOldest().get().getId());
         assertTrue(stats.getMessagesStats().getNewest().isPresent());
         assertNotNull(stats.getMessagesStats().getNewest().get().getId());
      }
   }

   @Test(dependsOnMethods = { "getStatsWithTotal" })
   public void delete() throws Exception {
      for (String zoneId : api.getConfiguredZones()) {
         QueueApi queueApi = api.getQueueApiForZoneAndQueue(zoneId, "jclouds-test");
         boolean success = queueApi.delete();

         assertTrue(success);
      }
   }

   @Test(dependsOnMethods = { "delete" })
   public void doesNotExist() throws Exception {
      for (String zoneId : api.getConfiguredZones()) {
         QueueApi queueApi = api.getQueueApiForZoneAndQueue(zoneId, "jclouds-test");
         boolean success = queueApi.exists();

         assertFalse(success);
      }
   }
}