1 files changed, 43 insertions, 15 deletions

diff --git a/rest-services/cbs-client/src/test/java/org/onap/dcaegen2/services/sdk/rest/services/cbs/client/impl/CbsClientImplIT.java b/rest-services/cbs-client/src/test/java/org/onap/dcaegen2/services/sdk/rest/services/cbs/client/impl/CbsClientImplIT.java
index 761cc5c1..309bb62f 100644
--- a/rest-services/cbs-client/src/test/java/org/onap/dcaegen2/services/sdk/rest/services/cbs/client/impl/CbsClientImplIT.java
+++ b/rest-services/cbs-client/src/test/java/org/onap/dcaegen2/services/sdk/rest/services/cbs/client/impl/CbsClientImplIT.java
@@ -24,14 +24,17 @@ import static org.onap.dcaegen2.services.sdk.rest.services.cbs.client.impl.Dummy
 import static org.onap.dcaegen2.services.sdk.rest.services.cbs.client.impl.DummyHttpServer.sendString;
 
 import com.google.gson.JsonObject;
+import io.vavr.collection.Stream;
 import java.time.Duration;
 import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
+import org.onap.dcaegen2.services.sdk.rest.services.model.logging.RequestDiagnosticContext;
 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.api.CbsClient;
 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.api.CbsClientFactory;
 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.api.EnvProperties;
 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.api.ImmutableEnvProperties;
+import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
 import reactor.test.StepVerifier;
 
@@ -41,21 +44,30 @@ import reactor.test.StepVerifier;
  */
 class CbsClientImplIT {
 
-    private static final String CONSUL_RESP = "[\n"
+    private static final String CONSUL_RESPONSE = "[\n"
             + "    {\n"
             + "        \"ServiceAddress\": \"HOST\",\n"
             + "        \"ServiceName\": \"the_cbs\",\n"
             + "        \"ServicePort\": PORT\n"
             + "    }\n"
             + "]\n";
-    private static final String RES_CONFIG = "/sample_config.json";
+    private static final String SAMPLE_CONFIG = "/sample_config.json";
+    private static final String SAMPLE_CONFIG_KEY = "keystore.path";
+    private static final String EXPECTED_CONFIG_VALUE = "/var/run/security/keystore.p12";
+    private static EnvProperties sampleEnvironment;
     private static DummyHttpServer server;
 
     @BeforeAll
     static void setUp() {
         server = DummyHttpServer.start(routes ->
                 routes.get("/v1/catalog/service/the_cbs", (req, resp) -> sendString(resp, lazyConsulResponse()))
-                        .get("/service_component/dcae-component", (req, resp) -> sendResource(resp, RES_CONFIG)));
+                        .get("/service_component/dcae-component", (req, resp) -> sendResource(resp, SAMPLE_CONFIG)));
+        sampleEnvironment = ImmutableEnvProperties.builder()
+                .appName("dcae-component")
+                .cbsName("the_cbs")
+                .consulHost(server.host())
+                .consulPort(server.port())
+                .build();
     }
 
     @AfterAll
@@ -64,28 +76,44 @@ class CbsClientImplIT {
     }
 
     @Test
-    void testCbsClient() {
+    void testCbsClientWithSingleCall() {
         // given
-        final EnvProperties env = ImmutableEnvProperties.builder()
-                .appName("dcae-component")
-                .cbsName("the_cbs")
-                .consulHost(server.host())
-                .consulPort(server.port())
-                .build();
-        final Mono<CbsClient> sut = CbsClientFactory.createCbsClient(env);
+        final Mono<CbsClient> sut = CbsClientFactory.createCbsClient(sampleEnvironment);
+        final RequestDiagnosticContext diagnosticContext = RequestDiagnosticContext.create();
+
+        // when
+        final Mono<JsonObject> result = sut.flatMap(cbsClient -> cbsClient.get(diagnosticContext));
+
+        // then
+        StepVerifier.create(result.map(this::sampleConfigValue))
+                .expectNext(EXPECTED_CONFIG_VALUE)
+                .expectComplete()
+                .verify(Duration.ofSeconds(5));
+    }
+
+    @Test
+    void testCbsClientWithPeriodicCall() {
+        // given
+        final Mono<CbsClient> sut = CbsClientFactory.createCbsClient(sampleEnvironment);
+        final RequestDiagnosticContext diagnosticContext = RequestDiagnosticContext.create();
 
         // when
-        final Mono<JsonObject> result = sut.flatMap(CbsClient::get);
+        final Flux<JsonObject> result = sut.flatMapMany(cbsClient -> cbsClient.get(diagnosticContext, Duration.ZERO, Duration.ofMillis(10)));
 
         // then
-        StepVerifier.create(result.map(obj -> obj.get("keystore.path").getAsString()))
-                .expectNext("/var/run/security/keystore.p12")
+        final int itemsToTake = 5;
+        StepVerifier.create(result.take(itemsToTake).map(this::sampleConfigValue))
+                .expectNextSequence(Stream.of(EXPECTED_CONFIG_VALUE).cycle(itemsToTake))
                 .expectComplete()
                 .verify(Duration.ofSeconds(5));
     }
 
+    private String sampleConfigValue(JsonObject obj) {
+        return obj.get(SAMPLE_CONFIG_KEY).getAsString();
+    }
+
     private static Mono<String> lazyConsulResponse() {
-        return Mono.just(CONSUL_RESP)
+        return Mono.just(CONSUL_RESPONSE)
                 .map(CbsClientImplIT::processConsulResponseTemplate);
     }