ms/blueprintsprocessor/functions/resource-resolution/src/test/kotlin/org/onap/ccsdk/cds/blueprintsprocessor/functions/resource/resolution/mock/MockBlueprintWebClientService.kt

   1 /*
   2  * Copyright © 2019 IBM, Bell Canada.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package org.onap.ccsdk.cds.blueprintsprocessor.functions.resource.resolution.mock
  17
  18 import org.apache.http.message.BasicHeader
  19 import org.mockserver.integration.ClientAndServer
  20 import org.mockserver.model.Header
  21 import org.mockserver.model.HttpRequest.request
  22 import org.mockserver.model.HttpResponse.response
  23 import org.onap.ccsdk.cds.blueprintsprocessor.rest.RestClientProperties
  24 import org.onap.ccsdk.cds.blueprintsprocessor.rest.service.BlueprintWebClientService
  25 import org.springframework.http.HttpHeaders
  26 import org.springframework.http.MediaType
  27 import java.nio.charset.Charset
  28 import java.util.*
  29
  30 class MockBlueprintWebClientService(private var restClientProperties: RestClientProperties): BlueprintWebClientService {
  31     private var mockServer: ClientAndServer
  32     private var port: String = if (restClientProperties.url.split(":")[2].isEmpty()) "8080"
  33                             else restClientProperties.url.split(":")[2]
  34     private var headers: Map<String, String>
  35
  36     init {
  37         mockServer  = ClientAndServer.startClientAndServer(port.toInt())
  38         headers =  defaultHeaders()
  39
  40         // Create expected requests and responses
  41         setRequest("GET", "/aai/v14/network/generic-vnfs/generic-vnf/123456")
  42         setRequest("GET", "/config/GENERIC-RESOURCE-API:services/service/10/service-data/vnfs/vnf/123456/" +
  43                 "vnf-data/vnf-topology/vnf-parameters-data/param/vnf_name")
  44         setRequestWithPayload("PUT", "/query",
  45                 "{\r\n\"start\": \"\\/nodes\\/vf-modules?vf-module-name=vf-module-name\",\r\n\"query\": \"\\/query\\/related-to?startingNodeType=vf-module&relatedToNodeType=generic-vnf\"\r\n}")
  46     }
  47
  48     override fun defaultHeaders(): Map<String, String> {
  49         val encodedCredentials = this.setBasicAuth("admin", "aaiTest")
  50         return mapOf(
  51                 HttpHeaders.CONTENT_TYPE to MediaType.APPLICATION_JSON_VALUE,
  52                 HttpHeaders.ACCEPT to MediaType.APPLICATION_JSON_VALUE,
  53                 HttpHeaders.AUTHORIZATION to "Basic $encodedCredentials")
  54     }
  55
  56     override fun host(uri: String): String {
  57         return restClientProperties.url + uri
  58     }
  59
  60     fun tearDown() {
  61         mockServer.close()
  62     }
  63
  64     override fun exchangeResource(method: String, path: String, payload: String): String {
  65         val header = arrayOf(BasicHeader(HttpHeaders.AUTHORIZATION, headers[HttpHeaders.AUTHORIZATION]))
  66         return when (method) {
  67             "POST" -> {
  68                 post(path, payload, header)
  69             }
  70             "PUT" -> {
  71                 put(path, payload, header)
  72             }
  73             else -> {
  74                 get(path, header)
  75             }
  76         }
  77     }
  78
  79     private fun setRequest(method: String, path: String) {
  80         val requestResponse = when (method) {
  81             "POST" -> {
  82                 "Post response"
  83             }
  84             "PUT" -> {
  85                 "Put response"
  86             }
  87             else -> {
  88                 "Get response"
  89             }
  90
  91         }
  92         mockServer.`when`(request().withHeaders(Header(HttpHeaders.AUTHORIZATION, headers[HttpHeaders.AUTHORIZATION]))
  93                         .withMethod(method)
  94                         .withPath(path)
  95         ).respond(response().withStatusCode(200).withBody("{\"aai-resource\":\"$requestResponse\"}"))
  96     }
  97
  98     private fun setRequestWithPayload(method: String, path: String, payload: String) {
  99         val requestResponse = when (method) {
 100             "POST" -> {
 101                 "Post response"
 102             }
 103             "PUT" -> {
 104                 "Put response"
 105             }
 106             else -> {
 107                 "Get response"
 108             }
 109
 110         }
 111         mockServer.`when`(request().withHeaders(Header(HttpHeaders.AUTHORIZATION, headers[HttpHeaders.AUTHORIZATION]))
 112                 .withMethod(method)
 113                 .withPath(path)
 114                 .withQueryStringParameter("format", "resource")
 115                 .withBody(payload)
 116         ).respond(response().withStatusCode(200).withBody("{\"aai-resource\":\"$requestResponse\"}"))
 117     }
 118
 119     private fun setBasicAuth(username: String, password: String): String {
 120         val credentialsString = "$username:$password"
 121         return Base64.getEncoder().encodeToString(
 122                 credentialsString.toByteArray(Charset.defaultCharset()))
 123     }
 124 }