|
2 | 2 |
|
3 | 3 | import com.arakelian.docker.junit.DockerRule; |
4 | 4 | import com.arakelian.docker.junit.model.ImmutableDockerConfig; |
5 | | -import com.spotify.docker.client.DefaultDockerClient; |
6 | 5 | import org.junit.Assume; |
7 | 6 | import org.junit.rules.TestRule; |
8 | 7 | import org.junit.runner.Description; |
9 | 8 | import org.junit.runners.model.Statement; |
| 9 | +import repackaged.com.arakelian.docker.junit.com.github.dockerjava.api.model.ExposedPort; |
| 10 | +import repackaged.com.arakelian.docker.junit.com.github.dockerjava.api.model.PortBinding; |
| 11 | +import repackaged.com.arakelian.docker.junit.com.github.dockerjava.core.DockerClientImpl; |
| 12 | +import repackaged.com.arakelian.docker.junit.com.github.dockerjava.okhttp.OkHttpDockerCmdExecFactory; |
10 | 13 |
|
11 | 14 | public class ToxiproxyRule implements TestRule { |
12 | 15 |
|
13 | 16 | private final DockerRule wrappedRule; |
14 | 17 |
|
15 | 18 | public ToxiproxyRule() { |
16 | 19 | wrappedRule = new DockerRule(ImmutableDockerConfig.builder() |
17 | | - .image("shopify/toxiproxy:2.1.3") |
18 | | - .name("text-toxiproxy") |
19 | | - .ports("8474") |
| 20 | + .image("shopify/toxiproxy:2.1.4") |
20 | 21 | .addStartedListener(container -> { |
21 | | - container.waitForPort("8474/tcp"); |
| 22 | + container.waitForPort(ExposedPort.tcp(8474)); |
| 23 | + }) |
| 24 | + .addHostConfigConfigurer(hostConfig -> { |
| 25 | + hostConfig.withPortBindings(PortBinding.parse("8474")); |
22 | 26 | }) |
23 | | - .alwaysRemoveContainer(true) |
24 | 27 | .build()); |
25 | | - |
26 | 28 | } |
27 | 29 |
|
28 | 30 | public ToxiproxyClient getToxiproxyClient() { |
29 | | - int mappedPort = wrappedRule.getContainer().getPortBinding("8474/tcp").getPort(); |
| 31 | + int mappedPort = wrappedRule.getContainer().getSimpleBinding(ExposedPort.tcp(8474)).getPort(); |
30 | 32 | ToxiproxyClient tp = new ToxiproxyClient("localhost", mappedPort); |
31 | 33 | return tp; |
32 | 34 | } |
33 | 35 |
|
34 | 36 | @Override |
35 | 37 | public Statement apply(Statement statement, Description description) { |
36 | 38 | try { |
37 | | - DefaultDockerClient client = DefaultDockerClient.fromEnv().connectTimeoutMillis(5000L).readTimeoutMillis(20000L).build(); |
38 | | - client.ping(); |
39 | | - client.close(); |
| 39 | + DockerClientImpl.getInstance().withDockerCmdExecFactory(new OkHttpDockerCmdExecFactory()).pingCmd().exec(); |
40 | 40 | } catch (Exception e) { |
41 | 41 | Assume.assumeNoException(e); |
42 | 42 | } |
43 | | - |
44 | 43 | return wrappedRule.apply(statement, description); |
45 | 44 | } |
46 | 45 | } |
0 commit comments