package com.github.dockerjava.core.command;

import static org.hamcrest.MatcherAssert.assertThat;

import static org.hamcrest.Matchers.isEmptyOrNullString;

import static org.hamcrest.Matchers.not;

import java.io.InputStream;

import java.lang.reflect.Method;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.nio.file.StandardOpenOption;

import org.apache.commons.compress.archivers.tar.TarArchiveEntry;

import org.apache.commons.compress.archivers.tar.TarArchiveInputStream;

import org.apache.commons.io.IOUtils;

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.exception.NotFoundException;

import com.github.dockerjava.api.command.CreateContainerResponse;

import com.github.dockerjava.client.AbstractDockerClientTest;

import com.github.dockerjava.core.util.CompressArchiveUtil;

@Test(groups = "integration")

public class CopyArchiveFromContainerCmdImplTest extends AbstractDockerClientTest {

@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 copyFromContainer() throws Exception {

// TODO extract this into a shared method

CreateContainerResponse container = dockerClient.createContainerCmd("busybox")

.withName("docker-java-itest-copyFromContainer").withCmd("touch", "/copyFromContainer").exec();

LOG.info("Created container: {}", container);

assertThat(container.getId(), not(isEmptyOrNullString()));

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

InputStream response = dockerClient.copyArchiveFromContainerCmd(container.getId(), "/copyFromContainer").exec();

Boolean bytesAvailable = response.available() > 0;

assertTrue(bytesAvailable, "The file was not copied from the container.");

// read the stream fully. Otherwise, the underlying stream will not be closed.

String responseAsString = asString(response);

assertNotNull(responseAsString);

assertTrue(responseAsString.length() > 0);

}

@Test

public void copyFromNonExistingContainer() throws Exception {

try {

dockerClient.copyArchiveFromContainerCmd("non-existing", "/test").exec();

fail("expected NotFoundException");

} catch (NotFoundException ignored) {

}

}

@Test

public void copyFromContainerBinaryFile() throws Exception {

CreateContainerResponse container = dockerClient.createContainerCmd("busybox")

.withName("docker-java-itest-copyFromContainerBinaryFile").exec();

LOG.info("Created container: {}", container);

assertThat(container.getId(), not(isEmptyOrNullString()));

Path temp = Files.createTempFile("", ".tar.gz");

Path binaryFile = Paths.get("src/test/resources/testCopyFromArchive/binary.dat");

CompressArchiveUtil.tar(binaryFile, temp, true, false);

try (InputStream uploadStream = Files.newInputStream(temp)) {

dockerClient.copyArchiveToContainerCmd(container.getId()).withTarInputStream(uploadStream).exec();

}

InputStream response = dockerClient.copyArchiveFromContainerCmd(container.getId(), "/binary.dat").exec();

Boolean bytesAvailable = response.available() > 0;

assertTrue(bytesAvailable, "The file was not copied from the container.");

try (TarArchiveInputStream tarInputStream = new TarArchiveInputStream(response)) {

TarArchiveEntry nextTarEntry = tarInputStream.getNextTarEntry();

assertEquals(nextTarEntry.getName(), "binary.dat");

try (InputStream binaryFileInputStream = Files.newInputStream(binaryFile, StandardOpenOption.READ)) {

assertTrue(IOUtils.contentEquals(binaryFileInputStream, tarInputStream));

}

assertNull(tarInputStream.getNextTarEntry(), "Nothing except binary.dat is expected to be copied.");

}

}

}