package com.baeldung.dockerapi;

import com.github.dockerjava.api.DockerClient;
import com.github.dockerjava.api.command.InspectImageResponse;
import com.github.dockerjava.api.model.Image;
import com.github.dockerjava.api.model.SearchItem;
import com.github.dockerjava.core.DockerClientBuilder;
import com.github.dockerjava.core.command.BuildImageResultCallback;
import com.github.dockerjava.core.command.PullImageResultCallback;
import com.github.dockerjava.core.command.PushImageResultCallback;
import org.junit.BeforeClass;
import org.junit.Test;

import java.io.File;
import java.util.List;
import java.util.concurrent.TimeUnit;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.greaterThan;
import static org.hamcrest.Matchers.lessThan;
import static org.hamcrest.Matchers.not;
import static org.hamcrest.core.Is.is;

public class ImageLiveTest {

    private static DockerClient dockerClient;

    @BeforeClass
    public static void setup() {
        dockerClient = DockerClientBuilder.getInstance().build();
    }

    @Test
    public void whenListingImages_thenReturnNonEmptyList() {

        //when
        List<Image> images = dockerClient.listImagesCmd().exec();

        //then
        assertThat(images.size(), is(not(0)));
    }

    @Test
    public void whenListingImagesWithIntermediateImages_thenReturnNonEmptyList() {

        //when
        List<Image> images = dockerClient.listImagesCmd()
                .withShowAll(true).exec();

        //then
        assertThat(images.size(), is(not(0)));
    }

    @Test
    public void whenListingDanglingImages_thenReturnNonNullList() {

        //when
        List<Image> images = dockerClient.listImagesCmd()
                .withDanglingFilter(true).exec();

        //then
        assertThat(images, is(not(null)));
    }

    @Test
    public void whenBuildingImage_thenMustReturnImageId() {

        //when
        String imageId = dockerClient.buildImageCmd()
                .withDockerfile(new File("src/test/resources/dockerapi/Dockerfile"))
                .withPull(true)
                .withNoCache(true)
                .withTag("alpine:git")
                .exec(new BuildImageResultCallback())
                .awaitImageId();

        //then
        assertThat(imageId, is(not(null)));
    }

    @Test
    public void givenListOfImages_whenInspectImage_thenMustReturnObject() {

        //given
        List<Image> images = dockerClient.listImagesCmd().exec();
        Image image = images.get(0);

        //when
        InspectImageResponse imageResponse
                = dockerClient.inspectImageCmd(image.getId()).exec();

        //then
        assertThat(imageResponse.getId(), is(image.getId()));
    }

    @Test
    public void givenListOfImages_whenTagImage_thenListMustIncrement() {

        //given
        List<Image> images = dockerClient.listImagesCmd().exec();
        Image image = images.get(0);

        //when
        dockerClient.tagImageCmd(image.getId(), "baeldung/alpine", "3.6.v2").exec();

        //then
        List<Image> imagesNow = dockerClient.listImagesCmd().exec();
        assertThat(imagesNow.size(), is(greaterThan(images.size())));
    }

    public void pushingAnImage() throws InterruptedException {

        dockerClient.pushImageCmd("baeldung/alpine")
                .withTag("3.6.v2")
                .exec(new PushImageResultCallback())
                .awaitCompletion(90, TimeUnit.SECONDS);
    }

    @Test
    public void whenPullingImage_thenImageListNotEmpty() throws InterruptedException {

        //when
        dockerClient.pullImageCmd("alpine")
                .withTag("latest")
                .exec(new PullImageResultCallback())
                .awaitCompletion(30, TimeUnit.SECONDS);

        //then
        List<Image> images = dockerClient.listImagesCmd().exec();
        assertThat(images.size(), is(not(0)));
    }

    @Test
    public void whenRemovingImage_thenImageListDecrease() {

        //when
        List<Image> images = dockerClient.listImagesCmd().exec();
        Image image = images.get(0);
        dockerClient.removeImageCmd(image.getId()).exec();

        //then
        List<Image> imagesNow = dockerClient.listImagesCmd().exec();
        assertThat(imagesNow.size(), is(lessThan(images.size())));
    }

    @Test
    public void whenSearchingImage_thenMustReturn25Items() {

        //when
        List<SearchItem> items = dockerClient.searchImagesCmd("Java").exec();

        //then
        assertThat(items.size(), is(25));
    }
}