package com.github.dockerjava.netty.exec;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.equalTo;
import static org.hamcrest.Matchers.is;
import static org.hamcrest.Matchers.isEmptyString;
import static org.hamcrest.Matchers.not;

import java.lang.reflect.Method;

import org.testng.ITestResult;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.AfterTest;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.Test;

import com.github.dockerjava.api.command.CreateContainerResponse;
import com.github.dockerjava.api.command.InspectContainerResponse;
import com.github.dockerjava.api.exception.DockerException;
import com.github.dockerjava.api.exception.NotFoundException;
import com.github.dockerjava.api.model.WaitResponse;
import com.github.dockerjava.core.command.WaitContainerResultCallback;
import com.github.dockerjava.netty.AbstractNettyDockerClientTest;

@Test(groups = "integration")
public class WaitContainerCmdExecTest extends AbstractNettyDockerClientTest {

    @BeforeTest
    public void beforeTest() throws Exception {
        super.beforeTest();
    }

    @AfterTest
    public void afterTest() {
        super.afterTest();
    }

    @BeforeMethod
    public void beforeMethod(Method method) {
        super.beforeMethod(method);
    }

    @AfterMethod
    public void afterMethod(ITestResult result) {
        super.afterMethod(result);
    }

    @Test
    public void testWaitContainer() throws DockerException {

        CreateContainerResponse container = dockerClient.createContainerCmd("busybox").withCmd("true").exec();

        LOG.info("Created container: {}", container.toString());
        assertThat(container.getId(), not(isEmptyString()));

        dockerClient.startContainerCmd(container.getId()).exec();

        int exitCode = dockerClient.waitContainerCmd(container.getId()).exec(new WaitContainerResultCallback())
                .awaitStatusCode();
        LOG.info("Container exit code: {}", exitCode);

        assertThat(exitCode, equalTo(0));

        InspectContainerResponse inspectContainerResponse = dockerClient.inspectContainerCmd(container.getId()).exec();
        LOG.info("Container Inspect: {}", inspectContainerResponse.toString());

        assertThat(inspectContainerResponse.getState().getRunning(), is(equalTo(false)));
        assertThat(inspectContainerResponse.getState().getExitCode(), is(equalTo(exitCode)));
    }

    @Test(expectedExceptions = NotFoundException.class)
    public void testWaitNonExistingContainer() throws DockerException {

        WaitContainerResultCallback callback = new WaitContainerResultCallback() {
            public void onNext(WaitResponse waitResponse) {
                fail("expected NotFoundException");
            };
        };

        dockerClient.waitContainerCmd("non-existing").exec(callback).awaitStatusCode();
    }

    @Test
    public void testWaitContainerAbort() throws Exception {

        CreateContainerResponse container = dockerClient.createContainerCmd("busybox").withCmd("sleep", "9999").exec();

        LOG.info("Created container: {}", container.toString());
        assertThat(container.getId(), not(isEmptyString()));

        dockerClient.startContainerCmd(container.getId()).exec();

        WaitContainerResultCallback callback = dockerClient.waitContainerCmd(container.getId()).exec(
                new WaitContainerResultCallback());

        Thread.sleep(5000);

        System.err.println("closing callback");

        callback.close();

        dockerClient.killContainerCmd(container.getId()).exec();

        InspectContainerResponse inspectContainerResponse = dockerClient.inspectContainerCmd(container.getId()).exec();
        LOG.info("Container Inspect: {}", inspectContainerResponse.toString());

        assertThat(inspectContainerResponse.getState().getRunning(), is(equalTo(false)));
    }
}

@@ -47,265 +47,511 @@
 ort 
-static org.hamcrest.MatcherAssert.assertThat;%0Aimport static org.hamcrest.Matchers.equalTo;%0Aimport static org.hamcrest.Matchers.is;%0Aimport static org.hamcrest.Matchers.isEmptyString;%0Aimport static org.hamcrest.Matchers.not
+com.github.dockerjava.api.command.CreateContainerResponse;%0Aimport com.github.dockerjava.api.command.InspectContainerResponse;%0Aimport com.github.dockerjava.api.exception.DockerClientException;%0Aimport com.github.dockerjava.api.exception.DockerException;%0Aimport com.github.dockerjava.api.exception.NotFoundException;%0Aimport com.github.dockerjava.api.model.WaitResponse;%0Aimport com.github.dockerjava.core.command.WaitContainerResultCallback
 ;%0A
-%0A
 import 
-java.lang.reflect.Method;%0A
+com.github.dockerjava.netty.AbstractNettyDockerClientTest;
 %0Aimp
@@ -796,444 +796,163 @@
 ort 
-com.github.dockerjava.api.command.CreateContainerResponse;%0Aimport com.github.dockerjava.api.command.InspectContainerResponse;%0Aimport com.github.dockerjava.api.exception.DockerException;%0Aimport com.github.dockerjava.api.exception.NotFoundException;%0Aimport com.github.dockerjava.api.model.WaitResponse;%0Aimport com.github.dockerjava.core.command.WaitContainerResultCallback;%0Aimport com.github.dockerjava.netty.AbstractNettyDockerClientTest
+java.lang.reflect.Method;%0Aimport java.util.concurrent.TimeUnit;%0A%0Aimport static org.hamcrest.MatcherAssert.assertThat;%0Aimport static org.hamcrest.Matchers.*
 ;%0A%0A@
@@ -2807,17 +2807,16 @@
        %7D
-;
 %0A       
@@ -3820,32 +3820,1165 @@
 e.toString());%0A%0A
+        assertThat(inspectContainerResponse.getState().getRunning(), is(equalTo(false)));%0A    %7D%0A%0A    @Test%0A    public void testWaitContainerTimeout() throws Exception %7B%0A%0A        CreateContainerResponse container = dockerClient.createContainerCmd(%22busybox%22).withCmd(%22sleep%22, %229999%22).exec();%0A%0A        LOG.info(%22Created container: %7B%7D%22, container.toString());%0A        assertThat(container.getId(), not(isEmptyString()));%0A%0A        dockerClient.startContainerCmd(container.getId()).exec();%0A%0A        WaitContainerResultCallback callback = dockerClient.waitContainerCmd(container.getId()).exec(%0A                new WaitContainerResultCallback());%0A        try %7B%0A            callback.awaitStatusCode(100, TimeUnit.MILLISECONDS);%0A            fail(%22Should throw exception on timeout.%22);%0A        %7D catch(DockerClientException e)%7B%0A            LOG.info(e.getMessage());%0A        %7D%0A%0A        dockerClient.killContainerCmd(container.getId()).exec();%0A%0A        InspectContainerResponse inspectContainerResponse = dockerClient.inspectContainerCmd(container.getId()).exec();%0A        LOG.info(%22Container Inspect: %7B%7D%22, inspectContainerResponse.toString());%0A%0A
         assertTh

Add test for WaitContainerResultCallback timeout in Netty implementation