Bugfix in RIC synchronization
[nonrtric.git] / policy-agent / src / test / java / org / oransc / policyagent / utils / MockA1ClientFactory.java
1 /*-
2  * ========================LICENSE_START=================================
3  * O-RAN-SC
4  * %%
5  * Copyright (C) 2019 Nordix Foundation
6  * %%
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ========================LICENSE_END===================================
19  */
20
21 package org.oransc.policyagent.utils;
22
23 import static org.mockito.Mockito.mock;
24 import static org.mockito.Mockito.spy;
25
26 import java.lang.invoke.MethodHandles;
27 import java.time.Duration;
28 import java.util.HashMap;
29 import java.util.Map;
30
31 import org.oransc.policyagent.clients.A1Client;
32 import org.oransc.policyagent.clients.A1ClientFactory;
33 import org.oransc.policyagent.configuration.ApplicationConfig;
34 import org.oransc.policyagent.repository.PolicyTypes;
35 import org.oransc.policyagent.repository.Ric;
36 import org.slf4j.Logger;
37 import org.slf4j.LoggerFactory;
38 import reactor.core.publisher.Mono;
39
40 public class MockA1ClientFactory extends A1ClientFactory {
41     private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
42     private final Map<String, MockA1Client> clients = new HashMap<>();
43     private final PolicyTypes policyTypes;
44     private Duration asynchDelay = Duration.ofSeconds(0);
45
46     public MockA1ClientFactory(PolicyTypes policyTypes) {
47         super(mock(ApplicationConfig.class));
48         this.policyTypes = policyTypes;
49     }
50
51     @Override
52     public Mono<A1Client> createA1Client(Ric ric) {
53         return Mono.just(getOrCreateA1Client(ric.name()));
54     }
55
56     public MockA1Client getOrCreateA1Client(String ricName) {
57         if (!clients.containsKey(ricName)) {
58             logger.debug("Creating client for RIC: {}", ricName);
59             MockA1Client client = spy(new MockA1Client(policyTypes, asynchDelay));
60             clients.put(ricName, client);
61         }
62         return clients.get(ricName);
63     }
64
65     /**
66      * Simulate network latency. The REST responses will be generated by separate
67      * threads
68      * 
69      * @param delay the delay between the request and the response
70      */
71     public void setResponseDelay(Duration delay) {
72         this.asynchDelay = delay;
73     }
74
75 }