/*
 * Copyright 2016 Red Hat Inc.
 *
 * Licensed 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 enmasse.smoketest;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.util.Arrays;
import java.util.List;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;

public class StoreAndForwardTopicTest extends VertxTestBase{
    @Before
    public void setupReplicas() throws InterruptedException {
        TestUtils.setReplicas("mytopic", "mytopic", 4, 10, TimeUnit.MINUTES);
    }

    @After
    public void teardownReplicas() throws InterruptedException {
        TestUtils.setReplicas("mytopic", "mytopic", 1, 10, TimeUnit.MINUTES);
    }

    @Test
    public void testMultipleSubscribers() throws InterruptedException, TimeoutException, ExecutionException {
        String dest = "mytopic";
        waitUntilReady(dest, 5, TimeUnit.MINUTES);
        EnMasseClient client = createClient(true);
        List<String> msgs = Arrays.asList("foo", "bar", "baz");

        List<Future<List<String>>> recvResults = Arrays.asList(
                client.recvMessages(dest, msgs.size()),
                client.recvMessages(dest, msgs.size()),
                client.recvMessages(dest, msgs.size()));

        assertThat(client.sendMessages(dest, msgs).get(1, TimeUnit.MINUTES), is(msgs.size()));

        assertThat(recvResults.get(0).get(1, TimeUnit.MINUTES).size(), is(msgs.size()));
        assertThat(recvResults.get(1).get(1, TimeUnit.MINUTES).size(), is(msgs.size()));
        assertThat(recvResults.get(2).get(1, TimeUnit.MINUTES).size(), is(msgs.size()));
    }
}

@@ -614,24 +614,139 @@
 smoketest;%0A%0A
+import org.apache.qpid.proton.amqp.messaging.ApplicationProperties;%0Aimport org.apache.qpid.proton.message.Message;%0A
 import org.j
@@ -757,16 +757,16 @@
 .After;%0A
-
 import o
@@ -831,16 +831,46 @@
 Arrays;%0A
+import java.util.Collections;%0A
 import j
@@ -1205,166 +1205,8 @@
 ase%7B
-%0A    @Before%0A    public void setupReplicas() throws InterruptedException %7B%0A        TestUtils.setReplicas(%22mytopic%22, %22mytopic%22, 4, 10, TimeUnit.MINUTES);%0A    %7D
 %0A%0A  
@@ -1476,32 +1476,110 @@
 tionException %7B%0A
+        TestUtils.setReplicas(%22mytopic%22, %22mytopic%22, 4, 10, TimeUnit.MINUTES);%0A
         String d
@@ -2262,32 +2262,32 @@
 (msgs.size()));%0A
-
         assertTh
@@ -2364,11 +2364,1393 @@
 );%0A    %7D
+%0A%0A    @Test%0A    public void testSubscriptionService() throws Exception %7B%0A        String topic = %22mytopic%22;%0A        String address = %22myaddress%22;%0A%0A        EnMasseClient ctrlClient = createClient(false);%0A        EnMasseClient client = createClient(true);%0A%0A        waitUntilReady(topic, 5, TimeUnit.MINUTES);%0A        Message sub = Message.Factory.create();%0A        sub.setAddress(%22$subctrl%22);%0A        sub.setCorrelationId(address);%0A        sub.setSubject(%22subscribe%22);%0A        sub.setApplicationProperties(new ApplicationProperties(Collections.singletonMap(%22root_address%22, topic)));%0A        ctrlClient.sendMessages(%22$subctrl%22, sub).get(5, TimeUnit.MINUTES);%0A%0A        Thread.sleep(10000);%0A%0A        List%3CString%3E msgs = Arrays.asList(%22foo%22, %22bar%22, %22baz%22);%0A        Future%3CList%3CString%3E%3E recvResult = client.recvMessages(address, msgs.size());%0A%0A        assertThat(client.sendMessages(topic, msgs).get(1, TimeUnit.MINUTES), is(msgs.size()));%0A        assertThat(recvResult.get(1, TimeUnit.MINUTES).size(), is(msgs.size()));%0A%0A        Message unsub = Message.Factory.create();%0A        unsub.setAddress(%22$subctrl%22);%0A        unsub.setCorrelationId(address);%0A        unsub.setApplicationProperties(new ApplicationProperties(Collections.singletonMap(%22root_address%22, topic)));%0A        unsub.setSubject(%22unsubscribe%22);%0A        ctrlClient.sendMessages(%22$subctrl%22, unsub).get(5, TimeUnit.MINUTES);%0A    %7D
 %0A%7D%0A

Add test for subscription service