From dd516e92c235adf4a885c6486c8fc2700e8bd149 Mon Sep 17 00:00:00 2001
From: Zixuan Liu <nodeces@gmail.com>
Date: Mon, 20 May 2024 12:03:23 +0800
Subject: [PATCH] Add TagMojo test

---
 .../io/fabric8/maven/docker/TagMojoTest.java  | 82 +++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 src/test/java/io/fabric8/maven/docker/TagMojoTest.java

diff --git a/src/test/java/io/fabric8/maven/docker/TagMojoTest.java b/src/test/java/io/fabric8/maven/docker/TagMojoTest.java
new file mode 100644
index 000000000..a5a1711fb
--- /dev/null
+++ b/src/test/java/io/fabric8/maven/docker/TagMojoTest.java
@@ -0,0 +1,82 @@
+package io.fabric8.maven.docker;
+
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import io.fabric8.maven.docker.access.AuthConfig;
+import io.fabric8.maven.docker.access.AuthConfigList;
+import io.fabric8.maven.docker.access.DockerAccessException;
+import io.fabric8.maven.docker.assembly.DockerAssemblyManager;
+import io.fabric8.maven.docker.config.AttestationConfiguration;
+import io.fabric8.maven.docker.config.BuildImageConfiguration;
+import io.fabric8.maven.docker.config.BuildXConfiguration;
+import io.fabric8.maven.docker.config.ImageConfiguration;
+import io.fabric8.maven.docker.service.BuildService;
+import io.fabric8.maven.docker.service.BuildXService;
+import io.fabric8.maven.docker.service.ImagePullManager;
+import io.fabric8.maven.docker.util.ImageName;
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Path;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
+import org.apache.maven.plugin.MojoExecutionException;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.junit.jupiter.api.io.TempDir;
+import org.mockito.ArgumentCaptor;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.junit.jupiter.MockitoExtension;
+
+@ExtendWith(MockitoExtension.class)
+class TagMojoTest extends MojoTestBase {
+
+    private static final String NON_NATIVE_PLATFORM = "linux/amd64";
+    private static final String NATIVE_PLATFORM = "linux/arm64";
+
+    @InjectMocks
+    private TagMojo tagMojo;
+
+    void tag(boolean skipTag, boolean buildx) throws IOException, MojoExecutionException {
+        givenMavenProject(tagMojo);
+
+        ImageConfiguration imageConfiguration = singleImageConfiguration(builder -> {
+            builder.skipTag(skipTag);
+            if (buildx) {
+                ArrayList<String> platforms = new ArrayList<>();
+                platforms.add(NON_NATIVE_PLATFORM);
+                platforms.add(NATIVE_PLATFORM);
+                BuildXConfiguration buildXConfiguration = new BuildXConfiguration.Builder().platforms(platforms).build();
+                builder.buildx(buildXConfiguration);
+            }
+        });
+        givenResolvedImages(tagMojo, Collections.singletonList(imageConfiguration));
+
+        tagMojo.executeInternal(serviceHub);
+
+        verify(buildService, times(skipTag || buildx ? 0 : 1)).tagImage(any(), any(), any(), any());
+    }
+
+    @Test
+    void skipTag() throws IOException, MojoExecutionException {
+        tag(true, false);
+    }
+
+    @Test
+    void tag() throws IOException, MojoExecutionException {
+        tag(false, false);
+    }
+
+    @Test
+    void tagWhenUsingBuildx() throws IOException, MojoExecutionException {
+        tag(false, true);
+    }
+}