package com.github.dockerjava.core.command;

import com.fasterxml.jackson.databind.JavaType;

import com.fasterxml.jackson.databind.ObjectMapper;

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.model.HostConfig;

import com.github.dockerjava.api.model.UpdateContainerResponse;

import com.github.dockerjava.client.AbstractDockerClientTest;

import com.github.dockerjava.core.RemoteApiVersion;

import org.testng.ITestResult;

import org.testng.SkipException;

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 java.io.IOException;

import java.lang.reflect.Method;

import static com.github.dockerjava.test.serdes.JSONSamples.testRoundTrip;

import static com.github.dockerjava.utils.TestUtils.getVersion;

import static org.hamcrest.CoreMatchers.is;

import static org.hamcrest.MatcherAssert.assertThat;

import static org.hamcrest.collection.IsCollectionWithSize.hasSize;

import static org.hamcrest.core.IsNull.notNullValue;

/**

* @author Kanstantsin Shautsou

*/

@Test(groups = "integration")

public class UpdateContainerCmdImplTest extends AbstractDockerClientTest {

public static final String BUSYBOX_IMAGE = "busybox";

@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(groups = "ignoreInCircleCi")

public void updateContainer() throws DockerException, IOException {

final RemoteApiVersion apiVersion = getVersion(dockerClient);

if (!apiVersion.isGreaterOrEqual(RemoteApiVersion.VERSION_1_22)) {

throw new SkipException("API version should be >= 1.22");

}

CreateContainerResponse response = dockerClient.createContainerCmd(BUSYBOX_IMAGE)

.withCmd("sleep", "9999")

.exec();

String containerId = response.getId();

dockerClient.startContainerCmd(containerId).exec();

InspectContainerResponse inspectBefore = dockerClient.inspectContainerCmd(containerId).exec();

LOG.debug("Inspect: {}", inspectBefore);

final Long memory = inspectBefore.getHostConfig().getMemory();

final UpdateContainerResponse updateResponse = dockerClient.updateContainerCmd(containerId)

.withBlkioWeight(300)

.withCpuShares(512)

.withCpuPeriod(100000)

.withCpuQuota(50000)

// .withCpusetCpus("0") // depends on env

.withCpusetMems("0")

// .withMemory(209715200L + 2L)

// .withMemorySwap(514288000L) Your kernel does not support swap limit capabilities, memory limited without swap.

// .withMemoryReservation(209715200L)

// .withKernelMemory(52428800) Can not update kernel memory to a running container, please stop it first.

.exec();

// true only on docker toolbox (1.10.1)

// assertThat(updateResponse.getWarnings(), hasSize(1));

// assertThat(updateResponse.getWarnings().get(0),

// is("Your kernel does not support Block I/O weight. Weight discarded."));

InspectContainerResponse inspectAfter = dockerClient.inspectContainerCmd(containerId).exec();

final HostConfig afterHostConfig = inspectAfter.getHostConfig();

// assertThat(afterHostConfig.getMemory(), is(209715200L + 2L));

// assertThat(afterHostConfig.getBlkioWeight(), is(300));

assertThat(afterHostConfig.getCpuShares(), is(512));

assertThat(afterHostConfig.getCpuPeriod(), is(100000));

assertThat(afterHostConfig.getCpuQuota(), is(50000));

assertThat(afterHostConfig.getCpusetMems(), is("0"));

// assertThat(afterHostConfig.getMemoryReservation(), is(209715200L));

// assertThat(afterHostConfig.getMemorySwap(), is(514288000L));

}

@Test(enabled = false, description = "impossible to serder because model bundled in cmd")

public void serDerDocs1() throws IOException {

final ObjectMapper mapper = new ObjectMapper();

final JavaType type = mapper.getTypeFactory().uncheckedSimpleType(UpdateContainerCmdImpl.class);

final UpdateContainerCmdImpl upd = testRoundTrip(RemoteApiVersion.VERSION_1_22,

"/containers/container/update/docs.json",

type

);

assertThat(upd, notNullValue());

}

}