forked from docker-java/docker-java
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathAbstractDockerClientTest.java
More file actions
146 lines (115 loc) · 3.66 KB
/
Copy pathAbstractDockerClientTest.java
File metadata and controls
146 lines (115 loc) · 3.66 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
package com.github.dockerjava.client;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringWriter;
import java.lang.reflect.Method;
import java.net.DatagramSocket;
import java.net.ServerSocket;
import org.apache.commons.io.IOUtils;
import org.apache.commons.io.LineIterator;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.testng.Assert;
import org.testng.ITestResult;
import com.github.dockerjava.api.DockerClient;
import com.github.dockerjava.api.DockerException;
import com.github.dockerjava.jaxrs.DockerClientBuilder;
import com.github.dockerjava.jaxrs.TestDockerCmdExecFactory;
public abstract class AbstractDockerClientTest extends Assert {
public static final Logger LOG = LoggerFactory
.getLogger(AbstractDockerClientTest.class);
protected DockerClient dockerClient;
protected TestDockerCmdExecFactory dockerCmdExecFactory = new TestDockerCmdExecFactory();
public void beforeTest() {
LOG.info("======================= BEFORETEST =======================");
LOG.info("Connecting to Docker server");
dockerClient = DockerClientBuilder.getInstance()
.withDockerCmdExecFactory(dockerCmdExecFactory)
.build();
LOG.info("Pulling image 'busybox'");
// need to block until image is pulled completely
asString(dockerClient.pullImageCmd("busybox").withTag("latest").exec());
assertNotNull(dockerClient);
LOG.info("======================= END OF BEFORETEST =======================\n\n");
}
public void afterTest() {
LOG.info("======================= END OF AFTERTEST =======================");
}
public void beforeMethod(Method method) {
LOG.info(String
.format("################################## STARTING %s ##################################",
method.getName()));
}
public void afterMethod(ITestResult result) {
for (String container : dockerCmdExecFactory.getContainerNames()) {
LOG.info("Cleaning up temporary container {}", container);
try {
dockerClient.removeContainerCmd(container).withForce().exec();
} catch (DockerException ignore) {
ignore.printStackTrace();
}
}
for (String image : dockerCmdExecFactory.getImageNames()) {
LOG.info("Cleaning up temporary image with {}", image);
try {
dockerClient.removeImageCmd(image).withForce().exec();
} catch (DockerException ignore) {
ignore.printStackTrace();
}
}
LOG.info(
"################################## END OF {} ##################################\n",
result.getName());
}
protected String asString(InputStream response) {
StringWriter logwriter = new StringWriter();
try {
LineIterator itr = IOUtils.lineIterator(
response, "UTF-8");
while (itr.hasNext()) {
String line = itr.next();
logwriter.write(line + (itr.hasNext() ? "\n" : ""));
//LOG.info("line: "+line);
}
return logwriter.toString();
} catch (IOException e) {
throw new RuntimeException(e);
} finally {
IOUtils.closeQuietly(response);
}
}
// UTIL
/**
* Checks to see if a specific port is available.
*
* @param port
* the port to check for availability
*/
public static boolean available(int port) {
if (port < 1100 || port > 60000) {
throw new IllegalArgumentException("Invalid start port: " + port);
}
ServerSocket ss = null;
DatagramSocket ds = null;
try {
ss = new ServerSocket(port);
ss.setReuseAddress(true);
ds = new DatagramSocket(port);
ds.setReuseAddress(true);
return true;
} catch (IOException e) {
} finally {
if (ds != null) {
ds.close();
}
if (ss != null) {
try {
ss.close();
} catch (IOException e) {
/* should not be thrown */
}
}
}
return false;
}
}