ms/blueprintsprocessor/modules/commons/grpc-lib/src/test/kotlin/org/onap/ccsdk/cds/blueprintsprocessor/grpc/service/MockTLSBluePrintProcessingServer.kt

   1 /*
   2  * Copyright © 2018-2019 AT&T Intellectual Property.
   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
  17 package org.onap.ccsdk.cds.blueprintsprocessor.grpc.service
  18
  19 import io.grpc.stub.StreamObserver
  20 import org.onap.ccsdk.cds.blueprintsprocessor.grpc.GRPCLibConstants
  21 import org.onap.ccsdk.cds.blueprintsprocessor.grpc.TLSAuthGrpcServerProperties
  22 import org.onap.ccsdk.cds.blueprintsprocessor.grpc.interceptor.GrpcServerLoggingInterceptor
  23 import org.onap.ccsdk.cds.controllerblueprints.common.api.EventType
  24 import org.onap.ccsdk.cds.controllerblueprints.common.api.Status
  25 import org.onap.ccsdk.cds.controllerblueprints.core.logger
  26 import org.onap.ccsdk.cds.controllerblueprints.processing.api.BluePrintProcessingServiceGrpc
  27 import org.onap.ccsdk.cds.controllerblueprints.processing.api.ExecutionServiceInput
  28 import org.onap.ccsdk.cds.controllerblueprints.processing.api.ExecutionServiceOutput
  29
  30
  31 val log = logger(MockTLSBluePrintProcessingServer::class)
  32
  33 /** For Integration testing stat this server, Set the working path to run this method */
  34 fun main() {
  35     try {
  36         val tlsAuthGrpcServerProperties = TLSAuthGrpcServerProperties().apply {
  37             port = 50052
  38             type = GRPCLibConstants.TYPE_TLS_AUTH
  39             certChain = "src/test/resources/tls-manual/py-executor-chain.pem"
  40             privateKey = "src/test/resources/tls-manual/py-executor-key.pem"
  41         }
  42         val server = TLSAuthGrpcServerService(tlsAuthGrpcServerProperties).serverBuilder()
  43                 .intercept(GrpcServerLoggingInterceptor())
  44                 .addService(MockTLSBluePrintProcessingServer())
  45                 .build()
  46         server.start()
  47         log.info("GRPC Serve started(${server.isShutdown}) on port(${server.port})...")
  48         server.awaitTermination()
  49     } catch (e: Exception) {
  50         log.error("Failed to start tls grpc integration server", e)
  51     }
  52
  53 }
  54
  55 class MockTLSBluePrintProcessingServer : BluePrintProcessingServiceGrpc.BluePrintProcessingServiceImplBase() {
  56     override fun process(responseObserver: StreamObserver<ExecutionServiceOutput>): StreamObserver<ExecutionServiceInput> {
  57
  58         return object : StreamObserver<ExecutionServiceInput> {
  59             override fun onNext(executionServiceInput: ExecutionServiceInput) {
  60                 log.info("Received requestId(${executionServiceInput.commonHeader.requestId})  " +
  61                         "subRequestId(${executionServiceInput.commonHeader.subRequestId})")
  62                 responseObserver.onNext(buildResponse(executionServiceInput))
  63                 responseObserver.onCompleted()
  64             }
  65
  66             override fun onError(error: Throwable) {
  67                 log.debug("Fail to process message", error)
  68                 responseObserver.onError(io.grpc.Status.INTERNAL
  69                         .withDescription(error.message)
  70                         .asException())
  71             }
  72
  73             override fun onCompleted() {
  74                 log.info("Completed")
  75             }
  76         }
  77     }
  78
  79     private fun buildResponse(input: ExecutionServiceInput): ExecutionServiceOutput {
  80         val status = Status.newBuilder().setCode(200)
  81                 .setEventType(EventType.EVENT_COMPONENT_EXECUTED)
  82                 .build()
  83         return ExecutionServiceOutput.newBuilder()
  84                 .setCommonHeader(input.commonHeader)
  85                 .setActionIdentifiers(input.actionIdentifiers)
  86                 .setStatus(status)
  87                 .build()
  88
  89     }
  90 }