27
27
import org .testcontainers .containers .wait .strategy .WaitStrategy ;
28
28
import org .testcontainers .images .builder .Transferable ;
29
29
import org .testcontainers .utility .DockerImageName ;
30
- import org .testcontainers .utility .MountableFile ;
31
30
import org .yaml .snakeyaml .Yaml ;
32
31
33
32
import java .io .IOException ;
41
40
import java .util .Map ;
42
41
import java .util .Set ;
43
42
44
- import static io .dapr .testcontainers .ContainerConstants .DAPR_PLACEMENT_IMAGE_TAG ;
45
- import static io .dapr .testcontainers .ContainerConstants .DAPR_RUNTIME_IMAGE_TAG ;
43
+ import static io .dapr .testcontainers .DaprContainerConstants .DAPR_PLACEMENT_IMAGE_TAG ;
44
+ import static io .dapr .testcontainers .DaprContainerConstants .DAPR_RUNTIME_IMAGE_TAG ;
45
+ import static io .dapr .testcontainers .DaprContainerConstants .DAPR_SCHEDULER_IMAGE_TAG ;
46
46
47
47
public class DaprContainer extends GenericContainer <DaprContainer > {
48
48
private static final Logger LOGGER = LoggerFactory .getLogger (DaprContainer .class );
49
49
private static final int DAPRD_DEFAULT_HTTP_PORT = 3500 ;
50
50
private static final int DAPRD_DEFAULT_GRPC_PORT = 50001 ;
51
51
private static final DaprProtocol DAPR_PROTOCOL = DaprProtocol .HTTP ;
52
52
private static final DockerImageName DEFAULT_IMAGE_NAME =
53
- DockerImageName .parse ("daprio/daprd" );
53
+ DockerImageName .parse (DAPR_RUNTIME_IMAGE_TAG );
54
54
private static final Yaml YAML_MAPPER = YamlMapperFactory .create ();
55
55
private static final YamlConverter <Component > COMPONENT_CONVERTER = new ComponentYamlConverter (YAML_MAPPER );
56
56
private static final YamlConverter <Subscription > SUBSCRIPTION_CONVERTER = new SubscriptionYamlConverter (YAML_MAPPER );
@@ -64,12 +64,12 @@ public class DaprContainer extends GenericContainer<DaprContainer> {
64
64
private final Set <Component > components = new HashSet <>();
65
65
private final Set <Subscription > subscriptions = new HashSet <>();
66
66
private final Set <HttpEndpoint > httpEndpoints = new HashSet <>();
67
- private final DaprLogLevel daprLogLevel = DaprLogLevel .INFO ;
68
- private final String appChannelAddress = "localhost" ;
69
- private final String placementService = "placement" ;
70
- private final String schedulerService = "scheduler" ;
71
- private final String placementDockerImageName = DAPR_PLACEMENT_IMAGE_TAG ;
72
- private final String schedulerDockerImageName = "daprio/scheduler" ;
67
+ private DaprLogLevel daprLogLevel = DaprLogLevel .INFO ;
68
+ private String appChannelAddress = "localhost" ;
69
+ private String placementService = "placement" ;
70
+ private String schedulerService = "scheduler" ;
71
+ private String placementDockerImageName = DAPR_PLACEMENT_IMAGE_TAG ;
72
+ private String schedulerDockerImageName = DAPR_SCHEDULER_IMAGE_TAG ;
73
73
74
74
private Configuration configuration ;
75
75
private DaprPlacementContainer placementContainer ;
@@ -141,6 +141,11 @@ public DaprContainer withPlacementService(String placementService) {
141
141
return this ;
142
142
}
143
143
144
+ public DaprContainer withSchedulerService (String schedulerService ) {
145
+ this .schedulerService = schedulerService ;
146
+ return this ;
147
+ }
148
+
144
149
public DaprContainer withAppName (String appName ) {
145
150
this .appName = appName ;
146
151
return this ;
@@ -166,6 +171,11 @@ public DaprContainer withPlacementImage(String placementDockerImageName) {
166
171
return this ;
167
172
}
168
173
174
+ public DaprContainer withSchedulerImage (String schedulerDockerImageName ) {
175
+ this .schedulerDockerImageName = schedulerDockerImageName ;
176
+ return this ;
177
+ }
178
+
169
179
public DaprContainer withReusablePlacement (boolean shouldReusePlacement ) {
170
180
this .shouldReusePlacement = shouldReusePlacement ;
171
181
return this ;
0 commit comments