src/test/java/org/onap/aai/modelloader/restclient/TracingTest.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2024 Deutsche Telekom AG Intellectual Property. All rights reserved.
   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 package org.onap.aai.modelloader.restclient;
  21
  22 import org.junit.jupiter.api.Test;
  23 import org.springframework.beans.factory.annotation.Autowired;
  24 import org.springframework.beans.factory.annotation.Value;
  25 import org.springframework.boot.test.context.SpringBootTest;
  26 import org.springframework.cloud.contract.wiremock.AutoConfigureWireMock;
  27 import org.springframework.http.HttpStatus;
  28 import org.springframework.web.client.RestTemplate;
  29
  30 import com.github.tomakehurst.wiremock.client.WireMock;
  31
  32 @SpringBootTest(properties = {
  33   "spring.sleuth.enabled=true",
  34   "spring.zipkin.baseUrl=http://localhost:${wiremock.server.port}"
  35 })
  36 @AutoConfigureWireMock(port = 0)
  37 public class TracingTest {
  38
  39   @Value("${wiremock.server.port}")
  40   private int wiremockPort;
  41
  42   @Autowired RestTemplate restTemplate;
  43
  44   @Test
  45   public void thatArtifactsCanBePushed() {
  46     WireMock.stubFor(
  47       WireMock.post(WireMock.urlEqualTo("/api/v2/spans"))
  48           .willReturn(
  49               WireMock.aResponse()
  50                   .withStatus(HttpStatus.OK.value())));
  51
  52     WireMock.stubFor(
  53       WireMock.get(WireMock.urlEqualTo("/"))
  54           .withHeader("X-B3-TraceId", WireMock.matching(".*"))
  55           .willReturn(
  56               WireMock.aResponse()
  57                   .withStatus(HttpStatus.OK.value())));
  58
  59
  60     String response = restTemplate.getForObject("http://localhost:" + wiremockPort + "/", String.class);
  61   }
  62
  63 }