public class MockA1ClientFactory extends A1ClientFactory {
private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
private final Map<String, MockA1Client> clients = new HashMap<>();
- private final PolicyTypes policyTypes;
+ private PolicyTypes policyTypes;
private Duration asynchDelay = Duration.ofSeconds(0);
public MockA1ClientFactory(PolicyTypes policyTypes) {
return clients.get(ricName);
}
+ public void setPolicyTypes(PolicyTypes policyTypes) {
+ this.policyTypes = policyTypes;
+ }
+
/**
* Simulate network latency. The REST responses will be generated by separate
* threads
this.asynchDelay = delay;
}
+ public void reset() {
+ this.asynchDelay = Duration.ofSeconds(0);
+ clients.clear();
+ }
+
+ public PolicyTypes getPolicyTypes() {
+ return this.policyTypes;
+ }
+
}