vid-app-common/src/test/java/org/onap/vid/mso/MsoUtilTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * VID
   4  * ================================================================================
   5  * Copyright (C) 2017 - 2019 AT&T Intellectual Property. All rights reserved.
   6  * Modifications Copyright (C) 2019 Nokia. All rights reserved.
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *      http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  */
  21
  22 package org.onap.vid.mso;
  23
  24 import static org.apache.http.HttpStatus.SC_OK;
  25 import static org.assertj.core.api.Assertions.assertThat;
  26
  27 import io.joshworks.restclient.http.HttpResponse;
  28 import org.onap.vid.testUtils.TestUtils;
  29 import org.testng.annotations.Test;
  30
  31 public class MsoUtilTest {
  32
  33     @Test
  34     public void shouldWrapRestObject() {
  35         // given
  36         String entity = "entity";
  37         RestObject<String> restObject = new RestObject<>();
  38         restObject.set(entity);
  39         restObject.setStatusCode(SC_OK);
  40         // when
  41         MsoResponseWrapper result = MsoUtil.wrapResponse(restObject);
  42         // then
  43         assertThat(result.getEntity()).isEqualTo(entity);
  44         assertThat(result.getStatus()).isEqualTo(SC_OK);
  45     }
  46
  47     @Test
  48     public void shouldWrapHttpResponse() throws Exception {
  49         // given
  50         HttpResponse<String> httpResponse = TestUtils.createTestHttpResponse(SC_OK, null);
  51         // when
  52         MsoResponseWrapper result = MsoUtil.wrapResponse(httpResponse);
  53         // then
  54         assertThat(result.getEntity()).isEqualTo(null);
  55         assertThat(result.getStatus()).isEqualTo(SC_OK);
  56     }
  57
  58     @Test
  59     public void shouldWrapHttpResponseWithEntity() throws Exception {
  60         // given
  61         String entity = "entity";
  62         HttpResponse<String> httpResponse = TestUtils.createTestHttpResponse(SC_OK, entity);
  63         // when
  64         MsoResponseWrapper result = MsoUtil.wrapResponse(httpResponse);
  65         // then
  66         assertThat(result.getEntity()).isEqualTo(entity);
  67         assertThat(result.getStatus()).isEqualTo(SC_OK);
  68     }
  69
  70 }