add task label filter test case

fengxx · fengxx · commit 0ad1f6d7e181 · 2018-03-09T17:22:09.000+08:00
diff --git a/src/test/java/com/github/dockerjava/cmd/swarm/ListTasksCmdExecIT.java b/src/test/java/com/github/dockerjava/cmd/swarm/ListTasksCmdExecIT.java
@@ -14,7 +14,9 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.util.Collections;
 import java.util.List;
+import java.util.Map;
 
 import static com.github.dockerjava.junit.DockerRule.DEFAULT_IMAGE;
 import static org.hamcrest.MatcherAssert.assertThat;
@@ -24,14 +26,16 @@
 public class ListTasksCmdExecIT extends SwarmCmdIT {
     public static final Logger LOG = LoggerFactory.getLogger(CreateServiceCmdExecIT.class);
     private static final String SERVICE_NAME = "inspect_task";
+    private static final String TASK_LABEL_KEY = "com.github.dockerjava.usage";
+    private static final String TASK_LABEL_VALUE = "test";
 
     @Test
     public void testListTasks() throws DockerException {
         dockerRule.getClient().initializeSwarmCmd(new SwarmSpec())
                 .withListenAddr("127.0.0.1")
                 .withAdvertiseAddr("127.0.0.1")
                 .exec();
-
+        Map<String, String> taskLabels = Collections.singletonMap(TASK_LABEL_KEY, TASK_LABEL_VALUE);
         CreateServiceResponse response = dockerRule.getClient().createServiceCmd(new ServiceSpec()
                 .withName(SERVICE_NAME)
                 .withMode(new ServiceModeConfig().withReplicated(
@@ -40,7 +44,7 @@ public void testListTasks() throws DockerException {
                 ))
                 .withTaskTemplate(new TaskSpec()
                         .withContainerSpec(new ContainerSpec()
-                                .withImage(DEFAULT_IMAGE))))
+                                .withImage(DEFAULT_IMAGE))).withLabels(taskLabels))
                 .exec();
         String serviceId = response.getId();
         //filtering with service id
@@ -60,6 +64,11 @@ public void testListTasks() throws DockerException {
         //filtering with state
         tasks = dockerRule.getClient().listTasksCmd().withStateFilter(TaskState.RUNNING).exec();
         assertThat(tasks, hasSize(2));
+        //filter labels
+        tasks = dockerRule.getClient().listTasksCmd().withLabelFilter(taskLabels).exec();
+        assertThat(tasks, hasSize(2));
+        tasks = dockerRule.getClient().listTasksCmd().withLabelFilter(TASK_LABEL_KEY + "=" + TASK_LABEL_VALUE).exec();
+        assertThat(tasks, hasSize(2));
         dockerRule.getClient().removeServiceCmd(SERVICE_NAME).exec();
     }
 }
