package org.jboss.resteasy.test.rx.rxjava2;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicReference;

import jakarta.ws.rs.ClientErrorException;
import jakarta.ws.rs.InternalServerErrorException;
import jakarta.ws.rs.client.ClientBuilder;
import jakarta.ws.rs.client.Entity;
import jakarta.ws.rs.core.GenericType;
import jakarta.ws.rs.core.MediaType;

import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.container.test.api.RunAsClient;
import org.jboss.arquillian.junit5.ArquillianExtension;
import org.jboss.resteasy.client.jaxrs.ResteasyClient;
import org.jboss.resteasy.client.jaxrs.ResteasyClientBuilder;
import org.jboss.resteasy.rxjava2.ObservableRxInvoker;
import org.jboss.resteasy.rxjava2.ObservableRxInvokerProvider;
import org.jboss.resteasy.test.rx.resource.Bytes;
import org.jboss.resteasy.test.rx.resource.RxScheduledExecutorService;
import org.jboss.resteasy.test.rx.resource.TRACE;
import org.jboss.resteasy.test.rx.resource.TestException;
import org.jboss.resteasy.test.rx.resource.TestExceptionMapper;
import org.jboss.resteasy.test.rx.resource.Thing;
import org.jboss.resteasy.test.rx.rxjava2.resource.Rx2ObservableResourceImpl;
import org.jboss.resteasy.utils.PortProviderUtil;
import org.jboss.resteasy.utils.TestUtil;
import org.jboss.shrinkwrap.api.Archive;
import org.jboss.shrinkwrap.api.asset.StringAsset;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.MethodOrderer.MethodName;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.TestMethodOrder;
import org.junit.jupiter.api.extension.ExtendWith;

import io.reactivex.Observable;

/**
 * @tpSubChapter Reactive classes
 * @tpChapter Integration tests
 * @tpSince RESTEasy 4.0
 *
 *          In these tests, the server uses Observables to create results asynchronously and streams the elements
 *          of the Observables as they are created.
 *
 *          The client makes invocations on an ObservableRxInvoker.
 */
@ExtendWith(ArquillianExtension.class)
@RunAsClient
@TestMethodOrder(MethodName.class)
public class Rx2ObservableTest {

    private static ResteasyClient client;
    private static CountDownLatch latch;
    private static AtomicInteger errors;

    private static final List<String> xStringList = new ArrayList<String>();
    private static final List<String> aStringList = new ArrayList<String>();
    private static final List<Thing> xThingList = new ArrayList<Thing>();
    private static final List<Thing> aThingList = new ArrayList<Thing>();
    private static final List<List<Thing>> xThingListList = new ArrayList<List<Thing>>();
    private static final List<List<Thing>> aThingListList = new ArrayList<List<Thing>>();
    private static final Entity<String> aEntity = Entity.entity("a", MediaType.TEXT_PLAIN_TYPE);
    private static final Entity<String> threeEntity = Entity.entity("3", MediaType.TEXT_PLAIN_TYPE);

    private static AtomicReference<Object> value = new AtomicReference<Object>();
    private static ArrayList<String> stringList = new ArrayList<String>();
    private static ArrayList<Thing> thingList = new ArrayList<Thing>();
    private static ArrayList<List<?>> thingListList = new ArrayList<List<?>>();
    private static ArrayList<byte[]> bytesList = new ArrayList<byte[]>();
    private static GenericType<List<Thing>> LIST_OF_THING = new GenericType<List<Thing>>() {
    };

    static {
        for (int i = 0; i < 3; i++) {
            xStringList.add("x");
        }
        for (int i = 0; i < 3; i++) {
            aStringList.add("a");
        }
        for (int i = 0; i < 3; i++) {
            xThingList.add(new Thing("x"));
        }
        for (int i = 0; i < 3; i++) {
            aThingList.add(new Thing("a"));
        }
        for (int i = 0; i < 2; i++) {
            xThingListList.add(xThingList);
        }
        for (int i = 0; i < 2; i++) {
            aThingListList.add(aThingList);
        }
    }

    @Deployment
    public static Archive<?> deploy() {

        WebArchive war = TestUtil.prepareArchive(Rx2ObservableTest.class.getSimpleName());
        war.addClass(Thing.class);
        war.addClass(TRACE.class);
        war.addClass(Bytes.class);
        war.addClass(RxScheduledExecutorService.class);
        war.addClass(TestException.class);
        war.setManifest(new StringAsset("Manifest-Version: 1.0\n"
                + "Dependencies: org.jboss.resteasy.resteasy-rxjava2 services\n"));
        return TestUtil.finishContainerPrepare(war, null, Rx2ObservableResourceImpl.class, TestExceptionMapper.class);
    }

    private static String generateURL(String path) {
        return PortProviderUtil.generateURL(path, Rx2ObservableTest.class.getSimpleName());
    }

    //////////////////////////////////////////////////////////////////////////////
    @BeforeAll
    public static void beforeClass() throws Exception {
        client = (ResteasyClient) ClientBuilder.newClient();
    }

    @BeforeEach
    public void before() throws Exception {
        stringList.clear();
        thingList.clear();
        thingListList.clear();
        bytesList.clear();
        latch = new CountDownLatch(1);
        errors = new AtomicInteger(0);
        value.set(null);
    }

    @AfterAll
    public static void after() throws Exception {
        client.close();
    }

    //////////////////////////////////////////////////////////////////////////////
    @SuppressWarnings("unchecked")
    @Test
    public void testGet() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/get/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable = (Observable<String>) invoker.get();
        observable.subscribe(
                (String o) -> stringList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xStringList, stringList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testGetThing() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/get/thing")).request().rx(ObservableRxInvoker.class);
        Observable<Thing> observable = (Observable<Thing>) invoker.get(Thing.class);
        observable.subscribe(
                (Thing o) -> thingList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xThingList, thingList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testGetThingList() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/get/thing/list")).request().rx(ObservableRxInvoker.class);
        Observable<List<Thing>> observable = (Observable<List<Thing>>) invoker.get(LIST_OF_THING);
        observable.subscribe(
                (List<?> l) -> thingListList.add(l),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xThingListList, thingListList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testGetBytes() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/get/bytes")).request().rx(ObservableRxInvoker.class);
        Observable<byte[]> observable = (Observable<byte[]>) invoker.get(byte[].class);
        observable.subscribe(
                (byte[] b) -> bytesList.add(b),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(3, bytesList.size());
        for (byte[] b : bytesList) {
            Assertions.assertTrue(Arrays.equals(Bytes.BYTES, b));
        }
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testPut() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/put/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable = (Observable<String>) invoker.put(aEntity);
        observable.subscribe(
                (String o) -> stringList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(aStringList, stringList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testPutThing() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/put/thing")).request().rx(ObservableRxInvoker.class);
        Observable<Thing> observable = (Observable<Thing>) invoker.put(aEntity, Thing.class);
        observable.subscribe(
                (Thing o) -> thingList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(aThingList, thingList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testPutThingList() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/put/thing/list")).request().rx(ObservableRxInvoker.class);
        Observable<List<Thing>> observable = (Observable<List<Thing>>) invoker.put(aEntity, LIST_OF_THING);
        observable.subscribe(
                (List<?> l) -> thingListList.add(l),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(aThingListList, thingListList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testPutBytes() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/put/bytes")).request().rx(ObservableRxInvoker.class);
        Observable<byte[]> observable = (Observable<byte[]>) invoker.put(threeEntity, byte[].class);
        observable.subscribe(
                (byte[] b) -> bytesList.add(b),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(3, bytesList.size());
        for (byte[] b : bytesList) {
            Assertions.assertTrue(Arrays.equals(Bytes.BYTES, b));
        }
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testPost() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/post/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable = (Observable<String>) invoker.post(aEntity);
        observable.subscribe(
                (String o) -> stringList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(aStringList, stringList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testPostThing() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/post/thing")).request().rx(ObservableRxInvoker.class);
        Observable<Thing> observable = (Observable<Thing>) invoker.post(aEntity, Thing.class);
        observable.subscribe(
                (Thing o) -> thingList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(aThingList, thingList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testPostThingList() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/post/thing/list")).request().rx(ObservableRxInvoker.class);
        Observable<List<Thing>> observable = (Observable<List<Thing>>) invoker.post(aEntity, LIST_OF_THING);
        observable.subscribe(
                (List<?> l) -> thingListList.add(l),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(aThingListList, thingListList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testPostBytes() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/post/bytes")).request().rx(ObservableRxInvoker.class);
        Observable<byte[]> observable = (Observable<byte[]>) invoker.post(threeEntity, byte[].class);
        observable.subscribe(
                (byte[] b) -> bytesList.add(b),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(3, bytesList.size());
        for (byte[] b : bytesList) {
            Assertions.assertTrue(Arrays.equals(Bytes.BYTES, b));
        }
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testDelete() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/delete/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable = (Observable<String>) invoker.delete();
        observable.subscribe(
                (String o) -> stringList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xStringList, stringList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testDeleteThing() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/delete/thing")).request().rx(ObservableRxInvoker.class);
        Observable<Thing> observable = (Observable<Thing>) invoker.delete(Thing.class);
        observable.subscribe(
                (Thing o) -> thingList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xThingList, thingList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testDeleteThingList() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/delete/thing/list")).request().rx(ObservableRxInvoker.class);
        Observable<List<Thing>> observable = (Observable<List<Thing>>) invoker.delete(LIST_OF_THING);
        observable.subscribe(
                (List<?> l) -> thingListList.add(l),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xThingListList, thingListList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testDeleteBytes() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/delete/bytes")).request().rx(ObservableRxInvoker.class);
        Observable<byte[]> observable = (Observable<byte[]>) invoker.delete(byte[].class);
        observable.subscribe(
                (byte[] b) -> bytesList.add(b),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(3, bytesList.size());
        for (byte[] b : bytesList) {
            Assertions.assertTrue(Arrays.equals(Bytes.BYTES, b));
        }
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testHead() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/head/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable = (Observable<String>) invoker.head();
        observable.subscribe(
                (String s) -> value.set(s), // HEAD - no body
                (Throwable t) -> throwableContains(t, "Input stream was empty"));
        Assertions.assertNull(value.get());
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testOptions() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/options/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable = (Observable<String>) invoker.options();
        observable.subscribe(
                (String o) -> stringList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xStringList, stringList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testOptionsThing() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/options/thing")).request().rx(ObservableRxInvoker.class);
        Observable<Thing> observable = (Observable<Thing>) invoker.options(Thing.class);
        observable.subscribe(
                (Thing o) -> thingList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xThingList, thingList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testOptionsThingList() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/options/thing/list")).request().rx(ObservableRxInvoker.class);
        Observable<List<Thing>> observable = (Observable<List<Thing>>) invoker.options(LIST_OF_THING);
        observable.subscribe(
                (List<?> l) -> thingListList.add(l),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xThingListList, thingListList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testOptionsBytes() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/options/bytes")).request().rx(ObservableRxInvoker.class);
        Observable<byte[]> observable = (Observable<byte[]>) invoker.options(byte[].class);
        observable.subscribe(
                (byte[] b) -> bytesList.add(b),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(3, bytesList.size());
        for (byte[] b : bytesList) {
            Assertions.assertTrue(Arrays.equals(Bytes.BYTES, b));
        }
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testTrace() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/trace/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable = (Observable<String>) invoker.trace();
        observable.subscribe(
                (String o) -> stringList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xStringList, stringList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testTraceThing() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/trace/thing")).request().rx(ObservableRxInvoker.class);
        Observable<Thing> observable = (Observable<Thing>) invoker.trace(Thing.class);
        observable.subscribe(
                (Thing o) -> thingList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xThingList, thingList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testTraceThingList() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/trace/thing/list")).request().rx(ObservableRxInvoker.class);
        Observable<List<Thing>> observable = (Observable<List<Thing>>) invoker.trace(LIST_OF_THING);
        observable.subscribe(
                (List<?> l) -> thingListList.add(l),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xThingListList, thingListList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testTraceBytes() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/trace/bytes")).request().rx(ObservableRxInvoker.class);
        Observable<byte[]> observable = (Observable<byte[]>) invoker.trace(byte[].class);
        observable.subscribe(
                (byte[] b) -> bytesList.add(b),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(3, bytesList.size());
        for (byte[] b : bytesList) {
            Assertions.assertTrue(Arrays.equals(Bytes.BYTES, b));
        }
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testMethodGet() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/get/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable = (Observable<String>) invoker.method("GET");
        observable.subscribe(
                (String o) -> stringList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xStringList, stringList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testMethodGetThing() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/get/thing")).request().rx(ObservableRxInvoker.class);
        Observable<Thing> observable = (Observable<Thing>) invoker.method("GET", Thing.class);
        observable.subscribe(
                (Thing o) -> thingList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xThingList, thingList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testMethodGetThingList() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/get/thing/list")).request().rx(ObservableRxInvoker.class);
        Observable<List<Thing>> observable = (Observable<List<Thing>>) invoker.method("GET", LIST_OF_THING);
        observable.subscribe(
                (List<?> l) -> thingListList.add(l),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(xThingListList, thingListList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testMethodGetBytes() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/get/bytes")).request().rx(ObservableRxInvoker.class);
        Observable<byte[]> observable = (Observable<byte[]>) invoker.method("GET", byte[].class);
        observable.subscribe(
                (byte[] b) -> bytesList.add(b),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(3, bytesList.size());
        for (byte[] b : bytesList) {
            Assertions.assertTrue(Arrays.equals(Bytes.BYTES, b));
        }
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testMethodPost() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/post/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable = (Observable<String>) invoker.method("POST", aEntity);
        observable.subscribe(
                (String o) -> stringList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(aStringList, stringList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testMethodPostThing() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/post/thing")).request().rx(ObservableRxInvoker.class);
        Observable<Thing> observable = (Observable<Thing>) invoker.method("POST", aEntity, Thing.class);
        observable.subscribe(
                (Thing o) -> thingList.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(aThingList, thingList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testMethodPostThingList() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/post/thing/list")).request().rx(ObservableRxInvoker.class);
        Observable<List<Thing>> observable = (Observable<List<Thing>>) invoker.method("POST", aEntity, LIST_OF_THING);
        observable.subscribe(
                (List<?> l) -> thingListList.add(l),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(aThingListList, thingListList);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testMethodPostBytes() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/post/bytes")).request().rx(ObservableRxInvoker.class);
        Observable<byte[]> observable = (Observable<byte[]>) invoker.method("POST", threeEntity, byte[].class);
        observable.subscribe(
                (byte[] b) -> bytesList.add(b),
                (Throwable t) -> errors.incrementAndGet(),
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(3, bytesList.size());
        for (byte[] b : bytesList) {
            Assertions.assertTrue(Arrays.equals(Bytes.BYTES, b));
        }
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testScheduledExecutorService() throws Exception {
        {
            RxScheduledExecutorService.used = false;
            ObservableRxInvoker invoker = client.target(generateURL("/get/string")).request().rx(ObservableRxInvoker.class);
            Observable<String> observable = (Observable<String>) invoker.get();
            observable.subscribe(
                    (String o) -> stringList.add(o),
                    (Throwable t) -> errors.incrementAndGet(),
                    () -> latch.countDown());
            boolean waitResult = latch.await(30, TimeUnit.SECONDS);
            Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
            Assertions.assertEquals(0, errors.get());
            Assertions.assertFalse(RxScheduledExecutorService.used);
            Assertions.assertEquals(xStringList, stringList);
        }

        {
            stringList.clear();
            latch = new CountDownLatch(1);
            RxScheduledExecutorService.used = false;
            RxScheduledExecutorService executor = new RxScheduledExecutorService();
            ResteasyClient client = ((ResteasyClientBuilder) ClientBuilder.newBuilder()).executorService(executor).build();
            client.register(ObservableRxInvokerProvider.class);
            ObservableRxInvoker invoker = client.target(generateURL("/get/string")).request().rx(ObservableRxInvoker.class);
            Observable<String> observable = (Observable<String>) invoker.get();
            stringList.clear();
            observable.subscribe(
                    (String o) -> stringList.add(o),
                    (Throwable t) -> errors.incrementAndGet(),
                    () -> latch.countDown());
            boolean waitResult = latch.await(30, TimeUnit.SECONDS);
            Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
            Assertions.assertEquals(0, errors.get());
            Assertions.assertTrue(RxScheduledExecutorService.used);
            Assertions.assertEquals(xStringList, stringList);
            client.close();
        }
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testUnhandledException() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/exception/unhandled")).request()
                .rx(ObservableRxInvoker.class);
        Observable<Thing> observable = (Observable<Thing>) invoker.get(Thing.class);
        AtomicReference<Object> value = new AtomicReference<Object>();
        observable.subscribe(
                (Thing t) -> thingList.add(t),
                (Throwable t) -> {
                    value.set(t);
                    latch.countDown();
                },
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Throwable t = (Throwable) value.get();
        Assertions.assertEquals(InternalServerErrorException.class, t.getClass());
        Assertions.assertTrue(t.getMessage().contains("500"));
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testHandledException() throws Exception {
        ObservableRxInvoker invoker = client.target(generateURL("/exception/handled")).request().rx(ObservableRxInvoker.class);
        Observable<Thing> observable = (Observable<Thing>) invoker.get(Thing.class);
        AtomicReference<Object> value = new AtomicReference<Object>();
        observable.subscribe(
                (Thing t) -> thingList.add(t),
                (Throwable t) -> {
                    value.set(t);
                    latch.countDown();
                },
                () -> latch.countDown());
        boolean waitResult = latch.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Throwable t = (Throwable) value.get();
        Assertions.assertEquals(ClientErrorException.class, t.getClass());
        Assertions.assertTrue(t.getMessage().contains("444"));
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testGetTwoClients() throws Exception {
        CountDownLatch cdl = new CountDownLatch(2);
        CopyOnWriteArrayList<String> list = new CopyOnWriteArrayList<String>();

        ResteasyClient client1 = (ResteasyClient) ClientBuilder.newClient();
        client1.register(ObservableRxInvokerProvider.class);
        ObservableRxInvoker invoker1 = client1.target(generateURL("/get/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable1 = (Observable<String>) invoker1.get();

        ResteasyClient client2 = (ResteasyClient) ClientBuilder.newClient();
        client2.register(ObservableRxInvokerProvider.class);
        ObservableRxInvoker invoker2 = client2.target(generateURL("/get/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable2 = (Observable<String>) invoker2.get();

        observable1.subscribe(
                (String o) -> list.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> cdl.countDown());

        observable2.subscribe(
                (String o) -> list.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> cdl.countDown());

        boolean waitResult = cdl.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(6, list.size());
        for (int i = 0; i < 6; i++) {
            Assertions.assertEquals("x", list.get(i));
        }
        client1.close();
        client2.close();
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testGetTwoInvokers() throws Exception {
        CountDownLatch cdl = new CountDownLatch(2);
        CopyOnWriteArrayList<String> list = new CopyOnWriteArrayList<String>();

        ObservableRxInvoker invoker1 = client.target(generateURL("/get/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable1 = (Observable<String>) invoker1.get();

        ObservableRxInvoker invoker2 = client.target(generateURL("/get/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable2 = (Observable<String>) invoker2.get();

        observable1.subscribe(
                (String o) -> list.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> cdl.countDown());

        observable2.subscribe(
                (String o) -> list.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> cdl.countDown());

        boolean waitResult = cdl.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(6, list.size());
        for (int i = 0; i < 6; i++) {
            Assertions.assertEquals("x", list.get(i));
        }
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testGetTwoObservables() throws Exception {
        CountDownLatch cdl = new CountDownLatch(2);
        CopyOnWriteArrayList<String> list = new CopyOnWriteArrayList<String>();

        ObservableRxInvoker invoker = client.target(generateURL("/get/string")).request().rx(ObservableRxInvoker.class);
        Observable<String> observable1 = (Observable<String>) invoker.get();
        Observable<String> observable2 = (Observable<String>) invoker.get();

        observable1.subscribe(
                (String o) -> list.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> cdl.countDown());

        observable2.subscribe(
                (String o) -> list.add(o),
                (Throwable t) -> errors.incrementAndGet(),
                () -> cdl.countDown());

        boolean waitResult = cdl.await(30, TimeUnit.SECONDS);
        Assertions.assertTrue(waitResult, "Waiting for event to be delivered has timed out.");
        Assertions.assertEquals(0, errors.get());
        Assertions.assertEquals(6, list.size());
        for (int i = 0; i < 6; i++) {
            Assertions.assertEquals("x", list.get(i));
        }
    }

    private static boolean throwableContains(Throwable t, String s) {
        while (t != null) {
            if (t.getMessage().contains(s)) {
                return true;
            }
            t = t.getCause();
        }
        return false;
    }
}
